"""Factory for creating marketing highlight projection records in tests."""

import datetime

import factory

from product_digital_marketing.models import highlight_projection


class HighlightProjectionFactory(factory.Factory):
    """Factory for test highlight projections."""

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

        model = highlight_projection.MarketingHighlightProjection

    pk = factory.Sequence(lambda n: n + 1)
    product_id = factory.Sequence(lambda n: n + 1)
    territory_id = factory.Sequence(lambda n: n + 1)
    spotify_projection = factory.Sequence(lambda n: n + 1)
    apple_projection = factory.Sequence(lambda n: n + 1)
    downloads_projection = factory.Sequence(lambda n: n + 1)
    highlight = factory.Sequence(
        lambda n: str('General drivers {}'.format(n + 1)))
    priority = 'A'
    updated_by = 'ALW:123456'
    updated_at = factory.Sequence(lambda n: datetime.datetime.now())