69 lines
3.1 KiB
Python
69 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
from flectra.tests import common
|
|
|
|
|
|
class TestCrmBranch(common.TransactionCase):
|
|
|
|
def setUp(self):
|
|
super(TestCrmBranch, self).setUp()
|
|
|
|
self.branch_1 = self.env.ref('base_branch_company.data_branch_1')
|
|
self.branch_2 = self.env.ref('base_branch_company.data_branch_2')
|
|
self.model_crm = self.env['crm.lead']
|
|
self.sale_user_group = self.env.ref('sales_team.group_sale_manager')
|
|
self.main_company = self.env.ref('base.main_company')
|
|
self.crm_user_group = self.env.ref('base.group_user')
|
|
|
|
self.user_1 = self.create_crm_user(self.main_company, self.branch_1, [self.branch_1],'user_1', [self.sale_user_group,
|
|
self.crm_user_group])
|
|
self.user_2 = self.create_crm_user(self.main_company, self.branch_2, [self.branch_2], 'user_2', [self.sale_user_group,
|
|
self.crm_user_group])
|
|
self.team1=self.env.ref('sales_team.team_sales_department')
|
|
self.team1.write({'branch_id': self.branch_1.id, 'user_id': self.user_1.id})
|
|
self.team2 = self.env.ref('sales_team.crm_team_1')
|
|
self.team2.write({'branch_id': self.branch_2.id, 'user_id': self.user_2.id})
|
|
|
|
self.lead_1 = self.lead_create(self.branch_1, self.team1, self.user_1.id)
|
|
self.lead_2 = self.lead_create(self.branch_2, self.team2, self.user_2.id)
|
|
|
|
def create_crm_user(self, main_company, branch, branch_ids, login_user, groups):
|
|
groups = [group.id for group in groups]
|
|
user_obj = self.env['res.users'].create({
|
|
'company_id': main_company.id,
|
|
'branch_ids': [(4, branch_id.id) for branch_id in branch_ids],
|
|
'default_branch_id': branch.id,
|
|
'company_ids': [(4, main_company.id)],
|
|
'groups_id': [(6, 0, groups)],
|
|
'login': login_user,
|
|
'name': 'CRM Test ' + login_user,
|
|
'email': 'demo@yourcompany.com',
|
|
'password': '123',
|
|
})
|
|
return user_obj
|
|
|
|
|
|
def lead_create(self, branch_id, team_id, user_id, ):
|
|
lead_name = 'CRM LEAD '
|
|
lead = self.model_crm.create({
|
|
'user_id': user_id,
|
|
'team_id': team_id.id,
|
|
'name': lead_name + branch_id.name,
|
|
'branch_id': branch_id.id,
|
|
})
|
|
return lead
|
|
|
|
def test_lead_authentication(self):
|
|
lead_ids = self.model_crm.sudo(self.user_2.id).search(
|
|
[('branch_id', '=', self.branch_1.id), ('id', '=', self.lead_1.id)])
|
|
self.assertEqual(lead_ids.ids, [], ('%s should not have'
|
|
' access to %s')
|
|
% (self.user_2.name, self.branch_1.name))
|
|
|
|
def test_team_branch(self):
|
|
lead = self.lead_create(self.branch_2, self.team2, self.user_2.id )
|
|
self.assertEqual(
|
|
lead.branch_id, self.branch_2, ('%s lead '
|
|
'should have %s as branch')
|
|
% (self.user_2.name,
|
|
self.branch_2.name))
|