"""Factory for the VendorRestrictedFeature model."""

import factory

from users.models.user_info import VendorRestrictedFeature


class VendorRestrictedFeatureFactory(factory.Factory):
    """Factory for creating VendorRestrictedFeature instances for testing."""

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

        model = VendorRestrictedFeature

    vendor_restricted_features_id = 1
    vendor_id = 1
    feature_id = 1
    preview = 1