audrius/server-backend
1
0
Fork 0
mirror of https://github.com/OCA/server-backend.git synced 2025-02-18 09:52:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

[FIX] base_user_role_company: fix tests

Browse Source
This commit is contained in:
Daniel Reis
2022-03-03 11:40:58 +00:00
parent f3c4d3d90e
commit 65218809a6
2 changed files with 23 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
base_user_role_company/models/user.py
View File

@@ -22,10 +22,9 @@ class ResUsers(models.Model):
def _get_enabled_roles(self): def _get_enabled_roles(self):
res = super()._get_enabled_roles() res = super()._get_enabled_roles()
# Enable only the Roles corresponing to the currently selected company # Enable only the Roles corresponing to the currently selected company
if self.env.user.role_line_ids: if self.role_line_ids:
curr_company = self.env.company
res = res.filtered( res = res.filtered(
lambda x: not x.company_id or x.company_id == curr_company lambda x: not x.company_id or x.company_id == self.env.company
) )
return res return res

41
base_user_role_company/tests/test_role_per_company.py
View File

@@ -11,11 +11,18 @@ class TestUserRoleCompany(TransactionCase):
self.Company = self.env["res.company"] self.Company = self.env["res.company"]
self.company1 = self.env.ref("base.main_company") self.company1 = self.env.ref("base.main_company")
self.company2 = self.Company.create({"name": "company2"}) self.company2 = self.Company.create({"name": "company2"})
# GROUPS for roles
self.groupA = self.env.ref("base.group_user")
self.groupB = self.env.ref("base.group_system")
self.groupC = self.env.ref("base.group_partner_manager")
# ROLES # ROLES
self.Role = self.env["res.users.role"] self.Role = self.env["res.users.role"]
self.roleA = self.Role.create({"name": "ROLE All Companies"}) self.roleA = self.Role.create({"name": "ROLE All Companies"})
self.roleA.implied_ids |= self.groupA
self.roleB = self.Role.create({"name": "ROLE Company 1"}) self.roleB = self.Role.create({"name": "ROLE Company 1"})
self.roleB.implied_ids |= self.groupB
self.roleC = self.Role.create({"name": "ROLE Company 1 and 2"}) self.roleC = self.Role.create({"name": "ROLE Company 1 and 2"})
self.roleC.implied_ids |= self.groupC
# USER # USER
# ==Role=== ==Company== C1 C2 C1+C2 # ==Role=== ==Company== C1 C2 C1+C2
# Role A Yes Yes Yes # Role A Yes Yes Yes
@@ -35,28 +42,22 @@ class TestUserRoleCompany(TransactionCase):
], ],
} }
self.test_user = self.User.create(user_vals) self.test_user = self.User.create(user_vals)
self.User = self.User.with_user(self.test_user)
def test_110_company_1(self): def test_110_company_1(self):
"Company 1 selected: Tech and Settings roles are activated" "Company 1 selected: Roles A, B and C are enabled"
self.User._set_session_active_roles([self.company1.id]) self.test_user.set_groups_from_roles(company_id=self.company1.id)
active_roles = self.test_user.role_line_ids.filtered("active_role").mapped( expected = self.groupA | self.groupB | self.groupC
"role_id" found = self.test_user.groups_id.filtered(lambda x: x in expected)
) self.assertEqual(expected, found)
self.assertEqual(active_roles, self.roleA | self.roleB | self.roleC)
def test_120_company_2(self): def test_120_company_2(self):
"Company 2 selected: only Tech role enabled" "Company 2 selected: Roles A and C are enabled"
self.User._set_session_active_roles([self.company2.id]) self.test_user.set_groups_from_roles(company_id=self.company2.id)
active_roles = self.test_user.role_line_ids.filtered("active_role").mapped( enabled = self.test_user.groups_id
"role_id" expected = self.groupA | self.groupC
) found = enabled.filtered(lambda x: x in expected)
self.assertEqual(active_roles, self.roleA | self.roleC) self.assertEqual(expected, found)
def test_130_company_1_2(self): not_expected = self.groupB
"Settings Role enabled for Company 1 and 2" found = enabled.filtered(lambda x: x in not_expected)
self.User._set_session_active_roles([self.company1.id, self.company2.id]) self.assertFalse(found)
active_roles = self.test_user.role_line_ids.filtered("active_role").mapped(
"role_id"
)
self.assertEqual(active_roles, self.roleA | self.roleC)