"""Factories for tests."""

from polyfactory.factories.pydantic_factory import ModelFactory

from src.models import (
    AbacusState,
    AccountPayee,
    AccountTaxInfo,
    NewTaxCorrection,
    NewTaxCorrectionVAT,
    NewTaxDetails,
    PayeeDetails,
    PaymentHold,
    PaymentHoldResult,
    PostTaxFormsInput,
    S3Event,
    StatementPeriod,
    TaxCorrection,
    TaxCorrectionVAT,
    TaxDetails,
    TaxDetailsDataloaderItem,
    TaxFormInfoDetailsItem,
    TaxFormInfoItem,
    UpdateAccountTaxInfo,
    USTaxFormW8BEN,
    USTaxFormW8BENE,
    USTaxFormW8ECI,
    USTaxFormW8IMY,
    USTaxFormW9,
)


class StatementPeriodFactory(ModelFactory[StatementPeriod]):
    """StatmentPeriod factory."""

    __model__ = StatementPeriod


class TaxCorrectionFactory(ModelFactory[TaxCorrection]):
    """TaxCorrection factory."""

    __model__ = TaxCorrection


class TaxCorrectionVATFactory(ModelFactory[TaxCorrectionVAT]):
    """TaxCorrectionVAT factory."""

    __model__ = TaxCorrectionVAT


class NewTaxCorrectionFactory(ModelFactory[NewTaxCorrection]):
    """NewTaxCorrection factory."""

    __model__ = NewTaxCorrection


class NewTaxCorrectionVATFactory(ModelFactory[NewTaxCorrectionVAT]):
    """NewTaxCorrectionVAT factory."""

    __model__ = NewTaxCorrectionVAT


class S3EventFactory(ModelFactory[S3Event]):
    """S3Event factory."""

    __model__ = S3Event


class PaymentHoldFactory(ModelFactory[PaymentHold]):
    """PaymentHold factory."""

    __model__ = PaymentHold


class PaymentHoldResultFactory(ModelFactory[PaymentHoldResult]):
    """PaymentHoldResult factory."""

    __model__ = PaymentHoldResult


class PayeeDetailsFactory(ModelFactory[PayeeDetails]):
    """PayeeDetails factory."""

    __model__ = PayeeDetails


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

    __model__ = AccountPayee


class AbacusStateFactory(ModelFactory[AbacusState]):
    """AbacusState factory."""

    __model__ = AbacusState


class TaxFormInfoDetailsItemFactory(ModelFactory[TaxFormInfoDetailsItem]):
    """TaxFormInfoDetailsItem factory."""

    __model__ = TaxFormInfoDetailsItem


class TaxFormInfoItemFactory(ModelFactory[TaxFormInfoItem]):
    """TaxFormInfoItem factory."""

    __model__ = TaxFormInfoItem


class USTaxFormW9Factory(ModelFactory[USTaxFormW9]):
    """USTaxFormW9 factory."""

    __model__ = USTaxFormW9


class USTaxFormW8BENFactory(ModelFactory[USTaxFormW8BEN]):
    """USTaxFormW8BEN factory."""

    __model__ = USTaxFormW8BEN


class USTaxFormW8BENEFactory(ModelFactory[USTaxFormW8BENE]):
    """USTaxFormW8BENE factory."""

    __model__ = USTaxFormW8BENE


class USTaxFormW8IMYFactory(ModelFactory[USTaxFormW8IMY]):
    """USTaxFormW8IMY factory."""

    __model__ = USTaxFormW8IMY


class USTaxFormW8ECIFactory(ModelFactory[USTaxFormW8ECI]):
    """USTaxFormW8ECI factory."""

    __model__ = USTaxFormW8ECI


class PostTaxFormsInputFactory(ModelFactory[PostTaxFormsInput]):
    """PostTaxFormsInput factory."""

    __model__ = PostTaxFormsInput


class AccountTaxInfoFactory(ModelFactory[AccountTaxInfo]):
    """AccountTaxInfo factory."""

    __model__ = AccountTaxInfo


class TaxDetailsFactory(ModelFactory[TaxDetails]):
    """TaxDetails factory."""

    __model__ = TaxDetails


class NewTaxDetailsFactory(ModelFactory[NewTaxDetails]):
    """NewTaxDetails factory."""

    __model__ = NewTaxDetails


class UpdateAccountTaxInfoFactory(ModelFactory[UpdateAccountTaxInfo]):
    """AccountTaxInfoFactory."""

    __model__ = UpdateAccountTaxInfo


class TaxDetailsDataloaderItemFactory(ModelFactory[TaxDetailsDataloaderItem]):
    """TaxDetailsDataloaderItem factory."""

    __model__ = TaxDetailsDataloaderItem