"""AccountPayee model tests."""

from abacus_state.models.account_payee import AccountPayee


def test_get_payees_by_ids(
    account_fixtures, payoneer_program_fixtures, account_payee_fixtures
):
    """Get account payees by ids."""
    account_payee_id = 1
    results = AccountPayee.get_payees_by_ids([account_payee_id])
    assert len(results) == 1
    assert results[0].account_payee_id == account_payee_id


def test_get_payees_by_ids_none_found(account_fixtures, payoneer_program_fixtures):
    """Get account payees by ids."""
    account_payee_id = 999
    results = AccountPayee.get_payees_by_ids([account_payee_id])
    assert len(results) == 0