"""Test Account Tax Info History schemas."""

import pytest
from abacus_common_logic.utils.dates import safe_format_date

from abacus_account.schemas.account_tax_info_history import AccountTaxInfoHistorySchema
from abacus_account.tests.utils.factories import AccountTaxInfoHistoryFactory


@pytest.mark.parametrize('country_of_tax_residence', ('USA', None))
def test_account_tax_info_history_schema(country_of_tax_residence):
    """Test account_tax_info_history serialization."""
    history = AccountTaxInfoHistoryFactory.create(
        country_of_tax_residence=country_of_tax_residence
    )
    res = AccountTaxInfoHistorySchema().dump(history)

    assert res['account_tax_info_history_id'] == history.account_tax_info_history_id
    assert res['account_id'] == history.account_id
    assert res['country_of_tax_residence'] == history.country_of_tax_residence
    assert res['is_sba_signed'] == history.is_sba_signed
    assert res['is_vat_exempt'] == history.is_vat_exempt
    assert res['is_tax_treaty_claimed'] == history.is_tax_treaty_claimed
    assert res['tax_employment_type'] == history.tax_employment_type
    assert (
        res['certificate_of_residence_expiration_date']
        == history.certificate_of_residence_expiration_date
    )
    assert res['is_wht_applicable'] == history.is_wht_applicable
    assert (
        res['is_resident_of_spanish_islands'] == history.is_resident_of_spanish_islands
    )
    assert res['wht_rate_override'] == history.wht_rate_override
    assert res['created_at'] == safe_format_date(history.created_at)
    assert res['last_modified'] == safe_format_date(history.last_modified)