"""Tests for reference_payment_entity model."""

from abacus_contract.models.reference_payment_entity import ReferencePaymentEntity


def test_create_reference_payment_entity():
    """Test to create reference_payment_entity record."""
    reference_payment_entity = ReferencePaymentEntity.create(
        payment_entity_name='KNR-UK',
        country_of_tax_reporting='UK'
    )
    result = ReferencePaymentEntity.query.all()
    assert len(result) == 1
    assert result[0] == reference_payment_entity