fix(ui): fix collapse/expand suite with file name search

packages/ui/client/composables/explorer/expand.ts

@@ -130,6 +130,18 @@ function expandAllNodes(nodes: UITaskTreeNode[], updateState: boolean) {
   }
 }
 
+/**
+ * Build the complete next explorer entry list by emitting an expanded node and its filtered
+ * subtree at the node's current position.
+ *
+ * `children` contains only entries produced by filtering the expanded subtree. This function
+ * walks the complete current `uiEntries`. When it reaches `node`, it emits the node unless
+ * `children` already contains it, followed by the children. Unrelated entries are copied
+ * unchanged, while existing entries with the same IDs are skipped to avoid duplicates.
+ *
+ * TODO: Make this a pure splice over explicit entries and keep expansion state changes in
+ * `runExpandNode`.
+ */
 function* collectExpandedNode(
   node: UITaskTreeNode,
   children: Set<UITaskTreeNode>,

packages/ui/client/composables/explorer/filter.ts

@@ -51,11 +51,13 @@ export function* filterNode(
   search: SearchMatcher,
   filter: Filter,
 ) {
+  // Parent node IDs that match or contain a matching descendant.
   const treeNodes = new Set<string>()
-
+  // Direct match state of visited parents, used to include their immediate children.
   const parentsMap = new Map<string, boolean>()
+  // Visited nodes with their effective match state, consumed bottom-up by filterParents.
   const list: FilterResult[] = []
-
+  // Matching file ID, or owning file ID when filtering an expanded subtree.
   let fileId: string | undefined
 
   if (filter.onlyTests) {
@@ -92,9 +94,14 @@ export function* filterNode(
     // when expanding a non-file node
     if (!fileId && !isFileNode(node) && 'fileId' in node) {
       fileId = node.fileId as string
+      const file = explorerTree.nodes.get(fileId)
+      if (file && matcher(file, search, filter)) {
+        treeNodes.add(fileId)
+      }
     }
   }
 
+  // TODO: Let filterParents own this traversal-local state.
   const filesToShow = new Set<string>()
 
   const entries = [...filterParents(

packages/ui/client/composables/explorer/state.ts

@@ -9,8 +9,11 @@ import { createTagsFilter } from '../../../../vitest/src/runtime/runner/utils/ta
 import { config } from '../client'
 import { explorerTree } from './index'
 
+/** All root file nodes, independent of the current filter and expansion state. */
 export const uiFiles = shallowRef<FileTreeNode[]>([])
+/** Flattened rows currently rendered by the explorer's virtual scroller. */
 export const uiEntries = shallowRef<UITaskTreeNode[]>([])
+/** Persisted node IDs used to restore individually expanded branches. */
 export const openedTreeItems = useLocalStorage<string[]>(
   'vitest-ui_task-tree-opened',
   [],

test/ui/test/ui.spec.ts

@@ -501,6 +501,10 @@ async function testError(page: Page) {
 async function testSuiteReport(page: Page) {
   const report = page.getByTestId('report')
 
+  await expect(getExplorerItem(page, 'sample.test.ts')).toBeVisible()
+  await page.getByPlaceholder('Search...').fill('suite-report')
+  await expect(getExplorerItem(page, 'sample.test.ts')).toHaveCount(0)
+
   await getExplorerItem(page, 'suite-report.test.ts').click()
   await expect(report).toContainText('before-all-marker')
   await expect(report).toContainText('direct-child-marker')