Fixing assertion

ethyca · Jul 12, 2024 · f27ed09 · f27ed09
1 parent 0a29f9a
commit f27ed09
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/tests/ops/service/connectors/test_saas_connector.py b/tests/ops/service/connectors/test_saas_connector.py
@@ -11,6 +11,7 @@
 
 from fides.api.common_exceptions import (
     AwaitingAsyncTaskCallback,
+    FidesopsException,
     SkippingConsentPropagation,
 )
 from fides.api.graph.execution import ExecutionNode
@@ -548,13 +549,15 @@ def test_request_with_invalid_output_template(
         execution_node = ExecutionNode(request_task)
         connector: SaaSConnector = get_connector(saas_example_connection_config)
 
-        assert connector.retrieve_data(
-            execution_node,
-            Policy(),
-            PrivacyRequest(id="123"),
-            request_task,
-            {"email": ["[email protected]"]},
-        ) == [{"id": "123", "email": "[email protected]"}]
+        with pytest.raises(FidesopsException) as exc:
+            assert connector.retrieve_data(
+                execution_node,
+                Policy(),
+                PrivacyRequest(id="123"),
+                request_task,
+                {"email": ["[email protected]"]},
+            ) == [{"id": "123", "email": "[email protected]"}]
+        assert "Failed to parse value as JSON" in str(exc)
 
 
 @pytest.mark.integration_saas