"""Factories for the Artist model."""

import factory

from artist.models.artist import Artist


class ArtistFactory(factory.Factory):
    """Artist factory class."""

    class Meta:
        """Meta class representing the model."""

        model = Artist

    artist_id = factory.Sequence(lambda n: n)
    vendor_id = factory.Sequence(lambda n: n)
    country_id = factory.Sequence(lambda n: n)
    name = factory.Faker('name')
    artist_type = 'artist'