1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo.addons.crm.tests.common import TestCrmCommon
from odoo.addons.iap.tests.common import MockIAPEnrich
from odoo.tests.common import users
class TestLeadEnrich(TestCrmCommon, MockIAPEnrich):
@classmethod
def setUpClass(cls):
super(TestLeadEnrich, cls).setUpClass()
cls.registry.enter_test_mode(cls.cr)
cls.leads = cls.env['crm.lead']
for x in range(0, 4):
cls.leads += cls.env['crm.lead'].create({
'name': 'Test %s' % x,
'email_from': 'test_mail_%s@megaexample.com' % x
})
@classmethod
def tearDownClass(cls):
cls.registry.leave_test_mode()
super().tearDownClass()
@users('user_sales_manager')
def test_enrich_internals(self):
leads = self.env['crm.lead'].browse(self.leads.ids)
leads[0].write({'partner_name': 'Already set', 'email_from': 'test@test1'})
leads.flush_recordset()
with self.mockIAPEnrichGateway(email_data={'test1': {'country_code': 'AU', 'state_code': 'NSW'}}):
leads.iap_enrich()
leads.flush_recordset()
self.assertEqual(leads[0].partner_name, 'Already set')
self.assertEqual(leads[0].country_id, self.env.ref('base.au'))
self.assertEqual(leads[0].state_id, self.env.ref('base.state_au_2'))
for lead in leads[1:]:
self.assertEqual(lead.partner_name, 'Simulator INC')
for lead in leads:
self.assertEqual(lead.street, 'Simulator Street')
# @users('sales_manager')
# def test_enrich_error_credit(self):
# leads = self.env['crm.lead'].browse(self.leads.ids)
# with self.mockIAPEnrichGateway(sim_error='credit'):
# leads.iap_enrich()
@users('user_sales_manager')
def test_enrich_error_jsonrpc_exception(self):
leads = self.env['crm.lead'].browse(self.leads.ids)
with self.mockIAPEnrichGateway(sim_error='jsonrpc_exception'):
leads.iap_enrich()
for lead in leads:
self.assertEqual(lead.street, False)
def test_lead_enrich_auto_setting(self):
cron = self.env.ref('crm_iap_enrich.ir_cron_lead_enrichment')
config = self.env['res.config.settings'].create({
'lead_enrich_auto': 'manual',
})
config.execute()
self.assertFalse(cron.active)
config.write({'lead_enrich_auto': 'auto'})
config.execute()
self.assertTrue(cron.active)
|
