Fix lint

Author: adamtheturtle

src/mock_vws/__init__.py

@@ -57,6 +57,9 @@ def __init__(  # pylint: disable=too-many-arguments
                 deterministic.
             base_vwq_url: The base URL for the VWQ API.
             base_vws_url: The base URL for the VWS API.
+            query_recognizes_deletion_seconds: The number of seconds after a
+                target has been deleted that the query endpoint will return a
+                500 response for on a match.
 
 
         Attributes:
@@ -122,7 +125,7 @@ def __enter__(self) -> 'MockVWS':
             mock_web_services_api=mock_vws_api,
             query_recognizes_deletion_seconds=(
                 self._query_recognizes_deletion_seconds
-            )
+            ),
         )
 
         date = email.utils.formatdate(None, localtime=False, usegmt=True)

src/mock_vws/_mock_web_query_api.py

@@ -10,7 +10,7 @@
 import io
 import uuid
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Set, Tuple
+from typing import Any, Callable, Dict, List, Set, Tuple, Union
 
 import pytz
 import wrapt
@@ -449,7 +449,7 @@ def __init__(
         client_access_key: str,
         client_secret_key: str,
         mock_web_services_api: MockVuforiaWebServicesAPI,
-        query_recognizes_deletion_seconds: int,
+        query_recognizes_deletion_seconds: Union[int, float],
     ) -> None:
         """
         Args:

tests/mock_vws/test_usage.py

@@ -433,11 +433,11 @@ class TestCustomQueryRecognizesDeletionSeconds:
     until it is not recognized by the query endpoint.
     """
 
-    def _seconds_to_recognize_deletion(
+    def _recognize_deletion_seconds(
         self,
         high_quality_image: io.BytesIO,
         vuforia_database_keys: VuforiaDatabaseKeys,
-    ) -> datetime.timedelta:
+    ) -> float:
         """
         XXX
         """
@@ -496,43 +496,48 @@ def test_default(
         vuforia_database_keys: VuforiaDatabaseKeys,
     ) -> None:
         """
-        XXX
+        By default it takes three seconds for the Query API on the mock to
+        recognize that a target has been deleted.
+
+        The real Query API takes between seven and thirty seconds.
+        See ``test_query`` for more information.
         """
         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(),
-        ) as mock:
-            seconds_to_recognize_deletion = self._seconds_to_recognize_deletion(
+        ):
+            recognize_deletion_seconds = self._recognize_deletion_seconds(
                 high_quality_image=high_quality_image,
                 vuforia_database_keys=vuforia_database_keys,
             )
 
         expected = 3
-        assert abs(expected - seconds_to_recognize_deletion) < 0.2
+        assert abs(expected - recognize_deletion_seconds) < 0.2
 
     def test_custom(
         self,
         high_quality_image: io.BytesIO,
         vuforia_database_keys: VuforiaDatabaseKeys,
     ) -> None:
         """
-        XXX
+        It is possible to use set a custom amount of time that it takes for the
+        Query API on the mock to recognize that a target has been deleted.
         """
-        # We choose a low time for a quick test
+        # We choose a low time for a quick test.
         query_recognizes_deletion = 0.1
         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(
+        ):
+            recognize_deletion_seconds = self._recognize_deletion_seconds(
                 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
+        assert abs(expected - recognize_deletion_seconds) < 0.2

tests/mock_vws/utils/__init__.py

@@ -408,7 +408,6 @@ def query(
         'Content-Type': content_type_header,
     }
 
-
     vwq_host = 'https://cloudreco.vuforia.com'
     response = requests.request(
         method=method,