minor: unit test fix

TencentBlueKing · Oct 19, 2023 · a2a9122 · a2a9122
1 parent 344df90
commit a2a9122
Showing 1 changed file with 10 additions and 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
@@ -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()