fix: ut test

apitable · Feb 6, 2024 · a2ff66b · a2ff66b
1 parent ea0262a
commit a2ff66b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/apitable.py/apitable/types/node.py b/apitable.py/apitable/types/node.py
@@ -10,6 +10,7 @@ class NodeTypeEnum(str, Enum):
     Form = 'Form'
     Dashboard = 'Dashboard'
     Mirror = 'Mirror'
+    Automation = 'Automation'
     ERROR = "ERROR NODE TYPE"
 
 

diff --git a/apitable.py/apitable/types/response.py b/apitable.py/apitable/types/response.py
@@ -104,9 +104,11 @@ class GETNodeListResponseData(BaseModel):
 class GETNodeListResponse(ResponseBase):
     data: GETNodeListResponseData
 
+
 class GETSearchNodeListResponseData(BaseModel):
     nodes: List[NodeSearchInfo]
 
+
 class GETSearchNodeListResponse(ResponseBase):
     data: GETSearchNodeListResponseData
 

diff --git a/apitable.py/test/test_get_nodes.py b/apitable.py/test/test_get_nodes.py
@@ -17,7 +17,7 @@ def test_space_root_nodes_all(self):
         space_root_nodes = self.apitable.space(SPACE_ID).nodes.all()
         self.assertIsInstance(space_root_nodes, list)
         first_node = space_root_nodes[0]
-        self.assertIn(first_node.type, ["Datasheet", "Folder", "Mirror", "Form", "ERROR NODE TYPE"])
+        self.assertIn(first_node.type, ["Datasheet", "Folder", "Mirror", "Form", "ERROR NODE TYPE", "Automation"])
 
     def test_get_node_detail(self):
         node = self.apitable.nodes.get(FOLDER_ID)