"""Factories for tests."""

from polyfactory.factories.pydantic_factory import ModelFactory

from src.models.ows_abacus_account import AccountPayeeModel, AccountPaymentTermModel
from src.models.ows_abacus_contract import ReferencePaymentEntityModel
from src.models.ows_abacus_state import (
    AbacusStateGetResponseItemModel,
    AbacusStatePutResponseModel,
)
from src.models.ows_payee import KycNotificationCreateResponseModel


class KycNotificationCreateResponseFactory(
    ModelFactory[KycNotificationCreateResponseModel]
):
    """KycNotificationCreateResponseModel factory."""

    __model__ = KycNotificationCreateResponseModel


class AbacusStateGetResponseItemFactory(ModelFactory[AbacusStateGetResponseItemModel]):
    """AbacusStateGetResponseItemModel factory."""

    __model__ = AbacusStateGetResponseItemModel


class AbacusStatePutResponseFactory(ModelFactory[AbacusStatePutResponseModel]):
    """AbacusStatePutResponseModel factory."""

    __model__ = AbacusStatePutResponseModel


class AccountPayeeFactory(ModelFactory[AccountPayeeModel]):
    """AccountPayeeModel factory."""

    __model__ = AccountPayeeModel


class AccountPaymentTermFactory(ModelFactory[AccountPaymentTermModel]):
    """AccountPaymentTermModel factory."""

    __model__ = AccountPaymentTermModel


class ReferencePaymentEntityFactory(ModelFactory[ReferencePaymentEntityModel]):
    """ReferencePaymentEntityModel factory."""

    __model__ = ReferencePaymentEntityModel