[ADD] base_user_role_company: add per company user roles

base_user_role_company/tests/test_role_per_company.py

@@ -0,0 +1,62 @@
+# Copyright 2021 Open Source Integrators
+# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
+
+from odoo.tests.common import TransactionCase
+
+
+class TestUserRoleCompany(TransactionCase):
+    def setUp(self):
+        super().setUp()
+        # COMPANIES
+        self.Company = self.env["res.company"]
+        self.company1 = self.env.ref("base.main_company")
+        self.company2 = self.Company.create({"name": "company2"})
+        # ROLES
+        self.Role = self.env["res.users.role"]
+        self.roleA = self.Role.create({"name": "ROLE All Companies"})
+        self.roleB = self.Role.create({"name": "ROLE Company 1"})
+        self.roleC = self.Role.create({"name": "ROLE Company 1 and 2"})
+        # USER
+        # ==Role=== ==Company== C1  C2  C1+C2
+        # Role A                Yes Yes Yes
+        # Role B    Company1    Yes
+        # Role C    Company1    Yes     Yes
+        # Role C    Company2        Yes Yes
+        self.User = self.env["res.users"]
+        user_vals = {
+            "name": "ROLES TEST USER",
+            "login": "test_user",
+            "company_ids": [(6, 0, [self.company1.id, self.company2.id])],
+            "role_line_ids": [
+                (0, 0, {"role_id": self.roleA.id}),
+                (0, 0, {"role_id": self.roleB.id, "company_id": self.company1.id}),
+                (0, 0, {"role_id": self.roleC.id, "company_id": self.company1.id}),
+                (0, 0, {"role_id": self.roleC.id, "company_id": self.company2.id}),
+            ],
+        }
+        self.test_user = self.User.create(user_vals)
+        self.User = self.User.with_user(self.test_user)
+
+    def test_110_company_1(self):
+        "Company 1 selected: Tech and Settings roles are activated"
+        self.User._set_session_active_roles([self.company1.id])
+        active_roles = self.test_user.role_line_ids.filtered("active_role").mapped(
+            "role_id"
+        )
+        self.assertEqual(active_roles, self.roleA | self.roleB | self.roleC)
+
+    def test_120_company_2(self):
+        "Company 2 selected: only Tech role enabled"
+        self.User._set_session_active_roles([self.company2.id])
+        active_roles = self.test_user.role_line_ids.filtered("active_role").mapped(
+            "role_id"
+        )
+        self.assertEqual(active_roles, self.roleA | self.roleC)
+
+    def test_130_company_1_2(self):
+        "Settings Role enabled for Company 1 and 2"
+        self.User._set_session_active_roles([self.company1.id, self.company2.id])
+        active_roles = self.test_user.role_line_ids.filtered("active_role").mapped(
+            "role_id"
+        )
+        self.assertEqual(active_roles, self.roleA | self.roleC)