19.0 vanilla

odoo-bringout-oca-ocb-iap/iap/tests/common.py

@@ -6,7 +6,7 @@ from unittest.mock import patch
 
 from odoo import exceptions
 from odoo.addons.iap.tools import iap_tools
-from odoo.addons.iap.models.iap_enrich_api import IapEnrichAPI
+from odoo.addons.iap.models.iap_enrich_api import IapEnrichApi
 from odoo.tests import common
 
 
@@ -49,11 +49,8 @@ class MockIAPEnrich(common.TransactionCase):
                         result[str(lead_id)].update(email_data[email])
                 return result
 
-        try:
-            with patch.object(IapEnrichAPI, '_contact_iap', side_effect=_contact_iap) as contact_iap_mock:
-                yield
-        finally:
-            pass
+        with patch.object(IapEnrichApi, '_contact_iap', side_effect=_contact_iap):
+            yield
 
     @classmethod
     def _init_iap_mock(cls):
@@ -137,6 +134,34 @@ class MockIAPEnrich(common.TransactionCase):
             'country_name': self.base_de.name,
         }
 
+    def _get_iap_dnb_company_data(self, base_name, service=None, add_values=None):
+        return {
+            'city': 'Mönchengladbach',
+            'country_code': self.base_de.code,
+            'domain': '%s.de' % base_name,
+            'duns': '123456789',
+            'employees': 314,
+            'entity_type': 'private',
+            'industry_code': 'J',
+            'name': '%s GmbH' % base_name,
+            'phone': '4930499193937',
+            'state_code': self.de_state_st.code,
+            'street': 'Mennrather Str. 123456',
+            'unspsc_codes': [
+                [541512, 'Computer Systems Design Services'],
+                [518210, 'Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services'],
+                [541611, 'Administrative Management and General Management Consulting Services'],
+                [87429902, 'Business management consultant'],
+                [73790200, 'Computer related consulting services'],
+                [73749902, 'Data processing service'],
+                [51, 'Data Processing']
+            ],
+            'vat': '7LRA8',
+            'website': '%s.de' % base_name,
+            'estimated_annual_revenue': '10M USD',
+            'zip': '41179'
+        }
+
     def _get_iap_contact_data(self, base_name, service=None, add_values=None):
         people_data = []
         for index in range(2):

odoo-bringout-oca-ocb-iap/iap/tests/test_iap.py

@@ -1,10 +1,30 @@
 # -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
-from odoo.tests.common import TransactionCase
+from odoo.exceptions import AccessError
+
+from odoo.addons.base.tests.common import TransactionCaseWithUserDemo
 
 
-class TestIAP(TransactionCase):
-    def test_get_account(self):
-        account = self.env["iap.account"].get("random_service_name")
-        self.assertTrue(account.account_token, "Must be able to read the field")
+class TestIAP(TransactionCaseWithUserDemo):
+    def test_get_iap_account(self):
+        user_demo = self.user_demo
+        user_admin = self.ref("base.user_admin")
+        service_name = 'random_service_name'
+
+        self.env['iap.service'].create({
+            'name': service_name,
+            'description': 'test service',
+            'unit_name': 'credit',
+            'integer_balance': True,
+            'technical_name': service_name,
+        })
+
+        account = self.env['iap.account'].with_user(user_demo).get('random_service_name')
+
+        # User demo can access an IAP account but not its token (must sudo)
+        with self.assertRaises(AccessError):
+            self.assertTrue(account.with_user(user_demo).account_token)
+
+        # Admin can access the token
+        self.assertTrue(account.with_user(user_admin).account_token, "Must be able to read the field")