"""Tests for reference_payment_entity model."""

from abacus_account.models.reference_payment_entity import ReferencePaymentEntity


def test_get_knr_payment_entities():
    """Test getting KNR reference payment entities."""
    result = ReferencePaymentEntity.get_knr_payment_entities()

    assert all(['knr' in res.payment_entity_name.lower() for res in result])


def test_get_payment_entity_payoneer_programs(
    payoneer_programs_fixture,
    reference_payment_entity_fixture
):
    """Test to get all payoneer programs."""
    items, total_count = \
        ReferencePaymentEntity.get_payment_entity_payoneer_programs(10, 0)

    assert total_count == 7

    assert items[0].reference_payment_entity_id == 1
    assert items[0].agreement_type is None
    assert items[0].reference_agreement_type_id is None
    assert items[0].payment_currency is None
    assert items[0].payoneer_program_id is None
    assert items[0].reference_payment_type_id is None
    assert items[0].payment_service is None

    assert items[1].reference_payment_entity_id == 2
    assert items[1].agreement_type is None
    assert items[1].reference_agreement_type_id is None
    assert items[1].payment_currency is None
    assert items[1].payoneer_program_id is None
    assert items[1].reference_payment_type_id is None
    assert items[1].payment_service is None

    assert items[2].reference_payment_entity_id == 3
    assert items[2].agreement_type is None
    assert items[2].reference_agreement_type_id is None
    assert items[2].payment_currency is None
    assert items[2].payoneer_program_id is None
    assert items[2].reference_payment_type_id is None
    assert items[2].payment_service is None

    assert items[3].reference_payment_entity_id == 4
    assert items[3].agreement_type is None
    assert items[3].reference_agreement_type_id is None
    assert items[3].payment_currency is None
    assert items[3].payoneer_program_id is None
    assert items[3].reference_payment_type_id is None
    assert items[3].payment_service is None

    assert items[4].reference_payment_entity_id == 5
    assert items[4].agreement_type == 'AWAL Core'
    assert items[4].reference_agreement_type_id == '1'
    assert items[4].payment_currency == 'GBP'
    assert items[4].payoneer_program_id == 10017819
    assert items[4].reference_payment_type_id == 7
    assert items[4].payment_service == 'service'

    assert items[5].reference_payment_entity_id == 6
    assert items[5].agreement_type == 'AWAL Core'
    assert items[5].reference_agreement_type_id == '1'
    assert items[5].payment_currency == 'USD'
    assert items[5].payoneer_program_id == 10017820
    assert items[5].reference_payment_type_id == 7
    assert items[5].payment_service == 'service'

    assert items[6].reference_payment_entity_id == 8
    assert items[6].agreement_type == 'AWAL Core'
    assert items[6].reference_agreement_type_id == '1'
    assert items[6].payment_currency == 'EUR'
    assert items[6].payoneer_program_id == 10017820
    assert items[6].reference_payment_type_id == 7
    assert items[6].payment_service == 'service'


def test_get_by_id():
    """Test getting reference payment entities by ID."""
    result = ReferencePaymentEntity.get_by_id(2)
    assert result.reference_payment_entity_id == 2
    assert result.payment_entity_name == 'AWAL-US'

    result = ReferencePaymentEntity.get_by_id(1000)
    assert result is None