Few changes

Shailesh Jagannath Padave · Shailesh Jagannath Padave · commit 5320b5a960cb · 2025-04-17T17:07:28.000+05:30
diff --git a/src/conductor/client/http/models/workflow_run.py b/src/conductor/client/http/models/workflow_run.py
@@ -3,6 +3,12 @@
 
 import six
 
+from conductor.client.http.models import Task
+
+terminal_status = ('COMPLETED', 'FAILED', 'TIMED_OUT', 'TERMINATED')
+successful_status = ('PAUSED', 'COMPLETED')
+running_status = ('RUNNING', 'PAUSED')
+
 class WorkflowRun(object):
     """NOTE: This class is auto generated by the swagger code generator program.
 
diff --git a/tests/unit/orkes/test_workflow_client.py b/tests/unit/orkes/test_workflow_client.py
@@ -86,7 +86,7 @@ def test_executeWorkflow(self, mock):
             startWorkflowReq, "request_id", None, 30
         )
         mock.assert_called_with(body=startWorkflowReq, request_id="request_id", name=WORKFLOW_NAME, version=1,
-                                wait_until_task_ref=None, wait_for_seconds=30)
+                                wait_until_task_ref=None, wait_for_seconds=30,  consistency='DURABLE', return_strategy='TARGET_WORKFLOW')
         self.assertEqual(workflowRun, expectedWfRun)
 
     @patch.object(WorkflowResourceApi, 'pause_workflow')

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,7 @@ def test_executeWorkflow(self, mock):`
`86`	`86`	`startWorkflowReq, "request_id", None, 30`
`87`	`87`	`)`
`88`	`88`	`mock.assert_called_with(body=startWorkflowReq, request_id="request_id", name=WORKFLOW_NAME, version=1,`
`89`		`- wait_until_task_ref=None, wait_for_seconds=30)`
	`89`	`+ wait_until_task_ref=None, wait_for_seconds=30, consistency='DURABLE', return_strategy='TARGET_WORKFLOW')`
`90`	`90`	`self.assertEqual(workflowRun, expectedWfRun)`
`91`	`91`
`92`	`92`	`@patch.object(WorkflowResourceApi, 'pause_workflow')`