test NEXUS-817: fixed retries in tests

Author: pawel-kmiecik

_test_contract/conftest.py

@@ -4,40 +4,66 @@
 
 import pytest
 
-from unstructured_client import UnstructuredClient
+from unstructured_client import UnstructuredClient, utils
 
 FAKE_API_KEY = "91pmLBeETAbXCpNylRsLq11FdiZPTk"
 
+
 @pytest.fixture(scope="module")
 def platform_client(platform_api_url) -> UnstructuredClient:
+    # settings the retry config to always try 3 times after a fail = 4 requests sent
     _client = UnstructuredClient(
         api_key_auth=FAKE_API_KEY,
         server_url=platform_api_url,
+        retry_config=utils.RetryConfig(
+            "backoff", utils.BackoffStrategy(
+                initial_interval=3000,
+                max_interval=3000,
+                exponent=1.0,
+                max_elapsed_time=8000
+            ),
+            retry_connection_errors=True
+        )
     )
     yield _client
 
+
 @pytest.fixture(scope="module")
 def serverless_client(serverless_api_url) -> UnstructuredClient:
+    # settings the retry config to always try 3 times after a fail = 4 requests sent
     _client = UnstructuredClient(
         api_key_auth=FAKE_API_KEY,
-        server_url=serverless_api_url
+        server_url=serverless_api_url,
+        retry_config = utils.RetryConfig(
+        "backoff", utils.BackoffStrategy(
+            initial_interval=3000,
+            max_interval=3000,
+            exponent=1.0,
+            max_elapsed_time=8000
+        ),
+        retry_connection_errors=True
+    )
     )
     yield _client
 
+
 @pytest.fixture(autouse=True)
 def mock_sleep(mocker, freezer):
     sleep_mock = mocker.patch("time.sleep")
     sleep_mock.side_effect = lambda seconds: freezer.tick(timedelta(seconds=seconds))
     yield sleep_mock
 
+
 @pytest.fixture(scope="module")
 def platform_api_url():
     return "https://platform.unstructuredapp.io"
 
+
 @pytest.fixture(scope="module")
 def serverless_api_url():
     return "https://api.unstructuredapp.io"
 
+
 @pytest.fixture(scope="module")
 def dummy_partitioned_text():
     return """[
@@ -82,8 +108,9 @@ def dummy_partitioned_text():
   }
 ]"""
 
+
 @pytest.fixture(scope="module")
 def doc_path() -> Path:
     samples_path = Path(__file__).resolve().parents[1] / "_sample_docs"
     assert samples_path.exists()
-    return samples_path
+    return samples_path

_test_contract/platform_api/test_jobs.py

@@ -119,6 +119,7 @@ def test_get_job_error(httpx_mock, platform_client: UnstructuredClient, platform
         headers={"Content-Type": "application/json"},
         json={"detail": "Internal server error"},
         url=url,
+        is_reusable=True,
     )
 
     with pytest.raises(SDKError) as e:
@@ -132,7 +133,7 @@ def test_get_job_error(httpx_mock, platform_client: UnstructuredClient, platform
     assert e.value.message == "API error occurred"
 
     requests = httpx_mock.get_requests()
-    assert len(requests) == 1
+    assert len(requests) == 4
     request = requests[0]
     assert request.method == "GET"
     assert request.url == url

_test_contract/test_retries.py

@@ -90,7 +90,7 @@ def test_partition_retries(
     requests = httpx_mock.get_requests()
     assert len(requests) == 3
     for request in requests:
-        assert request.method == "POSt"
+        assert request.method == "POST"
         assert request.url == url
 
     assert len(partition_response.elements) > 0