Start of test

Author: adamtheturtle

src/vws/exceptions.py

@@ -201,3 +201,9 @@ def response(self) -> Response:
         The response returned by Vuforia which included this error.
         """
         return self._response
+
+
+class TargetProcessingTimeout(Exception):
+    """
+    Exception raised when waiting for a target to be processed times out.
+    """

tests/test_vws.py

@@ -279,6 +279,43 @@ def test_wait_for_target_processed(
         report = client.get_target_summary_report(target_id=target_id)
         assert report['status'] != 'processing'
 
+    def test_custom_timeout(
+        self,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        It is possible to set a maximum timeout.
+        """
+        with MockVWS(processing_time_seconds=0.5) as mock:
+            database = VuforiaDatabase()
+            mock.add_database(database=database)
+            client = VWS(
+                server_access_key=database.server_access_key.decode(),
+                server_secret_key=database.server_secret_key.decode(),
+                base_vws_url=base_vws_url,
+            )
+
+            target_id = client.add_target(
+                name='x',
+                width=1,
+                image=high_quality_image,
+            )
+
+        report = client.get_target_summary_report(target_id=target_id)
+        assert report['status'] == 'processing'
+        with pytest.raises(TimeoutError):
+            client.wait_for_target_processed(
+                target_id=target_id,
+                timeout_seconds=0.1,
+            )
+
+        client.wait_for_target_processed(
+            target_id=target_id,
+            timeout_seconds=0.4,
+        )
+        report = client.get_target_summary_report(target_id=target_id)
+        assert report['status'] != 'processing'
+
 
 class TestGetDuplicateTargets:
     """