"""AccountPayee model tests.""" from abacus_account.models.account_payee import AccountPayee from abacus_account.tests.utils.factories import ( AccountFactory, AccountPayeeFactory, ReferencePaymentTypeFactory, ) def test_create(): """Create an account payee.""" account = AccountFactory.create() AccountPayee.create( account_id=account.account_id, payoneer_payee_id=20, payoneer_payee_name='Bad Bunny', payoneer_iframe_url='payoneer.com/iframedrogerabbit', payoneer_iframe_url_date='2022-03-01', payoneer_session_id='2000', ) results = AccountPayee.query.all() assert len(results) == 1 def test_create_with_payment_type(): """Create an account payee.""" account = AccountFactory.create() reference_payment_type = ReferencePaymentTypeFactory.create() AccountPayee.create( account_id=account.account_id, payoneer_payee_id=19, payoneer_payee_name='Person', payoneer_iframe_url='payoneer.com/iframedrogerabbit', payoneer_iframe_url_date='2023-05-05', payoneer_session_id='2001', reference_payment_type_id=reference_payment_type.reference_payment_type_id, ) results = AccountPayee.query.all() assert len(results) == 1 def test_get_by_account_id(): """Test getting account payee by account ID.""" account_payee = AccountPayeeFactory.create() result = AccountPayee.get_by_account_id(account_payee.account_id) assert result.account_id == account_payee.account_id assert result.account_payee_id == account_payee.account_payee_id assert result.payoneer_program_id == account_payee.payoneer_program_id assert result.payoneer_payee_id == account_payee.payoneer_payee_id assert result.payoneer_payee_name == account_payee.payoneer_payee_name assert result.payoneer_iframe_url == account_payee.payoneer_iframe_url assert result.payoneer_iframe_url_date == account_payee.payoneer_iframe_url_date assert result.payoneer_session_id == account_payee.payoneer_session_id assert result.sap_vendor_id == account_payee.sap_vendor_id assert result.reference_payment_type_id == account_payee.reference_payment_type_id