"""Test bank details model.""" from unittest.mock import patch from payee.logic.secure_document import ( create_secure_document_details, has_secure_document_details, ) from payee.models.bank_details import BankDetailsDocument from tests.constants import MOCK_AUDIT_FIELDS from tests.utils.factories import ( BankDetailsIndividualTypeFlattenFactory, BankDetailsWithAuditIndividualTypeFlattenFactory, ) @patch('payee.logic.secure_document.get_audit_fields') def test_bank_details(mock_get_audit_fields, mock_config): account_payee_id = '1' mock_get_audit_fields.return_value = MOCK_AUDIT_FIELDS banking_params = BankDetailsIndividualTypeFlattenFactory.build() create_params = {'account_payee_id': account_payee_id, **banking_params} create_secure_document_details( mock_config.SDM_CONFIG, BankDetailsDocument, **create_params ) result = has_secure_document_details(account_payee_id, BankDetailsDocument) assert ( result.original_values == BankDetailsWithAuditIndividualTypeFlattenFactory.build(**MOCK_AUDIT_FIELDS) )