Bump mock-vws and use sleep_fn in timeout tests

Author: adamtheturtle

pyproject.toml

@@ -77,7 +77,7 @@ optional-dependencies.dev = [
     "ty==0.0.16",
     "types-requests==2.32.4.20260107",
     "vulture==2.14",
-    "vws-python-mock==2026.2.15",
+    "vws-python-mock==2026.2.15.4",
     "vws-test-fixtures==2023.3.5",
     "yamlfix==1.19.1",
     "zizmor==1.22.0",

tests/test_query.py

@@ -1,5 +1,6 @@
 """Tests for the ``CloudRecoService`` querying functionality."""
 
+import datetime
 import io
 import uuid
 from typing import BinaryIO
@@ -63,8 +64,13 @@ def test_default_timeout(
         timeout.
         """
         with (
-            freeze_time(auto_tick_seconds=1),
-            MockVWS(response_delay_seconds=response_delay_seconds) as mock,
+            freeze_time() as frozen_datetime,
+            MockVWS(
+                response_delay_seconds=response_delay_seconds,
+                sleep_fn=lambda seconds: frozen_datetime.tick(
+                    delta=datetime.timedelta(seconds=seconds),
+                ),
+            ) as mock,
         ):
             database = VuforiaDatabase()
             mock.add_database(database=database)
@@ -82,40 +88,6 @@ def test_default_timeout(
                 matches = cloud_reco_client.query(image=image)
                 assert not matches
 
-    @staticmethod
-    @pytest.mark.parametrize(
-        argnames="custom_timeout",
-        argvalues=[60.5, 60.0, (5.0, 30.0)],
-    )
-    def test_custom_timeout(
-        image: io.BytesIO | BinaryIO,
-        custom_timeout: float | tuple[float, float],
-    ) -> None:
-        """It is possible to set a custom request timeout."""
-        with MockVWS() as mock:
-            database = VuforiaDatabase()
-            mock.add_database(database=database)
-            vws_client = VWS(
-                server_access_key=database.server_access_key,
-                server_secret_key=database.server_secret_key,
-            )
-            cloud_reco_client = CloudRecoService(
-                client_access_key=database.client_access_key,
-                client_secret_key=database.client_secret_key,
-                request_timeout_seconds=custom_timeout,
-            )
-
-            target_id = vws_client.add_target(
-                name="x",
-                width=1,
-                image=image,
-                active_flag=True,
-                application_metadata=None,
-            )
-            vws_client.wait_for_target_processed(target_id=target_id)
-            matches = cloud_reco_client.query(image=image)
-            assert len(matches) == 1
-
     @staticmethod
     def test_timeout_raises_on_slow_response(
         image: io.BytesIO | BinaryIO,

tests/test_vws.py

@@ -111,8 +111,13 @@ def test_default_timeout(
         timeout.
         """
         with (
-            freeze_time(auto_tick_seconds=1),
-            MockVWS(response_delay_seconds=response_delay_seconds) as mock,
+            freeze_time() as frozen_datetime,
+            MockVWS(
+                response_delay_seconds=response_delay_seconds,
+                sleep_fn=lambda seconds: frozen_datetime.tick(
+                    delta=datetime.timedelta(seconds=seconds),
+                ),
+            ) as mock,
         ):
             database = VuforiaDatabase()
             mock.add_database(database=database)
@@ -141,33 +146,6 @@ def test_default_timeout(
                     application_metadata=None,
                 )
 
-    @staticmethod
-    @pytest.mark.parametrize(
-        argnames="custom_timeout",
-        argvalues=[60.5, 60.0, (5.0, 30.0)],
-    )
-    def test_custom_timeout(
-        image: io.BytesIO | BinaryIO,
-        custom_timeout: float | tuple[float, float],
-    ) -> None:
-        """It is possible to set a custom request timeout."""
-        with MockVWS() as mock:
-            database = VuforiaDatabase()
-            mock.add_database(database=database)
-            vws_client = VWS(
-                server_access_key=database.server_access_key,
-                server_secret_key=database.server_secret_key,
-                request_timeout_seconds=custom_timeout,
-            )
-
-            vws_client.add_target(
-                name="x",
-                width=1,
-                image=image,
-                active_flag=True,
-                application_metadata=None,
-            )
-
     @staticmethod
     def test_timeout_raises_on_slow_response(
         image: io.BytesIO | BinaryIO,