|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). from odoo.tests.common import TransactionCase
class TestUserProfile(TransactionCase): def _helper_unpack_groups_role(self, role): role_group_ids = role.trans_implied_ids.ids role_group_ids.append(role.group_id.id) return sorted(set(role_group_ids))
def _helper_unpack_groups_group(self, group): group_ids = group.trans_implied_ids.ids group_ids.append(group.id) return sorted(set(group_ids))
def setUp(self): super().setUp() self.user_model = self.env["res.users"] self.role_model = self.env["res.users.role"]
self.default_user = self.env.ref("base.default_user") user_vals = { "name": "USER TEST (ROLES)", "login": "user_test_roles", } self.user_id = self.user_model.create(user_vals)
self.profile1_id = self.env["res.users.profile"].create({"name": "profile1"}) self.profile2_id = self.env["res.users.profile"].create({"name": "profile2"})
# role 1 self.group_user_id = self.env.ref("base.group_user") self.group_no_one_id = self.env.ref("base.group_no_one")
# role 2 self.group_system_id = self.env.ref("base.group_system") self.group_multi_company_id = self.env.ref("base.group_multi_company")
# role 3 self.group_erp_manager_id = self.env.ref("base.group_erp_manager") self.group_partner_manager_id = self.env.ref("base.group_partner_manager")
# roles 1 and 2 have a profile, role 3 no profile vals = { "name": "ROLE_1", "implied_ids": [(6, 0, [self.group_user_id.id, self.group_no_one_id.id])], "profile_id": self.profile1_id.id, } self.role1_id = self.role_model.create(vals) self.role1_group_ids = self._helper_unpack_groups_role(self.role1_id)
vals = { "name": "ROLE_2", "implied_ids": [ ( 6, 0, [self.group_system_id.id, self.group_multi_company_id.id], ) ], "profile_id": self.profile2_id.id, } self.role2_id = self.role_model.create(vals) self.role2_group_ids = self._helper_unpack_groups_role(self.role2_id)
vals = { "name": "ROLE_3", "implied_ids": [ ( 6, 0, [ self.group_erp_manager_id.id, self.group_partner_manager_id.id, ], ) ], } self.role3_id = self.role_model.create(vals) self.role3_group_ids = self._helper_unpack_groups_role(self.role3_id)
def test_filter_by_profile(self): line1_vals = {"role_id": self.role1_id.id, "user_id": self.user_id.id} self.user_id.write({"role_line_ids": [(0, 0, line1_vals)]}) line2_vals = {"role_id": self.role2_id.id, "user_id": self.user_id.id} self.user_id.write({"role_line_ids": [(0, 0, line2_vals)]}) self.assertEqual(self.user_id.profile_ids, self.profile1_id + self.profile2_id) self.assertEqual(self.user_id.profile_id, self.profile1_id) self.user_id.action_profile_change({"profile_id": self.profile1_id.id})
user_group_ids = sorted({group.id for group in self.user_id.groups_id}) expected_group_ids = sorted(set(self.role1_group_ids)) self.assertEqual(user_group_ids, expected_group_ids)
self.user_id.action_profile_change({"profile_id": self.profile2_id.id})
user_group_ids = sorted({group.id for group in self.user_id.groups_id}) expected_group_ids = sorted(set(self.role2_group_ids)) self.assertEqual(user_group_ids, expected_group_ids)
def test_allow_by_noprofile(self): line1_vals = {"role_id": self.role1_id.id, "user_id": self.user_id.id} self.user_id.write({"role_line_ids": [(0, 0, line1_vals)]}) line2_vals = {"role_id": self.role3_id.id, "user_id": self.user_id.id} self.user_id.write({"role_line_ids": [(0, 0, line2_vals)]}) self.assertEqual(self.user_id.profile_ids, self.profile1_id) user_group_ids = [] for group in self.user_id.groups_id: user_group_ids += self._helper_unpack_groups_group(group) user_group_ids = set(user_group_ids) expected_groups = set(self.role1_group_ids + self.role3_group_ids) self.assertEqual(user_group_ids, expected_groups)
|