added test cases to spp_programs_sp and spp_registry_group_hierarchy

Author: reichie020212

spp_programs_sp/tests/__init__.py

@@ -0,0 +1,4 @@
+from . import common
+from . import test_entitlement_manager_cash
+from . import test_entitlement_manager_default
+from . import test_entitlement_manager_inkind

spp_programs_sp/tests/common.py

@@ -0,0 +1,43 @@
+from odoo.tests.common import TransactionCase
+
+
+class Common(TransactionCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        country = cls.env.ref("base.iq")
+        cls.service_point_1 = cls.env["spp.service.point"].create(
+            {
+                "name": "Agent1",
+                "country_id": country.id,
+                "phone_no": "+9647001234567",
+                "is_disabled": False,
+            }
+        )
+        cls.registrant_1 = cls.env["res.partner"].create(
+            {
+                "name": "Registrant 1 [TEST]",
+                "is_registrant": True,
+                "is_group": True,
+                "service_point_ids": [(6, 0, cls.service_point_1.ids)],
+            }
+        )
+
+        cls.program = cls.env["g2p.program"].create(
+            {
+                "name": "Program 1",
+                "store_sp_in_entitlements": True,
+            }
+        )
+        cls.program.create_journal()
+
+    def _common_test_get_addl_entitlement_fields(self, entitlement_manager, mock_super):
+        mock_return_value = {"test": "test"}
+        mock_super.return_value = mock_return_value
+        result = entitlement_manager._get_addl_entitlement_fields(self.registrant_1)
+        self.assertEqual(result, {"test": "test", "service_point_ids": self.service_point_1})
+
+        mock_super.return_value = None
+        self.program.store_sp_in_entitlements = False
+        result = entitlement_manager._get_addl_entitlement_fields(self.registrant_1)
+        self.assertEqual(result, {"service_point_ids": None})

spp_programs_sp/tests/test_entitlement_manager_cash.py

@@ -0,0 +1,21 @@
+from unittest.mock import patch
+
+from .common import Common
+
+
+class CashEntitlementManagerTest(Common):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.cash_entitlement_manager = cls.env["g2p.program.entitlement.manager.cash"].create(
+            {
+                "name": "Entitlement Manager Cash 1",
+                "program_id": cls.program.id,
+            }
+        )
+
+    @patch(
+        "odoo.addons.spp_entitlement_cash.models.entitlement_manager.G2PCashEntitlementManager._get_addl_entitlement_fields"
+    )
+    def test_cash_get_addl_entitlement_fields(self, mock_super):
+        self._common_test_get_addl_entitlement_fields(self.cash_entitlement_manager, mock_super)

spp_programs_sp/tests/test_entitlement_manager_default.py

@@ -0,0 +1,21 @@
+from unittest.mock import patch
+
+from .common import Common
+
+
+class DefaultEntitlementManagerTest(Common):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.default_entitlement_manager = cls.env["g2p.program.entitlement.manager.default"].create(
+            {
+                "name": "Entitlement Manager Default 1",
+                "program_id": cls.program.id,
+            }
+        )
+
+    @patch(
+        "odoo.addons.spp_programs.models.managers.entitlement_manager_default.G2PDefaultEntitlementManagerCustom._get_addl_entitlement_fields"
+    )
+    def test_cash_get_addl_entitlement_fields(self, mock_super):
+        self._common_test_get_addl_entitlement_fields(self.default_entitlement_manager, mock_super)

spp_programs_sp/tests/test_entitlement_manager_inkind.py

@@ -0,0 +1,21 @@
+from unittest.mock import patch
+
+from .common import Common
+
+
+class DefaultEntitlementManagerTest(Common):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.inkind_entitlement_manager = cls.env["g2p.program.entitlement.manager.inkind"].create(
+            {
+                "name": "Entitlement Manager In-Kind 1",
+                "program_id": cls.program.id,
+            }
+        )
+
+    @patch(
+        "odoo.addons.spp_entitlement_in_kind.models.entitlement_manager.G2PInKindEntitlementManager._get_addl_entitlement_fields"
+    )
+    def test_cash_get_addl_entitlement_fields(self, mock_super):
+        self._common_test_get_addl_entitlement_fields(self.inkind_entitlement_manager, mock_super)

spp_registry_group_hierarchy/tests/__init__.py

@@ -0,0 +1 @@
+from . import test_group_membership

spp_registry_group_hierarchy/tests/test_group_membership.py

@@ -0,0 +1,87 @@
+from odoo import fields
+from odoo.exceptions import UserError
+from odoo.tests.common import TransactionCase
+
+
+class MembershipTest(TransactionCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+
+        cls.registrant_1 = cls.env["res.partner"].create(
+            {
+                "family_name": "Butay",
+                "given_name": "Red",
+                "name": "Red Butay",
+                "is_group": False,
+                "is_registrant": True,
+            }
+        )
+        cls.kind_1 = cls.env["g2p.group.kind"].create(
+            {
+                "name": "Test Kind",
+                "allow_all_member_type": True,
+            }
+        )
+        cls.group_1 = cls.env["res.partner"].create(
+            {
+                "name": "Group 1",
+                "is_group": True,
+                "is_registrant": True,
+                "kind": cls.kind_1.id,
+            }
+        )
+        cls.group_membership_1 = cls.env["g2p.group.membership"].create(
+            {
+                "group": cls.group_1.id,
+                "individual": cls.registrant_1.id,
+                "start_date": fields.Datetime.now(),
+            }
+        )
+
+    def test_compute_individual_domain(self):
+        self.group_membership_1._compute_individual_domain()
+
+        self.assertIsInstance(self.group_membership_1.individual_domain, list)
+        self.assertEqual(len(self.group_membership_1.individual_domain), 2)
+        self.assertIn(("is_registrant", "=", True), self.group_membership_1.individual_domain)
+        self.assertIn(("id", "!=", self.group_membership_1.group.id), self.group_membership_1.individual_domain)
+
+        self.group_1.kind = False
+        self.group_membership_1._compute_individual_domain()
+
+        self.assertIsInstance(self.group_membership_1.individual_domain, list)
+        self.assertEqual(len(self.group_membership_1.individual_domain), 2)
+        self.assertIn(("is_registrant", "=", True), self.group_membership_1.individual_domain)
+        self.assertIn(("is_group", "=", False), self.group_membership_1.individual_domain)
+
+    def test_open_member_form(self):
+        member_form = self.group_membership_1.open_member_form()
+
+        self.assertIsInstance(member_form, dict)
+        self.assertEqual(member_form["name"], "Individual Member")
+        self.assertEqual(member_form["view_mode"], "form")
+        self.assertEqual(member_form["res_model"], "res.partner")
+        self.assertEqual(member_form["res_id"], self.registrant_1.id)
+        self.assertEqual(member_form["view_id"], self.env.ref("g2p_registry_individual.view_individuals_form").id)
+        self.assertEqual(member_form["type"], "ir.actions.act_window")
+        self.assertEqual(member_form["target"], "new")
+        self.assertEqual(member_form["context"], {"default_is_group": False})
+        self.assertEqual(member_form["flags"], {"mode": "readonly"})
+
+        self.registrant_1.is_group = True
+        member_form = self.group_membership_1.open_member_form()
+        self.assertIsInstance(member_form, dict)
+        self.assertEqual(member_form["name"], "Group Membership")
+        self.assertEqual(member_form["view_mode"], "form")
+        self.assertEqual(member_form["res_model"], "res.partner")
+        self.assertEqual(member_form["res_id"], self.registrant_1.id)
+        self.assertEqual(member_form["view_id"], self.env.ref("g2p_registry_group.view_groups_form").id)
+        self.assertEqual(member_form["type"], "ir.actions.act_window")
+        self.assertEqual(member_form["target"], "new")
+        self.assertEqual(member_form["context"], {"default_is_group": True})
+        self.assertEqual(member_form["flags"], {"mode": "readonly"})
+
+        self.group_membership_1.individual = False
+        with self.assertRaisesRegex(UserError, "A group or individual must be speficied for this member."):
+            self.group_membership_1.open_member_form()