Merge pull request #892 from adamtheturtle/custom-wait-between-requests

Custom wait between requests

Author: adamtheturtle

tests/test_vws.py

@@ -279,6 +279,48 @@ def test_wait_for_target_processed(
         report = client.get_target_summary_report(target_id=target_id)
         assert report['status'] != 'processing'
 
+    def test_default_seconds_between_requests(
+        self,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        By default, 0.2 seconds are waited between polling requests.
+        """
+        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(),
+            )
+
+            target_id = client.add_target(
+                name='x',
+                width=1,
+                image=high_quality_image,
+            )
+
+            client.wait_for_target_processed(target_id=target_id)
+            report = client.get_database_summary_report()
+            expected_requests = (
+                # Add target request
+                1 +
+                # Database summary request
+                1 +
+                # Initial request
+                1 +
+                # Request after 0.2 seconds - not processed
+                1 +
+                # Request after 0.4 seconds - not processed
+                # This assumes that there is less than 0.1 seconds taken
+                # between the start of the target processing and the start of
+                # waiting for the target to be processed.
+                1 +
+                # Request after 0.6 seconds - processed
+                1
+            )
+            assert report['request_usage'] == expected_requests
+
 
 class TestGetDuplicateTargets:
     """