from polyfactory.factories.pydantic_factory import ModelFactory

from src.models import (
    Event,
    MovePayoneerProgramData,
    PayoneerStatus,
    PayoneerStatusData,
)


class EventFactory(ModelFactory[Event]):
    __model__ = Event


class PayoneerStatusFactory(ModelFactory[PayoneerStatus]):
    __model__ = PayoneerStatus


class PayoneerStatusDataFactory(ModelFactory[PayoneerStatusData]):
    __model__ = PayoneerStatusData


class MovePayoneerProgramDataFactory(ModelFactory[MovePayoneerProgramData]):
    __model__ = MovePayoneerProgramData