[fix] Preserve subgraph structure when clearing workflow (#4567)

Co-authored-by: github-actions <github-actions@github.com>

tests-ui/tests/utils/graphTraversalUtil.test.ts

@@ -7,6 +7,7 @@ import {
   findSubgraphByUuid,
   forEachNode,
   forEachSubgraphNode,
+  getAllNonIoNodesInSubgraph,
   getLocalNodeIdFromExecutionId,
   getNodeByExecutionId,
   getNodeByLocatorId,
@@ -757,5 +758,54 @@ describe('graphTraversalUtil', () => {
         ])
       })
     })
+
+    describe('getAllNonIoNodesInSubgraph', () => {
+      it('should filter out SubgraphInputNode and SubgraphOutputNode', () => {
+        const nodes = [
+          { id: 'input', constructor: { comfyClass: 'SubgraphInputNode' } },
+          { id: 'output', constructor: { comfyClass: 'SubgraphOutputNode' } },
+          { id: 'user1', constructor: { comfyClass: 'CLIPTextEncode' } },
+          { id: 'user2', constructor: { comfyClass: 'KSampler' } }
+        ] as LGraphNode[]
+
+        const subgraph = createMockSubgraph('sub-uuid', nodes)
+        const nonIoNodes = getAllNonIoNodesInSubgraph(subgraph)
+
+        expect(nonIoNodes).toHaveLength(2)
+        expect(nonIoNodes.map((n) => n.id)).toEqual(['user1', 'user2'])
+      })
+
+      it('should handle subgraph with only IO nodes', () => {
+        const nodes = [
+          { id: 'input', constructor: { comfyClass: 'SubgraphInputNode' } },
+          { id: 'output', constructor: { comfyClass: 'SubgraphOutputNode' } }
+        ] as LGraphNode[]
+
+        const subgraph = createMockSubgraph('sub-uuid', nodes)
+        const nonIoNodes = getAllNonIoNodesInSubgraph(subgraph)
+
+        expect(nonIoNodes).toHaveLength(0)
+      })
+
+      it('should handle subgraph with only user nodes', () => {
+        const nodes = [
+          { id: 'user1', constructor: { comfyClass: 'CLIPTextEncode' } },
+          { id: 'user2', constructor: { comfyClass: 'KSampler' } }
+        ] as LGraphNode[]
+
+        const subgraph = createMockSubgraph('sub-uuid', nodes)
+        const nonIoNodes = getAllNonIoNodesInSubgraph(subgraph)
+
+        expect(nonIoNodes).toHaveLength(2)
+        expect(nonIoNodes).toEqual(nodes)
+      })
+
+      it('should handle empty subgraph', () => {
+        const subgraph = createMockSubgraph('sub-uuid', [])
+        const nonIoNodes = getAllNonIoNodesInSubgraph(subgraph)
+
+        expect(nonIoNodes).toHaveLength(0)
+      })
+    })
   })
 })

tests-ui/tests/utils/typeGuardUtil.test.ts

@@ -0,0 +1,45 @@
+import { describe, expect, it } from 'vitest'
+
+import { isSubgraphIoNode } from '@/utils/typeGuardUtil'
+
+describe('typeGuardUtil', () => {
+  describe('isSubgraphIoNode', () => {
+    it('should identify SubgraphInputNode as IO node', () => {
+      const node = {
+        constructor: { comfyClass: 'SubgraphInputNode' }
+      } as any
+
+      expect(isSubgraphIoNode(node)).toBe(true)
+    })
+
+    it('should identify SubgraphOutputNode as IO node', () => {
+      const node = {
+        constructor: { comfyClass: 'SubgraphOutputNode' }
+      } as any
+
+      expect(isSubgraphIoNode(node)).toBe(true)
+    })
+
+    it('should not identify regular nodes as IO nodes', () => {
+      const node = {
+        constructor: { comfyClass: 'CLIPTextEncode' }
+      } as any
+
+      expect(isSubgraphIoNode(node)).toBe(false)
+    })
+
+    it('should handle nodes without constructor', () => {
+      const node = {} as any
+
+      expect(isSubgraphIoNode(node)).toBe(false)
+    })
+
+    it('should handle nodes without comfyClass', () => {
+      const node = {
+        constructor: {}
+      } as any
+
+      expect(isSubgraphIoNode(node)).toBe(false)
+    })
+  })
+})