mirror of
https://github.com/OCA/server-backend.git
synced 2025-02-18 09:52:42 +02:00
84 lines
3.6 KiB
Python
84 lines
3.6 KiB
Python
# Copyright 2021 Open Source Integrators
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
|
|
import mock
|
|
|
|
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, "company_id": None}),
|
|
(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.sudo(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)
|
|
|
|
def test_140_session_info(self):
|
|
"session_info sets active roles"
|
|
with mock.patch.object(
|
|
self.env['res.users'].__class__, '_set_session_active_roles'
|
|
) as mock_set_session_active_roles, mock.patch(
|
|
'odoo.addons.base_user_role_company.models.ir_http.request',
|
|
) as base_user_role_company_request, mock.patch(
|
|
'odoo.addons.base_setup.models.ir_http.request',
|
|
) as base_setup_request, mock.patch(
|
|
'odoo.addons.web_tour.models.ir_http.request',
|
|
) as web_tour_request, mock.patch(
|
|
'odoo.addons.web.models.ir_http.request',
|
|
) as web_request:
|
|
base_setup_request.env = self.env
|
|
web_request.env = self.env
|
|
web_tour_request.env = self.env
|
|
self.env['ir.http'].sudo(self.test_user).session_info()
|
|
mock_set_session_active_roles.assert_called_once()
|
|
base_user_role_company_request.httprequest.cookies.get.assert_called()
|