From a2a912273e3820656bd0eade0728c443c5e155ff Mon Sep 17 00:00:00 2001 From: crayon <873217631@qq.com> Date: Thu, 19 Oct 2023 17:33:48 +0800 Subject: [PATCH] minor: unit test fix --- .../models/taskflow/test_get_node_detail.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/gcloud/tests/taskflow3/models/taskflow/test_get_node_detail.py b/gcloud/tests/taskflow3/models/taskflow/test_get_node_detail.py index ad25618b1a..9ef3e3d82b 100644 --- a/gcloud/tests/taskflow3/models/taskflow/test_get_node_detail.py +++ b/gcloud/tests/taskflow3/models/taskflow/test_get_node_detail.py @@ -66,7 +66,7 @@ def test_include_data_is_false(self): taskflow.has_node = MagicMock(return_value=True) dispatcher = MagicMock() get_node_data_return = {"result": True, "data": {}} - get_node_detail_return = {"result": True, "data": {}} + get_node_detail_return = {"result": True, "data": {"id": "id", "parent_id": "parent_id"}} dispatcher.get_node_data = MagicMock(return_value=get_node_data_return) dispatcher.get_node_detail = MagicMock(return_value=get_node_detail_return) dispatcher_init = MagicMock(return_value=dispatcher) @@ -98,7 +98,15 @@ def test_include_data_is_false(self): loop=loop, subprocess_simple_inputs=False, ) - self.assertEqual(detail, {"code": 0, "data": {}, "message": "", "result": True}) + self.assertEqual( + detail, + { + "code": 0, + "data": {"auto_retry_info": {}, "id": "id", "parent_id": "parent_id"}, + "message": "", + "result": True, + }, + ) def test_success(self): taskflow = TaskFlowInstance()