minor: unit test fix

TencentBlueKing · Oct 19, 2023 · d354194 · d354194
1 parent bf4472f
commit d354194
Showing 1 changed file with 8 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
@@ -107,7 +107,7 @@ def test_success(self):
         taskflow.has_node = MagicMock(return_value=True)
         dispatcher = MagicMock()
         get_node_data_return = {"result": True, "data": {"data": "data"}}
-        get_node_detail_return = {"result": True, "data": {"detail": "detail"}}
+        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)
@@ -149,5 +149,11 @@ def test_success(self):
             subprocess_simple_inputs=False,
         )
         self.assertEqual(
-            detail, {"code": 0, "data": {"data": "data", "detail": "detail"}, "message": "", "result": True}
+            detail,
+            {
+                "code": 0,
+                "data": {"data": "data", "id": "id", "parent_id": "parent_id", "auto_retry_info": {}},
+                "message": "",
+                "result": True,
+            },
         )