"""Test account_contract_vat_detail schema."""

from abacus_account.schemas import AccountContractVatDetailSchema
from abacus_account.tests.utils.factories import AccountContractVatDetailFactory


def test_account_contract_vat_detail_schema(
    fresh_db, reference_signing_entity_fixture, account_contract_vat_detail_fixtures
):
    """Test account_contract_vat_detail schema."""
    account_vat_detail = AccountContractVatDetailFactory.build()
    res = AccountContractVatDetailSchema().dump(account_vat_detail)

    assert res == {
        'account_contract_vat_detail_id': account_vat_detail.account_contract_vat_detail_id,
        'contract_id': account_vat_detail.contract_id,
        'account_id': account_vat_detail.account_id,
        'is_vat_registered': account_vat_detail.is_vat_registered,
        'is_vat_eligible': account_vat_detail.is_vat_eligible,
        'vat_country': account_vat_detail.vat_country,
        'company_code': account_vat_detail.company_code,
    }