add test cases

Author: reichie020212

spp_oauth/tests/__init__.py

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

spp_oauth/tests/test_rsa_encode_decode.py

@@ -0,0 +1,30 @@
+from unittest.mock import patch
+
+from odoo.tests.common import TransactionCase
+
+from ..tools.rsa_encode_decode import calculate_signature, verify_and_decode_signature
+
+
+class TestRSA(TransactionCase):
+    @patch("odoo.addons.spp_oauth.tools.rsa_encode_decode.jwt.encode")
+    def test_calculate_signature(self, mock_encode):
+        mock_encode.return_value = "mocked_signature"
+
+        header = {"typ": "JWT"}
+        payload = {"data": "test"}
+        signature = calculate_signature(header, payload)
+
+        self.assertEqual(signature, "mocked_signature")
+        mock_encode.assert_called_once_with(headers=header, payload=payload, key="", algorithm="RS256")
+
+    @patch("odoo.addons.spp_oauth.tools.rsa_encode_decode.jwt.decode")
+    def test_verify_and_decode_signature(self, mock_decode):
+        mock_decode.return_value = {"data": "test"}
+
+        access_token = "mocked_access_token"
+
+        success, decoded = verify_and_decode_signature(access_token)
+
+        self.assertTrue(success)
+        self.assertEqual(decoded, {"data": "test"})
+        mock_decode.assert_called_once_with(access_token, key="", algorithms=["RS256"])

spp_pmt/tests/__init__.py

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

spp_pmt/tests/test_pmt.py

@@ -0,0 +1,71 @@
+from unittest.mock import MagicMock, patch
+
+from odoo import fields
+from odoo.tests.common import TransactionCase
+
+
+class PMTTest(TransactionCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.area_1 = cls.env["spp.area"].create(
+            {
+                "draft_name": "Area 1",
+            }
+        )
+        cls.registrant_1 = cls.env["res.partner"].create(
+            {
+                "family_name": "Butay",
+                "given_name": "Red",
+                "name": "Red Butay",
+                "is_group": False,
+                "is_registrant": True,
+                "area_id": cls.area_1.id,
+            }
+        )
+        cls.group_1 = cls.env["res.partner"].create(
+            {
+                "name": "Group 1",
+                "is_group": True,
+                "is_registrant": True,
+                "area_id": cls.area_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_area(self):
+        self.registrant_1._compute_area()
+        self.assertFalse(self.registrant_1.area_calc.id)
+
+        self.group_1._compute_area()
+        self.assertEqual(self.group_1.area_calc.id, self.area_1.id)
+
+    def test_compute_score(self):
+        self.group_1.compute_score("z_ind_grp_pmt_score")
+        self.assertEqual(self.group_1.z_ind_grp_pmt_score, 0)
+
+    @patch("odoo.addons.base.models.ir_model.IrModelFields.search")
+    def test_compute_score_with_weight(self, mock_search):
+        area_id = self.env["spp.fields.area"].create(
+            {
+                "name": self.area_1.id,
+                "weight": 1.0,
+            }
+        )
+        self.registrant_1.z_ind_grp_num_individuals = 1
+
+        mock_fields = MagicMock()
+        mock_fields.name = "z_ind_grp_num_individuals"
+        mock_fields.with_weight = True
+        mock_fields.target_type = "indv"
+        mock_fields.area_ids = area_id
+        mock_search.return_value = [mock_fields]
+
+        self.group_1.compute_score("z_ind_grp_pmt_score")
+        self.assertEqual(self.group_1.z_ind_grp_pmt_score, 1)

spp_user_roles/tests/__init__.py

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

spp_user_roles/tests/test_res_partner.py

@@ -0,0 +1,31 @@
+from unittest.mock import MagicMock, patch
+
+from odoo.tests.common import TransactionCase
+
+
+class ResPartnerTest(TransactionCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.center_area_1 = cls.env["spp.area"].create({"draft_name": "Center Area 1"})
+        cls.center_area_2 = cls.env["spp.area"].create({"draft_name": "Center Area 1"})
+        cls.partner_1 = cls.env["res.partner"].create({"name": "Partner 1", "area_id": cls.center_area_1.id})
+        cls.partner_2 = cls.env["res.partner"].create({"name": "Partner 2", "area_id": cls.center_area_2.id})
+
+    def test_search_read(self):
+        result = self.env["res.partner"].search_read(domain=[], fields=["id", "name", "area_id"])
+        self.assertEqual(len(result), self.env["res.partner"].search_count([]))
+
+        center_area_ids_mock = MagicMock()
+        center_area_ids_mock.ids = self.center_area_1.ids
+
+        with patch(
+            "odoo.addons.spp_user_roles.models.user.ResUsersCustomSPP.center_area_ids", new=center_area_ids_mock
+        ):
+            result = self.env["res.partner"].search_read(domain=[], fields=["id", "name", "area_id"])
+
+        self.assertEqual(len(result), 1)
+        self.assertEqual(
+            result,
+            [{"id": self.partner_1.id, "name": "Partner 1", "area_id": (self.center_area_1.id, "Center Area 1")}],
+        )