diff --git a/contract/tests/__init__.py b/contract/tests/__init__.py index 9de1b60ae..9a3586ace 100644 --- a/contract/tests/__init__.py +++ b/contract/tests/__init__.py @@ -1,5 +1,4 @@ -# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). - from . import test_contract from . import test_contract_manually_create_invoice from . import test_portal +from . import test_multicompany diff --git a/contract/tests/test_contract.py b/contract/tests/test_contract.py index 37eea84f6..087cea7ac 100644 --- a/contract/tests/test_contract.py +++ b/contract/tests/test_contract.py @@ -18,7 +18,7 @@ def to_date(date): return fields.Date.to_date(date) -class TestContractBase(common.TransactionCase): +class TestContractBase: @classmethod def setUpClass(cls): super().setUpClass() @@ -2370,3 +2370,7 @@ class TestContract(TestContractBase): action = self.contract.action_preview() self.assertIn("/my/contracts/", action["url"]) self.assertIn("access_token=", action["url"]) + + +class TestContractCase(TestContract, common.SavepointCase): + """ TESTS """ diff --git a/contract/tests/test_multicompany.py b/contract/tests/test_multicompany.py new file mode 100644 index 000000000..734e6c81c --- /dev/null +++ b/contract/tests/test_multicompany.py @@ -0,0 +1,113 @@ +# Copyright 2021 ACSONE SA/NV () +# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html). +from odoo.addons.account.tests.common import AccountTestInvoicingCommon + +from .test_contract import TestContractBase + + +class ContractMulticompanyCase(TestContractBase): + @classmethod + def setUpClass(cls): + super().setUpClass() + + cls.company_obj = cls.env["res.company"] + cls.company_1 = cls.env.ref("base.main_company") + cls.company_2 = cls.company_data_2["company"] + cls.env.user.company_ids |= cls.company_2 + + cls.contract_mc = ( + cls.env["contract.contract"] + .with_company(cls.company_2) + .create( + { + "name": "Test Contract MC", + "partner_id": cls.partner.id, + "pricelist_id": cls.partner.property_product_pricelist.id, + "line_recurrence": True, + "contract_type": "purchase", + "contract_line_ids": [ + ( + 0, + 0, + { + "product_id": cls.product_1.id, + "name": "Services from #START# to #END#", + "quantity": 1, + "uom_id": cls.product_1.uom_id.id, + "price_unit": 100, + "discount": 50, + "recurring_rule_type": "monthly", + "recurring_interval": 1, + "date_start": "2018-02-15", + "recurring_next_date": "2018-02-22", + }, + ) + ], + } + ) + ) + cls.line_vals = { + "contract_id": cls.contract_mc.id, + "product_id": cls.product_1.id, + "name": "Services from #START# to #END#", + "quantity": 1, + "uom_id": cls.product_1.uom_id.id, + "price_unit": 100, + "discount": 50, + "recurring_rule_type": "monthly", + "recurring_interval": 1, + "date_start": "2018-01-01", + "recurring_next_date": "2018-01-15", + "is_auto_renew": False, + } + cls.acct_line_mc = ( + cls.env["contract.line"].with_company(cls.company_2).create(cls.line_vals) + ) + + def test_cron_recurring_create_invoice_multi_company(self): + self.acct_line.date_start = "2018-01-01" + self.acct_line.recurring_invoicing_type = "post-paid" + self.acct_line.date_end = "2018-03-15" + + self.acct_line_mc.date_start = "2018-01-01" + self.acct_line_mc.recurring_invoicing_type = "post-paid" + self.acct_line_mc.date_end = "2018-03-15" + + contracts = self.contract2 + contracts_company_2 = self.env["contract.contract"].browse() + for _i in range(10): + contracts |= self.contract.copy() + for _i in range(10): + vals = ( + self.contract_mc.with_company(company=self.company_2) + .with_context(active_test=False) + .copy_data({"company_id": self.company_2.id}) + ) + contracts_company_2 |= self.contract_mc.with_company( + company=self.company_2 + ).create(vals) + self.env["contract.contract"].cron_recurring_create_invoice() + # Check company 1 + invoice_lines_company_1 = self.env["account.move.line"].search( + [("contract_line_id", "in", contracts.mapped("contract_line_ids").ids)] + ) + invoice_lines_company_2 = self.env["account.move.line"].search( + [ + ( + "contract_line_id", + "in", + contracts_company_2.mapped("contract_line_ids").ids, + ) + ] + ) + self.assertEqual( + len(contracts.mapped("contract_line_ids")), len(invoice_lines_company_1) + ) + self.assertEqual( + len(contracts_company_2.mapped("contract_line_ids")), + len(invoice_lines_company_2), + ) + + +class TestContractMultiCompany(ContractMulticompanyCase, AccountTestInvoicingCommon): + """ TESTS """