mirror of
https://github.com/OCA/server-backend.git
synced 2025-02-18 09:52:42 +02:00
[IMP] base_user_role: black, isort
This commit is contained in:
sebalix
committed by
Jesús Alan Ramos Rodríguez
Jesús Alan Ramos Rodríguez
parent
6337a5968e
commit
0e1b4653f7
@@ -7,61 +7,62 @@ from odoo.tests.common import TransactionCase
|
||||
|
||||
|
||||
class TestUserRole(TransactionCase):
|
||||
|
||||
def setUp(self):
|
||||
super(TestUserRole, self).setUp()
|
||||
self.user_model = self.env['res.users']
|
||||
self.role_model = self.env['res.users.role']
|
||||
self.user_model = self.env["res.users"]
|
||||
self.role_model = self.env["res.users.role"]
|
||||
|
||||
self.default_user = self.env.ref('base.default_user')
|
||||
self.default_user = self.env.ref("base.default_user")
|
||||
self.user_id = self.user_model.create(
|
||||
{'name': "USER TEST (ROLES)", 'login': 'user_test_roles'})
|
||||
{"name": "USER TEST (ROLES)", "login": "user_test_roles"}
|
||||
)
|
||||
|
||||
# ROLE_1
|
||||
self.group_user_id = self.env.ref('base.group_user')
|
||||
self.group_no_one_id = self.env.ref('base.group_no_one')
|
||||
self.group_user_id = self.env.ref("base.group_user")
|
||||
self.group_no_one_id = self.env.ref("base.group_no_one")
|
||||
vals = {
|
||||
'name': "ROLE_1",
|
||||
'implied_ids': [
|
||||
(6, 0, [self.group_user_id.id, self.group_no_one_id.id])],
|
||||
"name": "ROLE_1",
|
||||
"implied_ids": [(6, 0, [self.group_user_id.id, self.group_no_one_id.id])],
|
||||
}
|
||||
self.role1_id = self.role_model.create(vals)
|
||||
|
||||
# ROLE_2
|
||||
# Must have group_user in order to have sufficient groups. Check:
|
||||
# github.com/odoo/odoo/commit/c3717f3018ce0571aa41f70da4262cc946d883b4
|
||||
self.group_multi_currency_id = self.env.ref(
|
||||
'base.group_multi_currency')
|
||||
self.group_settings_id = self.env.ref('base.group_system')
|
||||
self.group_multi_currency_id = self.env.ref("base.group_multi_currency")
|
||||
self.group_settings_id = self.env.ref("base.group_system")
|
||||
vals = {
|
||||
'name': "ROLE_2",
|
||||
'implied_ids': [
|
||||
(6, 0, [self.group_user_id.id,
|
||||
"name": "ROLE_2",
|
||||
"implied_ids": [
|
||||
(
|
||||
6,
|
||||
0,
|
||||
[
|
||||
self.group_user_id.id,
|
||||
self.group_multi_currency_id.id,
|
||||
self.group_settings_id.id])],
|
||||
self.group_settings_id.id,
|
||||
],
|
||||
)
|
||||
],
|
||||
}
|
||||
self.role2_id = self.role_model.create(vals)
|
||||
self.company1 = self.env.ref('base.main_company')
|
||||
self.company2 = self.env['res.company'].create({'name': 'company2'})
|
||||
self.company1 = self.env.ref("base.main_company")
|
||||
self.company2 = self.env["res.company"].create({"name": "company2"})
|
||||
self.user_id.write(
|
||||
{'company_ids': [
|
||||
(4, self.company1.id, 0), (4, self.company2.id, 0)]})
|
||||
{"company_ids": [(4, self.company1.id, 0), (4, self.company2.id, 0)]}
|
||||
)
|
||||
|
||||
def test_role_1(self):
|
||||
self.user_id.write(
|
||||
{'role_line_ids': [(0, 0, {'role_id': self.role1_id.id})]})
|
||||
user_group_ids = sorted(set(
|
||||
[group.id for group in self.user_id.groups_id]))
|
||||
self.user_id.write({"role_line_ids": [(0, 0, {"role_id": self.role1_id.id})]})
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
role_group_ids = self.role1_id.trans_implied_ids.ids
|
||||
role_group_ids.append(self.role1_id.group_id.id)
|
||||
role_group_ids = sorted(set(role_group_ids))
|
||||
self.assertEqual(user_group_ids, role_group_ids)
|
||||
|
||||
def test_role_2(self):
|
||||
self.user_id.write(
|
||||
{'role_line_ids': [(0, 0, {'role_id': self.role2_id.id})]})
|
||||
user_group_ids = sorted(set(
|
||||
[group.id for group in self.user_id.groups_id]))
|
||||
self.user_id.write({"role_line_ids": [(0, 0, {"role_id": self.role2_id.id})]})
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
role_group_ids = self.role2_id.trans_implied_ids.ids
|
||||
role_group_ids.append(self.role2_id.group_id.id)
|
||||
role_group_ids = sorted(set(role_group_ids))
|
||||
@@ -69,12 +70,14 @@ class TestUserRole(TransactionCase):
|
||||
|
||||
def test_role_1_2(self):
|
||||
self.user_id.write(
|
||||
{'role_line_ids': [
|
||||
(0, 0, {'role_id': self.role1_id.id}),
|
||||
(0, 0, {'role_id': self.role2_id.id}),
|
||||
]})
|
||||
user_group_ids = sorted(set(
|
||||
[group.id for group in self.user_id.groups_id]))
|
||||
{
|
||||
"role_line_ids": [
|
||||
(0, 0, {"role_id": self.role1_id.id}),
|
||||
(0, 0, {"role_id": self.role2_id.id}),
|
||||
]
|
||||
}
|
||||
)
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
role1_group_ids = self.role1_id.trans_implied_ids.ids
|
||||
role1_group_ids.append(self.role1_id.group_id.id)
|
||||
role2_group_ids = self.role2_id.trans_implied_ids.ids
|
||||
@@ -88,15 +91,16 @@ class TestUserRole(TransactionCase):
|
||||
yesterday = today - datetime.timedelta(days=1)
|
||||
yesterday_str = fields.Date.to_string(yesterday)
|
||||
self.user_id.write(
|
||||
{'role_line_ids': [
|
||||
# Role 1 should be enabled
|
||||
(0, 0, {'role_id': self.role1_id.id, 'date_from': today_str}),
|
||||
# Role 2 should be disabled
|
||||
(0, 0,
|
||||
{'role_id': self.role2_id.id, 'date_to': yesterday_str}),
|
||||
]})
|
||||
user_group_ids = sorted(set(
|
||||
[group.id for group in self.user_id.groups_id]))
|
||||
{
|
||||
"role_line_ids": [
|
||||
# Role 1 should be enabled
|
||||
(0, 0, {"role_id": self.role1_id.id, "date_from": today_str}),
|
||||
# Role 2 should be disabled
|
||||
(0, 0, {"role_id": self.role2_id.id, "date_to": yesterday_str}),
|
||||
]
|
||||
}
|
||||
)
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
role1_group_ids = self.role1_id.trans_implied_ids.ids
|
||||
role1_group_ids.append(self.role1_id.group_id.id)
|
||||
role_group_ids = sorted(set(role1_group_ids))
|
||||
@@ -110,19 +114,20 @@ class TestUserRole(TransactionCase):
|
||||
|
||||
# Configure the user with role1 and role2
|
||||
self.user_id.write(
|
||||
{'role_line_ids': [
|
||||
(0, 0, {'role_id': self.role1_id.id}),
|
||||
(0, 0, {'role_id': self.role2_id.id}),
|
||||
]})
|
||||
{
|
||||
"role_line_ids": [
|
||||
(0, 0, {"role_id": self.role1_id.id}),
|
||||
(0, 0, {"role_id": self.role2_id.id}),
|
||||
]
|
||||
}
|
||||
)
|
||||
# Remove role2
|
||||
self.role2_id.unlink()
|
||||
user_group_ids = sorted(set([
|
||||
group.id for group in self.user_id.groups_id]))
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
self.assertEqual(user_group_ids, role1_group_ids)
|
||||
# Remove role1
|
||||
self.role1_id.unlink()
|
||||
user_group_ids = sorted(set([
|
||||
group.id for group in self.user_id.groups_id]))
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
self.assertEqual(user_group_ids, [])
|
||||
|
||||
def test_role_line_unlink(self):
|
||||
@@ -133,57 +138,71 @@ class TestUserRole(TransactionCase):
|
||||
|
||||
# Configure the user with role1 and role2
|
||||
self.user_id.write(
|
||||
{'role_line_ids': [
|
||||
(0, 0, {'role_id': self.role1_id.id}),
|
||||
(0, 0, {'role_id': self.role2_id.id}),
|
||||
]})
|
||||
{
|
||||
"role_line_ids": [
|
||||
(0, 0, {"role_id": self.role1_id.id}),
|
||||
(0, 0, {"role_id": self.role2_id.id}),
|
||||
]
|
||||
}
|
||||
)
|
||||
# Remove role2 from the user
|
||||
self.user_id.role_line_ids.filtered(
|
||||
lambda l: l.role_id.id == self.role2_id.id).unlink()
|
||||
user_group_ids = sorted(set([
|
||||
group.id for group in self.user_id.groups_id]))
|
||||
lambda l: l.role_id.id == self.role2_id.id
|
||||
).unlink()
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
self.assertEqual(user_group_ids, role1_group_ids)
|
||||
# Remove role1 from the user
|
||||
self.user_id.role_line_ids.filtered(
|
||||
lambda l: l.role_id.id == self.role1_id.id).unlink()
|
||||
user_group_ids = sorted(set([
|
||||
group.id for group in self.user_id.groups_id]))
|
||||
lambda l: l.role_id.id == self.role1_id.id
|
||||
).unlink()
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
self.assertEqual(user_group_ids, [])
|
||||
|
||||
def test_default_user_roles(self):
|
||||
self.default_user.write({
|
||||
'role_line_ids': [
|
||||
(0, 0, {
|
||||
'role_id': self.role1_id.id,
|
||||
}),
|
||||
(0, 0, {
|
||||
'role_id': self.role2_id.id,
|
||||
})
|
||||
]
|
||||
})
|
||||
user = self.user_model.create({
|
||||
'name': "USER TEST (DEFAULT ROLES)",
|
||||
'login': 'user_test_default_roles'
|
||||
})
|
||||
self.default_user.write(
|
||||
{
|
||||
"role_line_ids": [
|
||||
(0, 0, {"role_id": self.role1_id.id}),
|
||||
(0, 0, {"role_id": self.role2_id.id}),
|
||||
]
|
||||
}
|
||||
)
|
||||
user = self.user_model.create(
|
||||
{"name": "USER TEST (DEFAULT ROLES)", "login": "user_test_default_roles"}
|
||||
)
|
||||
roles = self.role_model.browse([self.role1_id.id, self.role2_id.id])
|
||||
self.assertEqual(user.role_ids, roles)
|
||||
|
||||
def test_user_role_different_company(self):
|
||||
self.user_id.write({'company_id': self.company1.id})
|
||||
self.user_id.write({'role_line_ids': [(0, 0, {
|
||||
'role_id': self.role2_id.id,
|
||||
'company_id': self.company2.id})]})
|
||||
self.user_id.write({"company_id": self.company1.id})
|
||||
self.user_id.write(
|
||||
{
|
||||
"role_line_ids": [
|
||||
(
|
||||
0,
|
||||
0,
|
||||
{"role_id": self.role2_id.id, "company_id": self.company2.id},
|
||||
)
|
||||
]
|
||||
}
|
||||
)
|
||||
# Check that user does not have any groups
|
||||
self.assertEquals(
|
||||
self.user_id.groups_id, self.env['res.groups'].browse())
|
||||
self.assertEquals(self.user_id.groups_id, self.env["res.groups"].browse())
|
||||
|
||||
def test_user_role_same_company(self):
|
||||
self.user_id.write({'company_id': self.company1.id})
|
||||
self.user_id.write({'role_line_ids': [(0, 0, {
|
||||
'role_id': self.role1_id.id,
|
||||
'company_id': self.company1.id})]})
|
||||
user_group_ids = sorted(set(
|
||||
[group.id for group in self.user_id.groups_id]))
|
||||
self.user_id.write({"company_id": self.company1.id})
|
||||
self.user_id.write(
|
||||
{
|
||||
"role_line_ids": [
|
||||
(
|
||||
0,
|
||||
0,
|
||||
{"role_id": self.role1_id.id, "company_id": self.company1.id},
|
||||
)
|
||||
]
|
||||
}
|
||||
)
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
role_group_ids = self.role1_id.trans_implied_ids.ids
|
||||
role_group_ids.append(self.role1_id.group_id.id)
|
||||
role_group_ids = sorted(set(role_group_ids))
|
||||
@@ -191,12 +210,15 @@ class TestUserRole(TransactionCase):
|
||||
self.assertEqual(user_group_ids, role_group_ids)
|
||||
|
||||
def test_user_role_no_company(self):
|
||||
self.user_id.write({'company_id': self.company1.id})
|
||||
self.user_id.write({'role_line_ids': [(0, 0, {
|
||||
'role_id': self.role2_id.id,
|
||||
'company_id': False})]})
|
||||
user_group_ids = sorted(set(
|
||||
[group.id for group in self.user_id.groups_id]))
|
||||
self.user_id.write({"company_id": self.company1.id})
|
||||
self.user_id.write(
|
||||
{
|
||||
"role_line_ids": [
|
||||
(0, 0, {"role_id": self.role2_id.id, "company_id": False})
|
||||
]
|
||||
}
|
||||
)
|
||||
user_group_ids = sorted({group.id for group in self.user_id.groups_id})
|
||||
role_group_ids = self.role2_id.trans_implied_ids.ids
|
||||
role_group_ids.append(self.role2_id.group_id.id)
|
||||
role_group_ids = sorted(set(role_group_ids))
|
||||
|
||||
Reference in New Issue
Block a user