3434from st2common .persistence import execution as ex_db_access
3535from st2common .persistence import liveaction as lv_db_access
3636from st2common .persistence import workflow as wf_db_access
37- from st2common .runners import base as runners
3837from st2common .services import action as ac_svc
3938from st2common .services import workflows as wf_svc
4039from st2common .transport import liveaction as lv_ac_xport
7675 wf_ex_xport .WorkflowExecutionPublisher ,
7776 'publish_state' ,
7877 mock .MagicMock (side_effect = mock_wf_ex_xport .MockWorkflowExecutionPublisher .publish_state ))
79- class OrquestaErrorHandlingTest (st2tests .ExecutionDbTestCase ):
78+ class OrquestaErrorHandlingTest (st2tests .WorkflowTestCase ):
8079 ensure_indexes = True
8180 ensure_indexes_models = [
8281 WorkflowExecutionDB ,
@@ -100,13 +99,6 @@ def setUpClass(cls):
10099 for pack in PACKS :
101100 actions_registrar .register_from_pack (pack )
102101
103- @classmethod
104- def get_runner_class (cls , runner_name ):
105- return runners .get_runner (runner_name , runner_name ).__class__
106-
107- def sort_wf_runtime_errors (self , errors ):
108- return sorted (errors , key = lambda x : x .get ('task_id' , None ))
109-
110102 def test_fail_inspection (self ):
111103 expected_errors = [
112104 {
@@ -181,7 +173,7 @@ def test_fail_input_rendering(self):
181173 tk_ex_dbs = wf_db_access .TaskExecution .query (workflow_execution = str (wf_ex_db .id ))
182174 self .assertEqual (len (tk_ex_dbs ), 0 )
183175 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
184- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
176+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
185177
186178 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
187179 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -215,7 +207,7 @@ def test_fail_vars_rendering(self):
215207 tk_ex_dbs = wf_db_access .TaskExecution .query (workflow_execution = str (wf_ex_db .id ))
216208 self .assertEqual (len (tk_ex_dbs ), 0 )
217209 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
218- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
210+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
219211
220212 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
221213 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -251,7 +243,7 @@ def test_fail_start_task_action(self):
251243 tk_ex_dbs = wf_db_access .TaskExecution .query (workflow_execution = str (wf_ex_db .id ))
252244 self .assertEqual (len (tk_ex_dbs ), 0 )
253245 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
254- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
246+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
255247
256248 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
257249 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -288,7 +280,7 @@ def test_fail_start_task_input_expr_eval(self):
288280 tk_ex_dbs = wf_db_access .TaskExecution .query (workflow_execution = str (wf_ex_db .id ))
289281 self .assertEqual (len (tk_ex_dbs ), 0 )
290282 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
291- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
283+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
292284
293285 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
294286 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -326,7 +318,7 @@ def test_fail_start_task_input_value_type(self):
326318 # Assert workflow and task executions failed.
327319 wf_ex_db = wf_db_access .WorkflowExecution .query (action_execution = str (ac_ex_db .id ))[0 ]
328320 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
329- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
321+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
330322
331323 tk_ex_db = wf_db_access .TaskExecution .query (workflow_execution = str (wf_ex_db .id ))[0 ]
332324 self .assertEqual (tk_ex_db .status , wf_statuses .FAILED )
@@ -376,7 +368,7 @@ def test_fail_next_task_action(self):
376368 self .assertEqual (tk_ex_db .status , wf_statuses .SUCCEEDED )
377369 wf_ex_db = wf_db_access .WorkflowExecution .get_by_id (wf_ex_db .id )
378370 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
379- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
371+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
380372
381373 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
382374 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -422,7 +414,7 @@ def test_fail_next_task_input_expr_eval(self):
422414 self .assertEqual (tk_ex_db .status , wf_statuses .SUCCEEDED )
423415 wf_ex_db = wf_db_access .WorkflowExecution .get_by_id (wf_ex_db .id )
424416 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
425- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
417+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
426418
427419 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
428420 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -471,7 +463,7 @@ def test_fail_next_task_input_value_type(self):
471463 # Assert workflow execution and task2 execution failed.
472464 wf_ex_db = wf_db_access .WorkflowExecution .get_by_id (str (wf_ex_db .id ))
473465 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
474- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
466+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
475467
476468 tk2_ex_db = wf_db_access .TaskExecution .query (task_id = 'task2' )[0 ]
477469 self .assertEqual (tk2_ex_db .status , wf_statuses .FAILED )
@@ -518,7 +510,7 @@ def test_fail_task_execution(self):
518510 # Assert workflow state and result.
519511 wf_ex_db = wf_db_access .WorkflowExecution .get_by_id (str (wf_ex_db .id ))
520512 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
521- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
513+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
522514
523515 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
524516 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -563,7 +555,7 @@ def test_fail_task_transition(self):
563555 self .assertEqual (tk_ex_db .status , wf_statuses .SUCCEEDED )
564556 wf_ex_db = wf_db_access .WorkflowExecution .get_by_id (wf_ex_db .id )
565557 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
566- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
558+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
567559
568560 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
569561 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -609,7 +601,7 @@ def test_fail_task_publish(self):
609601 self .assertEqual (tk_ex_db .status , wf_statuses .SUCCEEDED )
610602 wf_ex_db = wf_db_access .WorkflowExecution .get_by_id (wf_ex_db .id )
611603 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
612- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
604+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
613605
614606 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
615607 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -652,7 +644,7 @@ def test_fail_output_rendering(self):
652644 self .assertEqual (tk_ex_db .status , wf_statuses .SUCCEEDED )
653645 wf_ex_db = wf_db_access .WorkflowExecution .get_by_id (wf_ex_db .id )
654646 self .assertEqual (wf_ex_db .status , wf_statuses .FAILED )
655- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
647+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
656648
657649 lv_ac_db = lv_db_access .LiveAction .get_by_id (str (lv_ac_db .id ))
658650 self .assertEqual (lv_ac_db .status , ac_const .LIVEACTION_STATUS_FAILED )
@@ -770,7 +762,7 @@ def test_fail_manually(self):
770762 }
771763 ]
772764
773- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
765+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
774766
775767 def test_fail_manually_with_recovery_failure (self ):
776768 wf_file = 'fail-manually-with-recovery-failure.yaml'
@@ -833,4 +825,4 @@ def test_fail_manually_with_recovery_failure(self):
833825 }
834826 ]
835827
836- self .assertListEqual (self .sort_wf_runtime_errors (wf_ex_db .errors ), expected_errors )
828+ self .assertListEqual (self .sort_workflow_errors (wf_ex_db .errors ), expected_errors )
0 commit comments