flectra/addons/website_crm_partner_assign/tests/test_partner_assign.py
2018-01-16 02:34:37 -08:00

144 lines
6.7 KiB
Python

# -*- coding: utf-8 -*-
# Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details.
try:
from unittest.mock import patch
except ImportError:
from mock import patch
from flectra.exceptions import AccessError
from flectra.tests.common import TransactionCase
from flectra.addons.crm.tests.common import TestCrmCases
class TestPartnerAssign(TransactionCase):
def setUp(self):
super(TestPartnerAssign, self).setUp()
def geo_find(addr):
return {
'Wavre, Belgium': (50.7158956, 4.6128075),
'Cannon Hill Park, B46 3AG Birmingham, United Kingdom': (52.45216, -1.898578),
}.get(addr)
patcher = patch('flectra.addons.base_geolocalize.models.res_partner.geo_find', wraps=geo_find)
patcher.start()
self.addCleanup(patcher.stop)
patcher = patch('flectra.addons.website_crm_partner_assign.models.crm_lead.geo_find',
wraps=geo_find)
patcher.start()
self.addCleanup(patcher.stop)
def test_partner_assign(self):
""" Test the automatic assignation using geolocalisation """
partner2 = self.env.ref('base.res_partner_2')
lead = self.env.ref('crm.crm_case_21')
# In order to test find nearest Partner functionality and assign to opportunity,
# I Set Geo Lattitude and Longitude according to partner address.
partner2.geo_localize()
# I check Geo Latitude and Longitude of partner after set
self.assertTrue(50 < partner2.partner_latitude < 51, "Latitude is wrong: 50 < %s < 51" % partner2.partner_latitude)
self.assertTrue(3 < partner2.partner_longitude < 5, "Longitude is wrong: 3 < %s < 5" % partner2.partner_longitude)
# I assign nearest partner to opportunity.
lead.assign_partner()
# I check assigned partner of opportunity who is nearest Geo Latitude and Longitude of opportunity.
self.assertEqual(lead.partner_assigned_id, self.env.ref('base.res_partner_18'), "Opportuniy is not assigned nearest partner")
self.assertTrue(50 < lead.partner_latitude < 55, "Latitude is wrong: 50 < %s < 55" % lead.partner_latitude)
self.assertTrue(-4 < lead.partner_longitude < -1, "Longitude is wrong: -4 < %s < -1" % lead.partner_longitude)
# I forward this opportunity to its nearest partner.
context = dict(self.env.context, default_model='crm.lead', default_res_id=lead.id, active_ids=lead.ids)
lead_forwarded = self.env['crm.lead.forward.to.partner'].with_context(context).create({})
try:
lead_forwarded.action_forward()
except:
pass
class TestPartnerLeadPortal(TestCrmCases):
def setUp(self):
super(TestPartnerLeadPortal, self).setUp()
# Partner Grade
self.grade = self.env['res.partner.grade'].create({
'name': "Grade Test",
'partner_weight': 42,
'sequence': 3,
})
# Integrating user/partner, having a salesman
self.portal_user = self.env['res.users'].with_context({'no_reset_password': True, 'mail_notrack': True}).create({
'name': 'Super Customer Flectra Intregrating Partner',
'email': 'super.partner@ododo.com',
'login': 'superpartner',
'groups_id': [(4, self.env.ref('base.group_portal').id)],
'user_id': self.crm_salesman.id,
'grade_id': self.grade.id,
})
self.portal_partner = self.portal_user.partner_id
# New lead, assigned to the new portal
self.lead = self.env['crm.lead'].with_context(mail_notrack=True).create({
'type': "lead",
'name': "Test lead new",
'user_id': False,
'team_id': False,
'description': "This is the description of the test new lead.",
'partner_assigned_id': self.portal_partner.id
})
# Sales Team of crm_salesman
self.team = self.env['crm.team'].with_context(mail_notrack=True).create({
'name': 'Test Team FOR THE WIN',
'use_leads': True,
'use_opportunities': True,
'member_ids': [(6, 0, [self.crm_salesman.id])],
})
def test_partner_lead_accept(self):
""" Test an integrating partner accepting the lead """
team_before = self.lead.team_id
user_before = self.lead.user_id
self.lead.sudo(self.portal_user.id).partner_interested(comment="Oh yeah, I take that lead !")
self.assertEqual(self.lead.type, 'opportunity', 'Bad Type: accepted lead by portal user should become an opportunity.')
self.assertEqual(self.lead.team_id, team_before, 'Accepting lead does not change the sales team.')
self.assertEqual(self.lead.user_id, user_before, 'Accepting lead does not change the salesman.')
def test_partner_lead_decline(self):
""" Test an integrating partner decline the lead """
self.lead.sudo(self.portal_user.id).partner_desinterested(comment="No thanks, I have enough leads !", contacted=True, spam=False)
self.assertFalse(self.lead.partner_assigned_id.id, 'The partner_assigned_id of the declined lead should be False.')
self.assertTrue(self.portal_user.partner_id in self.lead.sudo().partner_declined_ids, 'Partner who has declined the lead should be in the declined_partner_ids.')
def test_lead_access_right(self):
""" Test another portal user can not write on every leads """
# portal user having no right
poor_portal_user = self.env['res.users'].with_context({'no_reset_password': True, 'mail_notrack': True}).create({
'name': 'Poor Partner (not integrating one)',
'email': 'poor.partner@ododo.com',
'login': 'poorpartner',
'groups_id': [(4, self.env.ref('base.group_portal').id)],
})
# try to accept a lead that is not mine
with self.assertRaises(AccessError):
self.lead.sudo(poor_portal_user.id).partner_interested(comment="Oh yeah, I take that lead !")
def test_lead_creation(self):
""" Test the opportinuty creation from portal """
data = self.env['crm.lead'].sudo(self.portal_user.id).create_opp_portal({
'title': "L'ours bleu",
'description': 'A good joke',
'contact_name': 'Renaud Rutten',
})
opportunity = self.env['crm.lead'].browse(data['id'])
salesmanteam = self.env['crm.team']._get_default_team_id(user_id=self.portal_user.user_id.id)
self.assertEqual(opportunity.team_id, salesmanteam, 'The created opportunity should have the same team as the salesman default team of the opportunity creator.')
self.assertEqual(opportunity.partner_assigned_id, self.portal_partner, 'Assigned Partner of created opportunity is the (portal) creator.')