"""Factory for creating release_artists associated with a release."""

import factory

from product_digital.models import release_artist


class ReleaseArtistFactory(factory.Factory):
    """Factory for test release artist."""

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

        model = release_artist.ReleaseArtist

    release_id = factory.Sequence(lambda n: n)
    release_artist_id = factory.Sequence(lambda n: n)
    artist_info_id = factory.Sequence(lambda n: n)
    upc = factory.Sequence(lambda n: n)
    role = 'primary_artist'
    artist_name = factory.Sequence(lambda n: 'TESTNAME{}'.format(n))