def test_success(self):
taskflow = TaskFlowInstance()
taskflow.id = 1
taskflow.engine_ver = 2
taskflow.has_node = MagicMock(return_value=True)
dispatcher = MagicMock()
dispatch_return = {"result": True, "data": {"data": "data"}}
dispatcher.dispatch = MagicMock(return_value=dispatch_return)
dispatcher_init = MagicMock(return_value=dispatcher)
node_id = "node_id"
username = "******"
inputs = "inputs"
with patch(TASKFLOW_MODEL_NODE_CMD_DISPATCHER, dispatcher_init):
detail = taskflow.spec_nodes_timer_reset(node_id=node_id,
username=username,
inputs=inputs)
dispatcher_init.assert_called_once_with(engine_ver=taskflow.engine_ver,
node_id=node_id,
taskflow_id=1)
dispatcher.dispatch.assert_has_calls([
call(command="forced_fail", operator=username),
call(command="retry", operator=username, inputs="inputs")
])
self.assertEqual(detail, dispatch_return)
def test_node_does_not_exist(self):
taskflow = TaskFlowInstance()
taskflow.id = 1
taskflow.has_node = MagicMock(return_value=False)
detail = taskflow.spec_nodes_timer_reset(node_id="node_id",
username="******",
inputs="inputs")
self.assertFalse(detail["result"])
self.assertEqual(detail["code"], err_code.REQUEST_PARAM_INVALID.code)