# -*- 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.')