"""Product factory."""

from factory import Factory
from factory.fuzzy import FuzzyChoice, FuzzyInteger

from video.constants.product import DISTRIBUTION_FORMAT_IDS
from video.models.ows.classes import product


class ProductFactory(Factory[product.Product]):
    """ProductFactory Class."""

    class Meta:
        """ProductFactory Meta Class."""

        model = product.Product

    product_id = FuzzyInteger(1, 999999)
    vendor_id = FuzzyInteger(1, 999999)
    subaccount_id = FuzzyInteger(0, 999999)
    distribution_format_id: int = FuzzyChoice(DISTRIBUTION_FORMAT_IDS)  # type: ignore[assignment]