fix: Fix handling of get_running_workflow_step() response

Author: Alan Christie

tests/test_test_api_adapter.py

@@ -240,7 +240,7 @@ def test_get_running_workflow_step():
     # Assert
     assert response["name"] == "step-1"
     assert not response["done"]
-    assert response["running_workflow"] == rwfid
+    assert response["running_workflow"]["id"] == rwfid
 
 
 def test_create_instance():

tests/wapi_adapter.py

@@ -137,7 +137,7 @@ def create_running_workflow_step(
             "name": step,
             "done": False,
             "success": False,
-            "running_workflow": running_workflow_id,
+            "running_workflow": {"id": running_workflow_id},
         }
         running_workflow_step[running_workflow_step_id] = record
 
@@ -300,7 +300,7 @@ def get_running_workflow_steps(self, *, running_workflow_id: str) -> dict[str, A
 
         steps = []
         for key, value in running_workflow_step.items():
-            if value["running_workflow"] == running_workflow_id:
+            if value["running_workflow"]["id"] == running_workflow_id:
                 item = value
                 item["id"] = key
                 steps.append(item)

workflow/workflow_engine.py

@@ -226,7 +226,7 @@ def _handle_pod_message(self, msg: PodMessage) -> None:
         step_name: str = rwfs_response["name"]
 
         # Get the step's running workflow record.
-        r_wfid: str = rwfs_response["running_workflow"]
+        r_wfid: str = rwfs_response["running_workflow"]["id"]
         assert r_wfid
         rwf_response, _ = self._wapi_adapter.get_running_workflow(
             running_workflow_id=r_wfid