from abacus_common_logic.models.base import db

from payee.models.reference_payment_entity import ReferencePaymentEntity


def test_get_all(fresh_db, reference_payment_entity_fixtures):
    result = ReferencePaymentEntity.get_all()
    assert (
        result
        == db.session.execute('select * from reference_payment_entity').fetchall()
    )


def test_get(fresh_db, reference_payment_entity_fixtures):
    result = ReferencePaymentEntity.get_by_id(1)
    assert (
        result
        == db.session.execute(
            'select * from reference_payment_entity where reference_payment_entity_id=1'
        ).first()
    )