minor: 修复单元测试

TencentBlueKing · Oct 18, 2023 · 9746c08 · 9746c08
1 parent 4a62eac
commit 9746c08
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/engine/test_engine_schedule.py b/tests/engine/test_engine_schedule.py
@@ -469,7 +469,7 @@ def test_schedule__schedule_done(node_id, state, pi, schedule, node, interrupter
     runtime.get_node.assert_called_once_with(node_id)
     runtime.get_execution_data.assert_called_once_with(node_id)
     runtime.set_execution_data.assert_called_once_with(node_id=node.id, data=execution_data)
-    runtime.get_data_inputs.assert_called_once_with(pi.root_pipeline_id)
+    runtime.get_data_inputs.assert_has_calls([call(node_id), call(node_id)])
     runtime.get_callback_data.assert_not_called()
     service.hook_dispatch.assert_called_once()
     handler.schedule.assert_called_once_with(
@@ -537,7 +537,8 @@ def test_schedule__schedule_done_with_node_reserved_rollback(node_id, state, pi,
 
     runtime.get_schedule.assert_called_once_with(schedule.id)
     runtime.node_finish.assert_called_once_with(pi.root_pipeline_id, node.id)
-    runtime.get_state.assert_called_once_with(node_id)
+
+    runtime.get_state.assert_has_calls([call(node_id), call(node_id)])
     runtime.get_node.assert_called_once_with(node_id)
     runtime.get_callback_data.assert_not_called()
     handler.schedule.assert_called_once_with(