[FIX] website_crm_partner_assign: mock geolocalisation during tests

1 files changed, 20 insertions, 0 deletions

diff --git a/addons/website_crm_partner_assign/tests/test_partner_assign.py b/addons/website_crm_partner_assign/tests/test_partner_assign.py
index f5c056fa839..6bf5e00af76 100644
--- a/addons/website_crm_partner_assign/tests/test_partner_assign.py
+++ b/addons/website_crm_partner_assign/tests/test_partner_assign.py
@@ -1,11 +1,31 @@
 # -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
+from mock import patch
+
 from odoo.tests.common import TransactionCase
 
 
 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('odoo.addons.base_geolocalize.models.res_partner.geo_find', wraps=geo_find)
+        patcher.start()
+        self.addCleanup(patcher.stop)
+
+        patcher = patch('odoo.addons.website_crm_partner_assign.models.crm_lead.geo_find',
+                        wraps=geo_find)
+        patcher.start()
+        self.addCleanup(patcher.stop)
+
     def test_00_partner_assign(self):
         partner2 = self.env.ref('base.res_partner_2')
         lead = self.env.ref('crm.crm_case_21')