Passing test on real

Author: adamtheturtle

tests/mock_vws/test_database_summary.py

@@ -381,17 +381,18 @@ def test_quotas(self, vuforia_database_keys: VuforiaDatabaseKeys) -> None:
 
 
 @pytest.mark.usefixtures('verify_mock_vuforia')
-class TestUsageMetrics:
+class TestRequestUsage:
     """
-    Tests for usage metrics.
+    Tests for the ``request_usage`` field.
     """
 
-    def test_request_usage(
+    def test_target_request(
         self,
         vuforia_database_keys: VuforiaDatabaseKeys,
     ) -> None:
         """
-        The ``request_usage`` count increases with each request.
+        The ``request_usage`` count increases with each request to the target
+        API.
         """
         response = database_summary(
             vuforia_database_keys=vuforia_database_keys,
@@ -406,17 +407,65 @@ def test_request_usage(
         new_request_usage = response.json()['request_usage']
         assert new_request_usage == original_request_usage + 1
 
-        query(
+    def test_bad_target_request(
+        self,
+        vuforia_database_keys: VuforiaDatabaseKeys,
+    ) -> None:
+        """
+        The ``request_usage`` count increases with each request to the target
+        API, even if it is a bad request.
+        """
+        response = database_summary(
             vuforia_database_keys=vuforia_database_keys,
-            body={},
         )
 
+        original_request_usage = response.json()['request_usage']
+
+        response = add_target_to_vws(
+            vuforia_database_keys=vuforia_database_keys,
+            # Missing data.
+            data={},
+        )
+
+        assert response.status_code == codes.BAD_REQUEST
+
         response = database_summary(
             vuforia_database_keys=vuforia_database_keys,
         )
         new_request_usage = response.json()['request_usage']
         assert new_request_usage == original_request_usage + 2
 
+    def test_query_request(
+        self,
+        vuforia_database_keys: VuforiaDatabaseKeys,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        The ``request_usage`` count does not increase with each query.
+        """
+        response = database_summary(
+            vuforia_database_keys=vuforia_database_keys,
+        )
+
+        original_request_usage = response.json()['request_usage']
+
+        image_content = high_quality_image.getvalue()
+        body = {'image': ('image.jpeg', image_content, 'image/jpeg')}
+        query_resp = query(
+            vuforia_database_keys=vuforia_database_keys,
+            body=body,
+        )
+
+        assert query_resp.status_code == codes.OK
+
+        response = database_summary(
+            vuforia_database_keys=vuforia_database_keys,
+        )
+        new_request_usage = response.json()['request_usage']
+        # The request usage goes up for the database summary request, not the
+        # query.
+        assert new_request_usage == original_request_usage + 1
+
 
 @pytest.mark.usefixtures('verify_mock_vuforia_inactive')
 class TestInactiveProject: