"""Tests for account_tax_info serialization."""

from datetime import date
from decimal import Decimal

from marshmallow import ValidationError
import pytest

from abacus_account.constants.constants import (
    DEFAULT_PAGE_LIMIT, DEFAULT_PAGE_OFFSET, TAX_EMPLOYMENT_TYPES
)
from abacus_account.schemas.account_tax_info import (
    AccountTaxInfoDetailSchema, AccountTaxInfoListInputSchema, AccountTaxInfoPutSchema
)
from tests.utils.factories import AccountTaxInfoFactory


@pytest.mark.parametrize(
    'country_of_tax_residence',
    ('USA', None)
)
def test_account_tax_info_detail_schema(country_of_tax_residence):
    """Test account_tax_info detail schema serialization."""
    account_tax_info = AccountTaxInfoFactory.build(
        account_id=1,
        country_of_tax_residence=country_of_tax_residence
    )

    result = AccountTaxInfoDetailSchema().dump(account_tax_info)
    assert result == {
        'account_tax_info_id': account_tax_info.account_tax_info_id,
        'account_id': account_tax_info.account_id,
        'country_of_tax_residence': account_tax_info.country_of_tax_residence,
        'is_sba_signed': account_tax_info.is_sba_signed,
        'is_vat_exempt': account_tax_info.is_vat_exempt,
        'is_tax_treaty_claimed': account_tax_info.is_tax_treaty_claimed,
        'tax_employment_type': account_tax_info.tax_employment_type,
        'certificate_of_residence_expiration_date':
            account_tax_info.certificate_of_residence_expiration_date,
        'is_wht_applicable': account_tax_info.is_wht_applicable,
        'is_resident_of_spanish_islands':
            account_tax_info.is_resident_of_spanish_islands,
        'wht_rate_override': None
    }


def test_account_tax_info_put_schema():
    """Test account_tax_info put schema validation."""
    account_tax_info = AccountTaxInfoFactory.build(account_id=1)

    test_data = {
        'country_of_tax_residence': account_tax_info.country_of_tax_residence,
        'is_sba_signed': account_tax_info.is_sba_signed,
        'is_vat_exempt': account_tax_info.is_vat_exempt,
        'is_tax_treaty_claimed': account_tax_info.is_tax_treaty_claimed
    }

    result = AccountTaxInfoPutSchema().load(test_data)
    assert result == test_data


def test_account_tax_info_put_schema_no_cotr():
    """Test account_tax_info put schema validation."""
    account_tax_info = AccountTaxInfoFactory.build(account_id=1)

    test_data = {
        'country_of_tax_residence': None,
        'is_sba_signed': account_tax_info.is_sba_signed,
        'is_vat_exempt': account_tax_info.is_vat_exempt,
        'is_tax_treaty_claimed': account_tax_info.is_tax_treaty_claimed
    }

    result = AccountTaxInfoPutSchema().load(test_data)
    assert result == test_data


@pytest.mark.parametrize(
    'tax_employment_type',
    (*TAX_EMPLOYMENT_TYPES, None)
)
def test_account_tax_info_put_schema_spanish_taxes(tax_employment_type):
    """Test account_tax_info put schema validation."""
    account_tax_info = AccountTaxInfoFactory.build(
        account_id=1,
        tax_employment_type=tax_employment_type
    )

    test_data = {
        'country_of_tax_residence': account_tax_info.country_of_tax_residence,
        'is_sba_signed': account_tax_info.is_sba_signed,
        'is_vat_exempt': account_tax_info.is_vat_exempt,
        'is_tax_treaty_claimed': account_tax_info.is_tax_treaty_claimed,
        'tax_employment_type': account_tax_info.tax_employment_type,
        'certificate_of_residence_expiration_date':
            account_tax_info.certificate_of_residence_expiration_date,
        'is_wht_applicable': account_tax_info.is_wht_applicable,
        'is_resident_of_spanish_islands':
            account_tax_info.is_resident_of_spanish_islands,
    }

    result = AccountTaxInfoPutSchema().load(test_data)
    assert result == test_data


@pytest.mark.parametrize(
    'tax_employment_type',
    (*TAX_EMPLOYMENT_TYPES, None)
)
def test_account_tax_info_put_schema_german_taxes(tax_employment_type):
    """Test account_tax_info put schema validation."""
    account_tax_info = AccountTaxInfoFactory.build(
        account_id=1,
        tax_employment_type=tax_employment_type,
        wht_rate_override=Decimal('10.01')
    )

    test_data = {
        'country_of_tax_residence': account_tax_info.country_of_tax_residence,
        'is_sba_signed': account_tax_info.is_sba_signed,
        'is_vat_exempt': account_tax_info.is_vat_exempt,
        'is_tax_treaty_claimed': account_tax_info.is_tax_treaty_claimed,
        'tax_employment_type': account_tax_info.tax_employment_type,
        'is_wht_applicable': account_tax_info.is_wht_applicable,
        'wht_rate_override': account_tax_info.wht_rate_override
    }

    result = AccountTaxInfoPutSchema().load(test_data)
    assert result == test_data


@pytest.mark.parametrize(
    'data,expected',
    [
        (
            {
                'certificate_of_residence_expiration_date_start': '2024-01-01',
                'certificate_of_residence_expiration_date_end': '2024-12-31',
                'limit': 10,
                'offset': 5,
                'account_ids': [1, 2, 3]
            },
            {
                'certificate_of_residence_expiration_date_start': date(2024, 1, 1),  # noqa: E501
                'certificate_of_residence_expiration_date_end': date(2024, 12, 31),  # noqa: E501
                'limit': 10,
                'offset': 5,
                'account_ids': [1, 2, 3]
            }
        ),
        (
            {
                'certificate_of_residence_expiration_date_start': None,
                'certificate_of_residence_expiration_date_end': None,
                'account_ids': None
            },
            {
                'certificate_of_residence_expiration_date_start': None,
                'certificate_of_residence_expiration_date_end': None,
                'limit': DEFAULT_PAGE_LIMIT,
                'offset': DEFAULT_PAGE_OFFSET,
                'account_ids': None
            }
        ),
    ]
)
def test_account_tax_info_list_input_schema_success(data, expected):
    """Test AccountTaxInfoListInputSchema."""
    schema = AccountTaxInfoListInputSchema()
    result = schema.load(data)
    assert result == expected


@pytest.mark.parametrize(
    'data,field',
    [
        (
            {
                'certificate_of_residence_expiration_date_start': 'invalid-date',
                'certificate_of_residence_expiration_date_end': '2024-12-31',
                'account_ids': [1, 2]
            },
            'certificate_of_residence_expiration_date_start'
        ),
        (
            {
                'limit': 0,
                'offset': 5,
                'account_ids': [1]
            },
            'limit'
        ),
        (
            {
                'limit': 10,
                'offset': -1,
                'account_ids': [1]
            },
            'offset'
        ),
        (
            {
                'limit': 10,
                'offset': 5,
                'account_ids': ['not-an-int']
            },
            'account_ids'
        ),
        (
            {
                'certificate_of_residence_expiration_date_start': '2024-12-31',
                'certificate_of_residence_expiration_date_end': '2024-01-01',
                'limit': 10,
                'offset': 5,
                'account_ids': [1, 2]
            },
            '_schema'
        ),
    ]
)
def test_account_tax_info_list_input_schema_failure(data, field):
    """Test AccountTaxInfoListInputSchema."""
    schema = AccountTaxInfoListInputSchema()
    with pytest.raises(ValidationError) as exc_info:
        schema.load(data)
    assert field in exc_info.value.messages