"""GrassInfo factory."""

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

from video.constants import access_control
from video.models.ows.classes import grass_info

from tests.unit.factories import custom_fuzzy_fields


class GrassInfoFactory(Factory[grass_info.GrassInfo]):
    """GrassInfoFactory Class."""

    class Meta:
        """GrassInfoFactory Meta Class."""

        model = grass_info.GrassInfo

    grass_account_type: str = FuzzyChoice(access_control.ACCOUNT_TYPES)  # type: ignore[assignment]
    grass_account_id = FuzzyInteger(1, 999999)
    orchard_user_id = custom_fuzzy_fields.FuzzyOrchardUserId()