"""Tests for reference_payment_entity serialization."""

from abacus_contract.schemas.reference_payment_entity import \
    ReferencePaymentEntitySchema
from tests.utils.factories import ReferencePaymentEntityFactory


def test_reference_payment_entity_schema():
    """Test reference_payment_entity serialization."""
    reference_payment_entity = ReferencePaymentEntityFactory.create()

    result = ReferencePaymentEntitySchema().dump(reference_payment_entity)
    assert result == {
        'reference_payment_entity_id':
            reference_payment_entity.reference_payment_entity_id,
        'payment_entity_name': 'AWAL-UK',
        'country_of_tax_reporting': 'UK'
    }