[14.0][ADD] contract_split module

contract_split/tests/test_contract_split.py

@@ -0,0 +1,123 @@
+# Copyright 2023 Foodles (https://www.foodles.com/)
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+from odoo.exceptions import ValidationError
+from odoo.tests import tagged
+
+from odoo.addons.contract.tests.test_contract import TestContractBase
+
+
+@tagged("post_install", "-at_install")
+class TestContractSplit(TestContractBase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+
+    def test_default_get(self):
+        wizard = (
+            self.env["split.contract"]
+            .with_context(active_id=self.contract3.id)
+            .create({})
+        )
+        self.assertEqual(self.contract3.partner_id.id, wizard.partner_id.id)
+        self.assertEqual(
+            self.contract3.invoice_partner_id.id, wizard.invoice_partner_id.id
+        )
+        self.assertEqual(self.contract3.id, wizard.main_contract_id.id)
+        self.assertEqual(3, len(wizard.split_line_ids.ids))
+
+    def test_no_split_because_no_qty_set(self):
+        wizard = (
+            self.env["split.contract"]
+            .with_context(active_id=self.contract3.id)
+            .create({})
+        )
+        wizard.partner_id = self.partner_2.id
+        initial_contracts_length = self.env["contract.contract"].search_count([])
+        # confirm wizard without setting to_split quantities
+        wizard.action_split_contract()
+        # nothing should have changed. No new contract created and original
+        # contract remains untouched
+        self.assertEqual(3, len(self.contract3.contract_line_ids.ids))
+        self.assertEqual(
+            initial_contracts_length, self.env["contract.contract"].search_count([])
+        )
+
+    def test_split_one_line_full_qty(self):
+        wizard = (
+            self.env["split.contract"]
+            .with_context(active_id=self.contract3.id)
+            .create({})
+        )
+        wizard.partner_id = self.partner_2.id
+        initial_contracts_length = self.env["contract.contract"].search_count([])
+        # set quantity to split in the wizard
+        wizard.split_line_ids[0].quantity_to_split = wizard.split_line_ids[
+            0
+        ].original_qty
+        # confirm wizard with setting to_split quantities
+        new_contract = wizard.action_split_contract()
+        # A new contract must have been created.
+        self.assertEqual(
+            initial_contracts_length + 1, self.env["contract.contract"].search_count([])
+        )
+        # new contract has now the splitted line
+        self.assertEqual(self.partner2.id, new_contract.partner_id.id)
+        self.assertEqual(1, len(new_contract.contract_line_ids.ids))
+        self.assertEqual(
+            self.contract3.id,
+            new_contract.contract_line_ids.mapped("splitted_from_contract_id"),
+        )
+        # Original contract has now only 2 lines (3 at the beginning)
+        self.assertEqual(2, len(self.contract3.contract_line_ids.ids))
+
+    def test_split_one_line_one_qty(self):
+        # Set a qty = 2 in one line of a contract
+        self.contract3.contract_line_ids.filtered(
+            lambda line: line.name == "Line"
+        ).quantity = 2
+        wizard = (
+            self.env["split.contract"]
+            .with_context(active_id=self.contract3.id)
+            .create({})
+        )
+        wizard.partner_id = self.partner_2.id
+        initial_contracts_length = self.env["contract.contract"].search_count([])
+        # set quantity to split in the wizard
+        wizard.split_line_ids.filtered(lambda l: l.name == "Line").quantity_to_split = 1
+        # confirm wizard with setting to_split quantities
+        new_contract = wizard.action_split_contract()
+        # A new contract must have been created.
+        self.assertEqual(
+            initial_contracts_length + 1, self.env["contract.contract"].search_count([])
+        )
+        # new contract has partner2 as partner_id
+        self.assertEqual(self.partner2.id, new_contract.partner_id.id)
+        # new contract has now the splitted line with a qty of one
+        self.assertEqual(1, len(new_contract.contract_line_ids.ids))
+        self.assertEqual(1, new_contract.contract_line_ids.quantity)
+        self.assertEqual(
+            self.contract3.id,
+            new_contract.contract_line_ids.mapped("splitted_from_contract_id"),
+        )
+        # Original contract still has 3 lines but with a qty=1 in the last line named "Line"
+        self.assertEqual(3, len(self.contract3.contract_line_ids.ids))
+        self.assertEqual(
+            1,
+            self.contract3.contract_line_ids.filtered(
+                lambda l: l.name == "Line"
+            ).quantity,
+        )
+
+    def test_split_with_more_quantity_should_raise_error(self):
+        wizard = (
+            self.env["split.contract"]
+            .with_context(active_id=self.contract3.id)
+            .create({})
+        )
+        # set quantity to split in the wizard
+        wizard.split_line_ids[0].quantity_to_split = (
+            wizard.split_line_ids[0].original_qty + 2
+        )
+        # confirm wizard with setting to_split quantities that should raise an error
+        with self.assertRaises(ValidationError):
+            wizard.action_split_contract()