"""Test PaymentEntityPayoneerProgram schemas."""

from abacus_account.schemas.payment_entity_payoneer_program import (
    PaymentEntityPayoneerProgramListSchema,
    PaymentEntityPayoneerProgramSchema,
)
from abacus_account.tests.utils.factories import (
    PaymentEntityPayoneerProgramFactory,
    ReferencePayoneerProgramFactory,
)


def test_payment_entity_payoneer_program_schema(reference_payment_entity_fixture):
    """Test payment_entity_payoneer_program serialization."""
    payoneer_program = ReferencePayoneerProgramFactory.create(payoneer_program_id=201)

    payment_entity_payoneer_program = PaymentEntityPayoneerProgramFactory.create(
        reference_payment_entity_id=5,
        payoneer_program_id=payoneer_program.payoneer_program_id,
    )

    res = PaymentEntityPayoneerProgramSchema().dump(payment_entity_payoneer_program)
    assert (
        res['payment_entity_payoneer_program_id']
        == payment_entity_payoneer_program.payment_entity_payoneer_program_id
    )
    assert res['payment_currency'] == payment_entity_payoneer_program.payment_currency
    assert (
        res['payoneer_program_id']
        == payment_entity_payoneer_program.payoneer_program_id
    )
    assert res['agreement_type_id'] == payment_entity_payoneer_program.agreement_type_id
    assert (
        res['reference_payment_entity_id']
        == payment_entity_payoneer_program.reference_payment_entity_id
    )


def test_payment_entity_payoneer_program_list_schema():
    """Test payment_entity_payoneer_program list serialization."""
    mock_data = {
        'agreement_type': 'AWAL Core',
        'payment_currency': 'EUR',
        'payoneer_program_id': 100176119,
        'reference_payment_entity_id': 1,
    }

    result = PaymentEntityPayoneerProgramListSchema().dump(mock_data)
    assert result == mock_data