Merge pull request #636 from adamtheturtle/test-query-success-false

Add test for querying a target with success: failed

Author: adamtheturtle

src/mock_vws/_mock_web_query_api.py

@@ -602,7 +602,10 @@ def query(
                     content_type = 'text/html; charset=ISO-8859-1'
                     context.headers['Content-Type'] = content_type
                     return Path(match_processing_resp_file).read_text()
-                if target.active_flag and not target.delete_date:
+                if (
+                    target.active_flag and not target.delete_date
+                    and target.status == TargetStatuses.SUCCESS.value
+                ):
                     matches.add(target)
 
         results: List[Dict[str, Any]] = []

tests/mock_vws/test_query.py

@@ -1511,3 +1511,47 @@ def test_deleted_inactive(
 
         assert_query_success(response=response)
         assert response.json()['results'] == []
+
+
+@pytest.mark.usefixtures('verify_mock_vuforia')
+class TestTargetStatusFailed:
+    """
+    Tests for targets with the status "failed".
+    """
+
+    def test_status_failed(
+        self,
+        png_rgb: io.BytesIO,
+        vuforia_database_keys: VuforiaDatabaseKeys,
+    ) -> None:
+        """
+        Targets with the status "failed" are not found in query results.
+        """
+        image_content = png_rgb.getvalue()
+        image_data_encoded = base64.b64encode(image_content).decode('ascii')
+        add_target_data = {
+            'name': 'example_name',
+            'width': 1,
+            'image': image_data_encoded,
+        }
+        response = add_target_to_vws(
+            vuforia_database_keys=vuforia_database_keys,
+            data=add_target_data,
+        )
+
+        target_id = response.json()['target_id']
+
+        wait_for_target_processed(
+            target_id=target_id,
+            vuforia_database_keys=vuforia_database_keys,
+        )
+
+        body = {'image': ('image.jpeg', image_content, 'image/jpeg')}
+
+        response = query(
+            vuforia_database_keys=vuforia_database_keys,
+            body=body,
+        )
+
+        assert_query_success(response=response)
+        assert response.json()['results'] == []