"""Test PaymentEntityPayoneerProgram schemas."""

from abacus_account.schemas.payment_entity_payoneer_program import \
    PaymentEntityPayoneerProgramListSchema
from abacus_account.schemas.payment_entity_payoneer_program import \
    PaymentEntityPayoneerProgramSchema
from 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,
        'reference_payment_type_id': 7,
        'payment_service': 'service',
    }

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