Odoo18-Base/addons/account_payment/tests/common.py
2025-03-10 11:12:23 +07:00

95 lines
3.7 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from unittest.mock import patch
from contextlib import contextmanager
from odoo.addons.account.tests.common import AccountTestInvoicingCommon
from odoo.addons.payment.tests.common import PaymentCommon
class AccountPaymentCommon(PaymentCommon, AccountTestInvoicingCommon):
@classmethod
def setUpClass(cls, *kw):
# chart_template_ref is dropped on purpose because not needed for account_payment tests.
super().setUpClass()
with cls.mocked_get_payment_method_information(cls):
cls.dummy_provider_method = cls.env['account.payment.method'].sudo().create({
'name': 'Dummy method',
'code': 'none',
'payment_type': 'inbound'
})
cls.dummy_provider.journal_id = cls.company_data['default_journal_bank']
cls.account = cls.company.account_journal_payment_credit_account_id
cls.invoice = cls.env['account.move'].create({
'move_type': 'entry',
'date': '2019-01-01',
'currency_id': cls.currency_euro.id,
'partner_id': cls.partner.id,
'line_ids': [
(0, 0, {
'account_id': cls.account.id,
'debit': 100.0,
'credit': 0.0,
'amount_currency': 200.0,
}),
(0, 0, {
'account_id': cls.account.id,
'debit': 0.0,
'credit': 100.0,
'amount_currency': -200.0,
}),
],
})
def setUp(self):
self.enable_reconcile_after_done_patcher = False
super().setUp()
#=== Utils ===#
@contextmanager
def mocked_get_payment_method_information(self):
Method_get_payment_method_information = self.env['account.payment.method']._get_payment_method_information
def _get_payment_method_information(*args, **kwargs):
res = Method_get_payment_method_information()
res['none'] = {'mode': 'electronic', 'domain': [('type', '=', 'bank')]}
return res
with patch.object(self.env.registry['account.payment.method'], '_get_payment_method_information', _get_payment_method_information):
yield
@contextmanager
def mocked_get_default_payment_method_id(self):
def _get_default_payment_method_id(*args, **kwargs):
return self.dummy_provider_method.id
with patch.object(self.env.registry['payment.provider'], '_get_default_payment_method_id', _get_default_payment_method_id):
yield
@classmethod
def _prepare_provider(cls, provider_code='none', company=None, update_values=None):
""" Override of `payment` to prepare and return the first provider matching the given
provider and company.
If no provider is found in the given company, we duplicate the one from the base company.
All other providers belonging to the same company are disabled to avoid any interferences.
:param str provider_code: The code of the provider to prepare.
:param recordset company: The company of the provider to prepare, as a `res.company` record.
:param dict update_values: The values used to update the provider.
:return: The provider to prepare, if found.
:rtype: recordset of `payment.provider`
"""
provider = super()._prepare_provider(provider_code, company, update_values)
if not provider.journal_id:
provider.journal_id = cls.env['account.journal'].search(
[('company_id', '=', provider.company_id.id), ('type', '=', 'bank')],
limit=1,
)
return provider