"""Factory for the VendContactRoles model."""

import factory

from users.models.user_info import VendContactRoles


class VendContactRolesFactory(factory.Factory):
    """Factory for creating VendContactRoles instances for testing."""

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

        model = VendContactRoles

    _id = factory.Sequence(lambda n: n)
    vend_contact_id = factory.Sequence(lambda n: n)
    role_id = factory.Sequence(lambda n: n)