Get tests to pass on 3.11

sartography · Oct 9, 2024 · 379282c · 379282c
1 parent 9485883
commit 379282c
Showing 1 changed file with 12 additions and 13 deletions.
diff --git a/tests/mock_connectors/connector-example/src/connector_example/commands/combine_strings.py b/tests/mock_connectors/connector-example/src/connector_example/commands/combine_strings.py
@@ -1,7 +1,7 @@
 """Simple Example Command."""
 from typing import Any
 
-from spiffworkflow_connector_command.command_interface import CommandResultDictV2
+from spiffworkflow_connector_command.command_interface import CommandResponseDict
 from spiffworkflow_connector_command.command_interface import ConnectorCommand
 from spiffworkflow_connector_command.command_interface import ConnectorProxyResponseDict
 
@@ -20,19 +20,18 @@ def __init__(
         self.arg1 = arg1
         self.arg2 = arg2
 
-    def execute(self, config: Any, task_data: Any) -> CommandResultDictV2:
+    def execute(self, config: Any, task_data: Any) -> CommandResponseDict:
         """Execute."""
-
-        return_response: ConnectorProxyResponseDict = {
-            "command_response": {"example_response": "whatever you want", "arg1": self.arg1, "arg2": self.arg2},
-            "spiff__logs": [],
-            "error": None,
-        }
-        result: CommandResultDictV2 = {
-            "response": return_response,
-            "status": 200,
+
+        return {
+            "body": {
+                "command_response": {
+                    "example_response": "whatever you want",
+                    "arg1": self.arg1,
+                    "arg2": self.arg2
+                },
+            },
+            "http_status": 200,
             "mimetype": "application/json",
         }
 
-        return result
-