"""Test for ledger_vat_summary schema."""
import datetime

from moneyhub.constants.constants import VatCategory
from moneyhub.schemas.ledger_vat_summary import LedgerVatSummaryBackfillSchema
from moneyhub.schemas.ledger_vat_summary import LedgerVatSummarySchema


def test_ledger_vat_summary_backfill_schema():
    """Test ledger vat summary schema."""
    expected = LedgerVatSummaryBackfillSchema(
        account_id=1,
        contract_id=1,
        vat_category=VatCategory.GROSS_REVENUE,
        payee_currency_code='USD',
        vat_currency_code='USD',
        base_amount_payee_currency=0.001406240000,
        vat_rate=None,
        vat_amount_payee_currency=None,
        vat_amount_vat_currency=None,
        wht_amount_payee_currency=None,
        wht_amount_vat_currency=None,
        wht_rate=None,
        net_amount_payee_currency=0.001406240000,
    )
    data = {
        'account_id': 1,
        'contract_id': 1,
        'vat_category': VatCategory.GROSS_REVENUE,
        'payee_currency_code': 'USD',
        'vat_currency_code': 'USD',
        'base_amount_payee_currency': 0.001406240000,
        'vat_rate': None,
        'vat_amount_payee_currency': None,
        'vat_amount_vat_currency': None,
        'wht_amount_payee_currency': None,
        'wht_amount_vat_currency': None,
        'wht_rate': None,
        'net_amount_payee_currency': 0.001406240000,
    }

    actual = LedgerVatSummaryBackfillSchema(**data)

    assert actual == expected


def test_ledger_vat_summary_schema():
    """Test ledger vat summary schema."""
    expected = LedgerVatSummarySchema(
        ledger_vat_summary_id=1,
        abacus_event_id=1,
        account_id=1,
        statement_period_id=285,
        activity_statement_period_id=284,
        contract_id=1,
        vat_category=VatCategory.GROSS_REVENUE,
        payee_currency_code='USD',
        vat_currency_code='USD',
        base_amount_payee_currency=0.001406240000,
        vat_rate=None,
        vat_amount_payee_currency=None,
        vat_amount_vat_currency=None,
        wht_amount_payee_currency=None,
        wht_amount_vat_currency=None,
        wht_rate=None,
        net_amount_payee_currency=0.001406240000,
        abacus_exempt_reason=None,
        is_reporting_only=True,
        created_at=datetime.datetime(2022, 11, 17, 22, 2, 42),
        created_by='me',
        last_modified=datetime.datetime(2022, 12, 17, 22, 2, 42),
        last_modified_by='me'
    )
    data = {
        'ledger_vat_summary_id': 1,
        'abacus_event_id': 1,
        'account_id': 1,
        'statement_period_id': 285,
        'activity_statement_period_id': 284,
        'contract_id': 1,
        'vat_category': VatCategory.GROSS_REVENUE,
        'payee_currency_code': 'USD',
        'vat_currency_code': 'USD',
        'base_amount_payee_currency': 0.001406240000,
        'vat_rate': None,
        'vat_amount_payee_currency': None,
        'vat_amount_vat_currency': None,
        'wht_amount_payee_currency': None,
        'wht_amount_vat_currency': None,
        'wht_rate': None,
        'net_amount_payee_currency': 0.001406240000,
        'abacus_exempt_reason': None,
        'is_reporting_only': True,
        'created_at': '2022-11-17T22:02:42',
        'created_by': 'me',
        'last_modified': '2022-12-17T22:02:42',
        'last_modified_by': 'me'
    }

    actual = LedgerVatSummarySchema(**data)

    assert expected == actual