You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
4.7 KiB
107 lines
4.7 KiB
# -*- coding: utf-8 -*-
|
|
# Copyright 2014 ABF OSIELL <http://osiell.com>
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
import datetime
|
|
|
|
from openerp.tools import DEFAULT_SERVER_DATE_FORMAT
|
|
from openerp.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_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]
|
|
vals = {
|
|
'name': u"ROLE_1",
|
|
'implied_ids': [6, 0, [self.group_user_id, self.group_no_one_id]],
|
|
}
|
|
self.role1_id = self.role_model.create(self.cr, self.uid, 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]
|
|
vals = {
|
|
'name': u"ROLE_2",
|
|
'implied_ids': [6, 0, [self.group_multi_currency_id,
|
|
self.group_sale_manager_id]],
|
|
}
|
|
self.role2_id = self.role_model.create(self.cr, self.uid, 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)
|
|
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)
|
|
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],
|
|
{'role_line_ids': [
|
|
(0, 0, {'role_id': self.role1_id}),
|
|
(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]))
|
|
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)
|
|
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],
|
|
{'role_line_ids': [
|
|
# Role 1 should be enabled
|
|
(0, 0, {'role_id': self.role1_id, 'date_from': today_str}),
|
|
# Role 2 should be disabled
|
|
(0, 0, {'role_id': self.role2_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)
|
|
role_group_ids = sorted(set(role1_group_ids))
|
|
self.assertEqual(user_group_ids, role_group_ids)
|