"""Factory for the VendorPrivilege model."""

import factory

from users.models.user_info import VendorPrivilege


class VendorPrivilegeFactory(factory.Factory):
    """Factory for creating VendorPrivilege instances for testing."""

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

        model = VendorPrivilege

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