|
|
|
@ -4,104 +4,93 @@ |
|
|
|
|
|
|
|
import datetime |
|
|
|
|
|
|
|
from openerp.tools import DEFAULT_SERVER_DATE_FORMAT |
|
|
|
from openerp.tests.common import TransactionCase |
|
|
|
from odoo.tools import DEFAULT_SERVER_DATE_FORMAT |
|
|
|
from odoo.tests.common import TransactionCase |
|
|
|
|
|
|
|
|
|
|
|
class TestUserRole(TransactionCase): |
|
|
|
|
|
|
|
def setUp(self): |
|
|
|
super(TestUserRole, self).setUp() |
|
|
|
self.imd_model = self.registry('ir.model.data') |
|
|
|
self.user_model = self.registry('res.users') |
|
|
|
self.role_model = self.registry('res.users.role') |
|
|
|
self.user_model = self.env['res.users'] |
|
|
|
self.role_model = self.env['res.users.role'] |
|
|
|
|
|
|
|
self.user_id = self.user_model.create( |
|
|
|
self.cr, self.uid, |
|
|
|
{'name': u"USER TEST (ROLES)", 'login': 'user_test_roles'}) |
|
|
|
|
|
|
|
# ROLE_1 |
|
|
|
self.group_user_id = self.imd_model.get_object_reference( |
|
|
|
self.cr, self.uid, 'base', 'group_user')[1] |
|
|
|
self.group_no_one_id = self.imd_model.get_object_reference( |
|
|
|
self.cr, self.uid, 'base', 'group_no_one')[1] |
|
|
|
self.group_user_id = self.env.ref('base.group_user') |
|
|
|
self.group_no_one_id = self.env.ref('base.group_no_one') |
|
|
|
vals = { |
|
|
|
'name': u"ROLE_1", |
|
|
|
'implied_ids': [6, 0, [self.group_user_id, self.group_no_one_id]], |
|
|
|
'implied_ids': [ |
|
|
|
(6, 0, [self.group_user_id.id, self.group_no_one_id.id])], |
|
|
|
} |
|
|
|
self.role1_id = self.role_model.create(self.cr, self.uid, vals) |
|
|
|
self.role1_id = self.role_model.create(vals) |
|
|
|
|
|
|
|
# ROLE_2 |
|
|
|
self.group_multi_currency_id = self.imd_model.get_object_reference( |
|
|
|
self.cr, self.uid, 'base', 'group_multi_currency')[1] |
|
|
|
self.group_sale_manager_id = self.imd_model.get_object_reference( |
|
|
|
self.cr, self.uid, 'base', 'group_sale_manager')[1] |
|
|
|
self.group_multi_currency_id = self.env.ref( |
|
|
|
'base.group_multi_currency') |
|
|
|
self.group_settings_id = self.env.ref('base.group_system') |
|
|
|
vals = { |
|
|
|
'name': u"ROLE_2", |
|
|
|
'implied_ids': [6, 0, [self.group_multi_currency_id, |
|
|
|
self.group_sale_manager_id]], |
|
|
|
'implied_ids': [ |
|
|
|
(6, 0, [self.group_multi_currency_id.id, |
|
|
|
self.group_settings_id.id])], |
|
|
|
} |
|
|
|
self.role2_id = self.role_model.create(self.cr, self.uid, vals) |
|
|
|
self.role2_id = self.role_model.create(vals) |
|
|
|
|
|
|
|
def test_role_1(self): |
|
|
|
role1 = self.role_model.browse(self.cr, self.uid, self.role1_id) |
|
|
|
self.user_model.write( |
|
|
|
self.cr, self.uid, [self.user_id], |
|
|
|
{'role_line_ids': [(0, 0, {'role_id': self.role1_id})]}) |
|
|
|
user = self.user_model.browse(self.cr, self.uid, self.user_id) |
|
|
|
user_group_ids = sorted(set([group.id for group in user.groups_id])) |
|
|
|
role_group_ids = role1.implied_ids.ids |
|
|
|
role_group_ids.append(role1.group_id.id) |
|
|
|
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])) |
|
|
|
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): |
|
|
|
role2 = self.role_model.browse(self.cr, self.uid, self.role2_id) |
|
|
|
self.user_model.write( |
|
|
|
self.cr, self.uid, [self.user_id], |
|
|
|
{'role_line_ids': [(0, 0, {'role_id': self.role2_id})]}) |
|
|
|
user = self.user_model.browse(self.cr, self.uid, self.user_id) |
|
|
|
user_group_ids = sorted(set([group.id for group in user.groups_id])) |
|
|
|
role_group_ids = role2.implied_ids.ids |
|
|
|
role_group_ids.append(role2.group_id.id) |
|
|
|
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])) |
|
|
|
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)) |
|
|
|
self.assertEqual(user_group_ids, role_group_ids) |
|
|
|
|
|
|
|
def test_role_1_2(self): |
|
|
|
role1 = self.role_model.browse(self.cr, self.uid, self.role1_id) |
|
|
|
role2 = self.role_model.browse(self.cr, self.uid, self.role2_id) |
|
|
|
self.user_model.write( |
|
|
|
self.cr, self.uid, [self.user_id], |
|
|
|
self.user_id.write( |
|
|
|
{'role_line_ids': [ |
|
|
|
(0, 0, {'role_id': self.role1_id}), |
|
|
|
(0, 0, {'role_id': self.role2_id}), |
|
|
|
(0, 0, {'role_id': self.role1_id.id}), |
|
|
|
(0, 0, {'role_id': self.role2_id.id}), |
|
|
|
]}) |
|
|
|
user = self.user_model.browse(self.cr, self.uid, self.user_id) |
|
|
|
user_group_ids = sorted(set([group.id for group in user.groups_id])) |
|
|
|
role1_group_ids = role1.implied_ids.ids |
|
|
|
role1_group_ids.append(role1.group_id.id) |
|
|
|
role2_group_ids = role2.implied_ids.ids |
|
|
|
role2_group_ids.append(role2.group_id.id) |
|
|
|
user_group_ids = sorted(set( |
|
|
|
[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 |
|
|
|
role2_group_ids.append(self.role2_id.group_id.id) |
|
|
|
role_group_ids = sorted(set(role1_group_ids + role2_group_ids)) |
|
|
|
self.assertEqual(user_group_ids, role_group_ids) |
|
|
|
|
|
|
|
def test_role_1_2_with_dates(self): |
|
|
|
role1 = self.role_model.browse(self.cr, self.uid, self.role1_id) |
|
|
|
today = datetime.date.today() |
|
|
|
today_str = today.strftime(DEFAULT_SERVER_DATE_FORMAT) |
|
|
|
yesterday = today - datetime.timedelta(days=1) |
|
|
|
yesterday_str = yesterday.strftime(DEFAULT_SERVER_DATE_FORMAT) |
|
|
|
self.user_model.write( |
|
|
|
self.cr, self.uid, [self.user_id], |
|
|
|
self.user_id.write( |
|
|
|
{'role_line_ids': [ |
|
|
|
# Role 1 should be enabled |
|
|
|
(0, 0, {'role_id': self.role1_id, 'date_from': today_str}), |
|
|
|
(0, 0, {'role_id': self.role1_id.id, 'date_from': today_str}), |
|
|
|
# Role 2 should be disabled |
|
|
|
(0, 0, {'role_id': self.role2_id, 'date_to': yesterday_str}), |
|
|
|
(0, 0, |
|
|
|
{'role_id': self.role2_id.id, 'date_to': yesterday_str}), |
|
|
|
]}) |
|
|
|
user = self.user_model.browse(self.cr, self.uid, self.user_id) |
|
|
|
user_group_ids = sorted(set([group.id for group in user.groups_id])) |
|
|
|
role1_group_ids = role1.implied_ids.ids |
|
|
|
role1_group_ids.append(role1.group_id.id) |
|
|
|
user_group_ids = sorted(set( |
|
|
|
[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)) |
|
|
|
self.assertEqual(user_group_ids, role_group_ids) |
Adrien Peiffer (ACSONE)
8 years ago
Dave Lasley