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

import factory

from product_digital.models import catalog_transfer


class CatalogTransferFactory(factory.Factory):
    """Factory for test catalog_transfers records."""

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

        model = catalog_transfer.CatalogTransfer

    release_id = factory.Sequence(lambda n: n)
    original_distributor = factory.Sequence(lambda n: 'Test Distributor {}'.format(n))
    same_isrc_upc = 'Y'