Fix tests on query

Author: adamtheturtle

src/mock_vws/_mock_web_query_api.py

@@ -20,7 +20,7 @@
 from requests_mock.request import _RequestObjectProxy
 from requests_mock.response import _Context
 
-from mock_vws._constants import ResultCodes, TargetStatuses
+from mock_vws._constants import ResultCodes, States, TargetStatuses
 from mock_vws._mock_common import Route, json_dump, set_content_length_header
 from mock_vws._mock_web_services_api import MockVuforiaWebServicesAPI, Target
 
@@ -29,6 +29,46 @@
 ROUTES = set([])
 
 
+@wrapt.decorator
+def validate_project_state(
+    wrapped: Callable[..., str],
+    instance: Any,
+    args: Tuple[_RequestObjectProxy, _Context],
+    kwargs: Dict,
+) -> str:
+    """
+    Validate the state of the project.
+
+    Args:
+        wrapped: An endpoint function for `requests_mock`.
+        instance: The class that the endpoint function is in.
+        args: The arguments given to the endpoint function.
+        kwargs: The keyword arguments given to the endpoint function.
+
+    Returns:
+        The result of calling the endpoint.
+        A `FORBIDDEN` response with an InactiveProject result code if the
+        project is inactive.
+    """
+    _, context = args
+
+    if instance.mock_web_services_api.state != States.PROJECT_INACTIVE:
+        return wrapped(*args, **kwargs)
+
+    context.status_code = codes.FORBIDDEN
+    transaction_id = uuid.uuid4().hex
+    result_code = ResultCodes.INACTIVE_PROJECT.value
+
+    # The response has an unusual format of separators, so we construct it
+    # manually.
+    return (
+        '{"transaction_id": '
+        f'"{transaction_id}",'
+        f'"result_code":"{result_code}"'
+        '}'
+    )
+
+
 @wrapt.decorator
 def validate_image_format(
     wrapped: Callable[..., str],
@@ -593,6 +633,7 @@ def decorator(method: Callable[..., str]) -> Callable[..., str]:
             validate_content_type_header,
             validate_accept_header,
             validate_auth_header_exists,
+            validate_project_state,
             set_content_length_header,
         ]