"""Factory for creating marketing highlights records in tests."""

import datetime

import factory

from product_digital_marketing.models import highlight


class HighlightFactory(factory.Factory):
    """Factory for test highlights."""

    class Meta:
        """Meta class around the highlight model."""

        model = highlight.MarketingHighlight

    pk = factory.Sequence(lambda n: n + 1)
    product_id = factory.Sequence(lambda n: n + 1)
    global_highlight = factory.Sequence(
        lambda n: str('General drivers {}'.format(n + 1)))
    spotify_highlight = factory.Sequence(
        lambda n: str('Spotify Drivers {}'.format(n + 1)))
    apple_highlight = factory.Sequence(
        lambda n: str('Apple Drivers {}'.format(n + 1)))
    spotify_total = factory.Sequence(lambda n: n + 1000)
    apple_total = factory.Sequence(lambda n: n + 2000)
    downloads_total = factory.Sequence(lambda n: n + 3000)
    updated_by = 'ALW:123456'
    updated_at = factory.Sequence(lambda n: datetime.datetime.now())