add tests ensuring timeout is passed

auth0/v3/test/authentication/test_base.py

@@ -1,5 +1,7 @@
 import base64
 import json
+from time import sleep
+
 import mock
 import sys
 import requests
@@ -43,15 +45,15 @@ def test_telemetry_disabled(self):
 
     @mock.patch('requests.post')
     def test_post(self, mock_post):
-        ab = AuthenticationBase('auth0.com', telemetry=False)
+        ab = AuthenticationBase('auth0.com', telemetry=False, timeout=(10, 2))
 
         mock_post.return_value.status_code = 200
         mock_post.return_value.text = '{"x": "y"}'
 
         data = ab.post('the-url', data={'a': 'b'}, headers={'c': 'd'})
 
         mock_post.assert_called_with(url='the-url', json={'a': 'b'},
-                headers={'c': 'd', 'Content-Type': 'application/json'}, timeout=5.0)
+                headers={'c': 'd', 'Content-Type': 'application/json'}, timeout=(10, 2))
 
         self.assertEqual(data, {'x': 'y'})
 
@@ -174,15 +176,15 @@ def test_post_error_with_no_response_text(self, mock_post):
 
     @mock.patch('requests.get')
     def test_get(self, mock_get):
-        ab = AuthenticationBase('auth0.com', telemetry=False)
+        ab = AuthenticationBase('auth0.com', telemetry=False, timeout=(10, 2))
 
         mock_get.return_value.status_code = 200
         mock_get.return_value.text = '{"x": "y"}'
 
         data = ab.get('the-url', params={'a': 'b'}, headers={'c': 'd'})
 
         mock_get.assert_called_with(url='the-url', params={'a': 'b'},
-                headers={'c': 'd', 'Content-Type': 'application/json'}, timeout=5.0)
+                headers={'c': 'd', 'Content-Type': 'application/json'}, timeout=(10, 2))
 
         self.assertEqual(data, {'x': 'y'})
 

auth0/v3/test/management/test_blacklists.py

@@ -5,6 +5,12 @@
 
 class TestBlacklists(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Blacklists(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.blacklists.RestClient')
     def test_get(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_client_grants.py

@@ -5,6 +5,12 @@
 
 class TestClientGrants(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = ClientGrants(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.client_grants.RestClient')
     def test_all(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_clients.py

@@ -5,6 +5,12 @@
 
 class TestClients(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Clients(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.clients.RestClient')
     def test_all(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_connections.py

@@ -5,6 +5,12 @@
 
 class TestConnection(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Connections(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.connections.RestClient')
     def test_all(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_custom_domains.py

@@ -5,6 +5,12 @@
 
 class TestCustomDomains(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = CustomDomains(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.custom_domains.RestClient')
     def test_get_all(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_device_credentials.py

@@ -5,6 +5,12 @@
 
 class TestDeviceCredentials(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = DeviceCredentials(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.device_credentials.RestClient')
     def test_get(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_email_endpoints.py

@@ -5,6 +5,12 @@
 
 class TestClients(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = EmailTemplates(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.email_templates.RestClient')
     def test_create(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_emails.py

@@ -5,6 +5,12 @@
 
 class TestEmails(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Emails(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.emails.RestClient')
     def test_get(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_grants.py

@@ -5,6 +5,12 @@
 
 class TestGrants(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Grants(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.grants.RestClient')
     def test_get_all(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_guardian.py

@@ -5,6 +5,12 @@
 
 class TestGuardian(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Guardian(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.guardian.RestClient')
     def test_all_factors(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_jobs.py

@@ -5,6 +5,12 @@
 
 class TestJobs(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Jobs(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.jobs.RestClient')
     def test_get(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_logs.py

@@ -5,6 +5,12 @@
 
 class TestLogs(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Logs(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.logs.RestClient')
     def test_search(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_resource_servers.py

@@ -5,6 +5,12 @@
 
 class TestResourceServers(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = ResourceServers(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.resource_servers.RestClient')
     def test_create(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_rest.py

@@ -2,14 +2,111 @@
 import sys
 import json
 import base64
-import requests
+
 import mock
+import requests
+
 from ...management.rest import RestClient
 from ...exceptions import Auth0Error
 
 
 class TestRest(unittest.TestCase):
 
+    def test_get_can_timeout(self):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=0.00001)
+
+        with self.assertRaises(requests.exceptions.Timeout):
+            rc.get('http://auth0.com')
+
+    def test_post_can_timeout(self):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=0.00001)
+
+        with self.assertRaises(requests.exceptions.Timeout):
+            rc.post('http://auth0.com')
+
+    def test_put_can_timeout(self):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=0.00001)
+
+        with self.assertRaises(requests.exceptions.Timeout):
+            rc.put('http://auth0.com')
+
+    def test_patch_can_timeout(self):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=0.00001)
+
+        with self.assertRaises(requests.exceptions.Timeout):
+            rc.patch('http://auth0.com')
+
+    def test_delete_can_timeout(self):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=0.00001)
+
+        with self.assertRaises(requests.exceptions.Timeout):
+            rc.delete('http://auth0.com')
+
+    @mock.patch('requests.get')
+    def test_get_custom_timeout(self, mock_get):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=(10, 2))
+        headers = {
+            'Authorization': 'Bearer a-token',
+            'Content-Type': 'application/json',
+        }
+        mock_get.return_value.text = '["a", "b"]'
+        mock_get.return_value.status_code = 200
+
+        rc.get('the-url')
+        mock_get.assert_called_with('the-url', params=None, headers=headers, timeout=(10, 2))
+
+    @mock.patch('requests.post')
+    def test_post_custom_timeout(self, mock_post):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=(10, 2))
+        headers = {
+            'Authorization': 'Bearer a-token',
+            'Content-Type': 'application/json',
+        }
+        mock_post.return_value.text = '["a", "b"]'
+        mock_post.return_value.status_code = 200
+
+        rc.post('the-url')
+        mock_post.assert_called_with('the-url', json=None, headers=headers, timeout=(10, 2))
+
+    @mock.patch('requests.put')
+    def test_put_custom_timeout(self, mock_put):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=(10, 2))
+        headers = {
+            'Authorization': 'Bearer a-token',
+            'Content-Type': 'application/json',
+        }
+        mock_put.return_value.text = '["a", "b"]'
+        mock_put.return_value.status_code = 200
+
+        rc.put('the-url')
+        mock_put.assert_called_with('the-url', json=None, headers=headers, timeout=(10, 2))
+
+    @mock.patch('requests.patch')
+    def test_patch_custom_timeout(self, mock_patch):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=(10, 2))
+        headers = {
+            'Authorization': 'Bearer a-token',
+            'Content-Type': 'application/json',
+        }
+        mock_patch.return_value.text = '["a", "b"]'
+        mock_patch.return_value.status_code = 200
+
+        rc.patch('the-url')
+        mock_patch.assert_called_with('the-url', json=None, headers=headers, timeout=(10, 2))
+
+    @mock.patch('requests.delete')
+    def test_delete_custom_timeout(self, mock_delete):
+        rc = RestClient(jwt='a-token', telemetry=False, timeout=(10, 2))
+        headers = {
+            'Authorization': 'Bearer a-token',
+            'Content-Type': 'application/json',
+        }
+        mock_delete.return_value.text = '["a", "b"]'
+        mock_delete.return_value.status_code = 200
+
+        rc.delete('the-url')
+        mock_delete.assert_called_with('the-url', params={}, json=None, headers=headers, timeout=(10, 2))
+
     @mock.patch('requests.get')
     def test_get(self, mock_get):
         rc = RestClient(jwt='a-token', telemetry=False)

auth0/v3/test/management/test_roles.py

@@ -5,6 +5,12 @@
 
 class TestRoles(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Roles(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.roles.RestClient')
     def test_list(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_rules.py

@@ -5,6 +5,12 @@
 
 class TestRules(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Rules(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.rules.RestClient')
     def test_all(self, mock_rc):
         mock_instance = mock_rc.return_value

auth0/v3/test/management/test_rules_configs.py

@@ -3,7 +3,13 @@
 from ...management.rules_configs import RulesConfigs
 
 
-class TestRules(unittest.TestCase):
+class TestRulesConfigs(unittest.TestCase):
+
+    def test_init_with_optionals(self):
+        t = RulesConfigs(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
 
     @mock.patch('auth0.v3.management.rules_configs.RestClient')
     def test_all(self, mock_rc):

auth0/v3/test/management/test_stats.py

@@ -3,7 +3,13 @@
 from ...management.stats import Stats
 
 
-class TestTickets(unittest.TestCase):
+class TestStats(unittest.TestCase):
+
+    def test_init_with_optionals(self):
+        t = Stats(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
 
     @mock.patch('auth0.v3.management.stats.RestClient')
     def test_active_users(self, mock_rc):

auth0/v3/test/management/test_tenants.py

@@ -5,6 +5,12 @@
 
 class TestTenants(unittest.TestCase):
 
+    def test_init_with_optionals(self):
+        t = Tenants(domain='domain', token='jwttoken', telemetry=False, timeout=(10, 2))
+        self.assertEqual(t.client.timeout, (10, 2))
+        telemetry_header = t.client.base_headers.get('Auth0-Client', None)
+        self.assertEqual(telemetry_header, None)
+
     @mock.patch('auth0.v3.management.tenants.RestClient')
     def test_get(self, mock_rc):
         mock_instance = mock_rc.return_value