"""Factory for the VendorRoles model."""

import factory

from users.models.user_info import VendorRoles


class VendorRolesFactory(factory.Factory):
    """Factory for creating VendorRoles instances for testing."""

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

        model = VendorRoles

    _id = factory.Sequence(lambda n: n)
    role = factory.Sequence(lambda n: n)