def test_success(self):
taskflow = TaskFlowInstance()
taskflow.pk = 1
old_pipeline = PipelineInstance()
taskflow.pipeline_instance = old_pipeline
taskflow.pipeline_instance.clone = MagicMock(
return_value=PipelineInstance())
username = "******"
with patch(TASKINSTANCE_SAVE, MagicMock()):
task = taskflow.clone(username)
self.assertIsNone(task.id)
self.assertFalse(task.is_deleted)
taskflow.save.assert_called_once()
old_pipeline.clone.assert_called_once_with(username)
def test_success(self):
taskflow = TaskFlowInstance()
taskflow.id = 1
taskflow.engine_ver = 2
taskflow.has_node = MagicMock(return_value=True)
taskflow.project_id = 3
taskflow.pipeline_instance = PipelineInstance()
dispatcher = MagicMock()
get_node_data_return = {"code": 0, "data": {"data": "data"}, "message": "", "result": True}
dispatcher.get_node_data = MagicMock(return_value=get_node_data_return)
dispatcher_init = MagicMock(return_value=dispatcher)
node_id = "node_id"
username = "******"
component_code = "component_code"
subprocess_stack = ["1"]
loop = 1
with patch(TASKFLOW_MODEL_NODE_CMD_DISPATCHER, dispatcher_init):
data = taskflow.get_node_data(
node_id=node_id,
username=username,
component_code=component_code,
subprocess_stack=subprocess_stack,
loop=loop,
)
dispatcher_init.assert_called_once_with(engine_ver=taskflow.engine_ver, node_id=node_id, taskflow_id=1)
dispatcher.get_node_data.assert_called_once_with(
username=username,
component_code=component_code,
loop=loop,
pipeline_instance=taskflow.pipeline_instance,
subprocess_stack=subprocess_stack,
project_id=taskflow.project_id,
)
self.assertEqual(data, get_node_data_return)
