Passing test for custom query deletion recognition

Author: adamtheturtle

src/mock_vws/__init__.py

@@ -37,6 +37,7 @@ def __init__(  # pylint: disable=too-many-arguments
         client_secret_key: Optional[str]=None,
         database_name: Optional[str]=None,
         processing_time_seconds: Union[int, float]=0.5,
+        query_recognizes_deletion_seconds: Union[int, float]=3,
     ) -> None:
         """
         Args:
@@ -95,6 +96,10 @@ def __init__(  # pylint: disable=too-many-arguments
         self._base_vws_url = base_vws_url
         self._base_vwq_url = base_vwq_url
 
+        self._query_recognizes_deletion_seconds = (
+            query_recognizes_deletion_seconds
+        )
+
     def __enter__(self) -> 'MockVWS':
         """
         Start an instance of a Vuforia mock with access keys set from
@@ -115,8 +120,9 @@ def __enter__(self) -> 'MockVWS':
             client_access_key=self.client_access_key,
             client_secret_key=self.client_secret_key,
             mock_web_services_api=mock_vws_api,
-            #  TODO change here
-            query_recognizes_deletion_seconds=3,
+            query_recognizes_deletion_seconds=(
+                self._query_recognizes_deletion_seconds
+            )
         )
 
         date = email.utils.formatdate(None, localtime=False, usegmt=True)

tests/mock_vws/test_usage.py

@@ -499,7 +499,6 @@ def test_default(
         XXX
         """
         with MockVWS(
-            real_http=False,
             client_access_key=vuforia_database_keys.client_access_key.decode(),
             client_secret_key=vuforia_database_keys.client_secret_key.decode(),
             server_access_key=vuforia_database_keys.server_access_key.decode(),
@@ -512,3 +511,27 @@ def test_default(
 
         expected = 3
         assert abs(expected - seconds_to_recognize_deletion) < 0.2
+
+    def test_custom(
+        self,
+        high_quality_image: io.BytesIO,
+        vuforia_database_keys: VuforiaDatabaseKeys,
+    ) -> None:
+        """
+        XXX
+        """
+        query_recognizes_deletion = 0.5
+        with MockVWS(
+            client_access_key=vuforia_database_keys.client_access_key.decode(),
+            client_secret_key=vuforia_database_keys.client_secret_key.decode(),
+            server_access_key=vuforia_database_keys.server_access_key.decode(),
+            server_secret_key=vuforia_database_keys.server_secret_key.decode(),
+            query_recognizes_deletion_seconds=query_recognizes_deletion,
+        ) as mock:
+            seconds_to_recognize_deletion = self._seconds_to_recognize_deletion(
+                high_quality_image=high_quality_image,
+                vuforia_database_keys=vuforia_database_keys,
+            )
+
+        expected = query_recognizes_deletion
+        assert abs(expected - seconds_to_recognize_deletion) < 0.2