"""Factory for creating lyrics records in tests."""

import factory

from lyrics.models import lyrics


class LyricsFactory(factory.Factory):
    """Factory for test lyrics."""

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

        model = lyrics.Lyrics

    highlight_id = factory.Sequence(lambda n: n + 1)
    mkt_program_id = lyrics.LYRICS_PROGRAM_ID
    entity = lyrics.TRACK_ENTITY
    entity_id = factory.Sequence(lambda n: n + 1)
    subject = None
    description = factory.Sequence(lambda n: str('Lyrics {}'.format(n + 1)))
    attachment = 'N'
    scope = 'internal_only'
    client = lyrics.ALW_CLIENT