feat NEXUS-817: added freezgun to control time-dependent operations

Author: pawel-kmiecik

_test_contract/conftest.py

@@ -0,0 +1,21 @@
+from datetime import timedelta
+
+import pytest
+
+from unstructured_client import UnstructuredClient
+
+FAKE_API_KEY = "91pmLBeETAbXCpNylRsLq11FdiZPTk"
+
+@pytest.fixture(scope="module")
+def client(platform_api_url) -> UnstructuredClient:
+    _client = UnstructuredClient(
+        api_key_auth=FAKE_API_KEY,
+        server_url=platform_api_url,
+    )
+    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

_test_contract/platform_api/conftest.py

@@ -1,29 +1,8 @@
-import os
-
 import pytest
 
-from unstructured_client import UnstructuredClient, RetryConfig
-from unstructured_client.utils import BackoffStrategy
-
-FAKE_API_KEY = "91pmLBeETAbXCpNylRsLq11FdiZPTk"
-
 
 @pytest.fixture(scope="module")
 def platform_api_url():
     return "https://platform.unstructuredapp.io"
 
 
-@pytest.fixture(scope="module")
-def client(platform_api_url) -> UnstructuredClient:
-    _client = UnstructuredClient(
-        api_key_auth=FAKE_API_KEY,
-        server_url=platform_api_url,
-        retry_config=RetryConfig(
-            strategy="backoff",
-            retry_connection_errors=False,
-            backoff=BackoffStrategy(
-                max_elapsed_time=0, max_interval=0, exponent=0, initial_interval=0
-            ),
-        ),
-    )
-    yield _client

gen.yaml

@@ -19,6 +19,7 @@ python:
       deepdiff: '>=6.0'
       pytest: '>=8.3.3'
       pytest-asyncio: '>=0.24.0'
+      pytest-freezgun: '>=0.4.2'
       pytest-mock: '>=3.14.0'
       types-aiofiles: '>=24.1.0'
       uvloop: '>=0.20.0'

pyproject.toml

@@ -38,6 +38,7 @@ mypy = "==1.14.1"
 pylint = "==3.2.3"
 pytest = ">=8.3.3"
 pytest-asyncio = ">=0.24.0"
+pytest-freezgun = ">=0.4.2"
 pytest-mock = ">=3.14.0"
 types-aiofiles = ">=24.1.0"
 types-python-dateutil = "^2.9.0.20240316"