improve tests

tests/mocks/test_retry.py

@@ -13,9 +13,9 @@
 from .table_responses import list_response
 
 
-class NoRetriesEndpoint(Endpoint):
-    def __init__(self, root_url, token):
-        super().__init__(root_url, 'no-retries', token, max_requests_retries=0)
+class RetriesEndpoint(Endpoint):
+    def __init__(self, root_url, token, retries_count):
+        super().__init__(root_url, 'retries', token, max_requests_retries=retries_count)
 
     def list(self):
         return self._get(self.base_url)
@@ -29,7 +29,8 @@ def setUp(self):
         token = 'dummy_token'
         base_url = 'https://connection.keboola.com/'
         self.tables = Tables(base_url, token)
-        self.no_retries = NoRetriesEndpoint(base_url, token)
+        self.no_retries = RetriesEndpoint(base_url, token, 0)
+        self.two_retries = RetriesEndpoint(base_url, token, 2)
 
     @responses.activate
     @patch('time.sleep', return_value=None)
@@ -104,17 +105,65 @@ def test_no_retries(self, sleep_mock):
         responses.add(
             responses.Response(
                 method='GET',
-                url='https://connection.keboola.com/v2/storage/no-retries',
+                url='https://connection.keboola.com/v2/storage/retries',
                 json=list_response,
                 status=502
             )
         )
         responses.add(
             responses.Response(
                 method='GET',
-                url='https://connection.keboola.com/v2/storage/no-retries',
+                url='https://connection.keboola.com/v2/storage/retries',
                 json=list_response,
             )
         )
         with self.assertRaises(requests.exceptions.HTTPError):
             self.no_retries.list()
+
+    @responses.activate
+    @patch('time.sleep', return_value=None)
+    def test_two_retries_fail(self, sleep_mock):
+        """
+        Request wont be retried for endpoints that configured it.
+        """
+        for i in range(3):
+            responses.add(
+                responses.Response(
+                    method='GET',
+                    url='https://connection.keboola.com/v2/storage/retries',
+                    json=list_response,
+                    status=502
+                )
+            )
+        responses.add(
+            responses.Response(
+                method='GET',
+                url='https://connection.keboola.com/v2/storage/retries',
+                json=list_response,
+            )
+        )
+        with self.assertRaises(requests.exceptions.HTTPError):
+            self.two_retries.list()
+
+    @responses.activate
+    @patch('time.sleep', return_value=None)
+    def test_two_retries_ok(self, sleep_mock):
+        """
+        Request wont be retried for endpoints that configured it.
+        """
+        responses.add(
+            responses.Response(
+                method='GET',
+                url='https://connection.keboola.com/v2/storage/retries',
+                json=list_response,
+                status=502
+            )
+        )
+        responses.add(
+            responses.Response(
+                method='GET',
+                url='https://connection.keboola.com/v2/storage/retries',
+                json=list_response,
+            )
+        )
+        assert isinstance(self.two_retries.list(), list)