"""Factory for the Vendor model."""

import factory

from users.models.user_info import Vendor


class VendorFactory(factory.Factory):
    """Factory for creating Vendor instances for testing."""

    class Meta:
        """Meta definition for the factory."""

        model = Vendor

    vendor_id = factory.Sequence(lambda n: n)
    company = factory.Sequence(lambda n: n)
    is_distributor = factory.Sequence(lambda n: n)
    language = factory.Sequence(lambda n: n)
    label_identifier = 'Catalog'
    name = factory.Sequence(lambda n: 'label {}'.format(n))
    number_format = factory.Sequence(lambda n: n)
    owner = factory.Sequence(lambda n: n)
    status = 'signed'
    support_contact_email = factory.Sequence(lambda n: 'email {}'.format(n))
    wel_email_sender = 2233
    migrated_to_abacus = 0