feat(cli): show test "path" when filtering

packages/vitest/src/node/stdin.ts

@@ -3,9 +3,12 @@ import type { Writable } from 'node:stream'
 import c from 'tinyrainbow'
 import prompt from 'prompts'
 import { relative, resolve } from 'pathe'
-import { getTests, isWindows, stdout } from '../utils'
+import type { File, Task } from '@vitest/runner'
+import { isAtomTest } from '@vitest/runner/utils'
+import { isWindows, stdout } from '../utils'
 import { toArray } from '../utils/base'
 import type { Vitest } from './core'
+import type { FilterType } from './watch-filter'
 import { WatchFilter } from './watch-filter'
 
 const keys = [
@@ -36,6 +39,38 @@ ${keys
   )
 }
 
+function* traverseFilteredTestNames(parentName: string, filter: RegExp, t: Task): Generator<FilterType> {
+  if (isAtomTest(t)) {
+    if (t.name.match(filter)) {
+      const displayName = `${parentName} > ${t.name}`
+      yield { name: t.name, toString: () => displayName }
+    }
+  }
+  else {
+    parentName = parentName.length ? `${parentName} > ${t.name}` : t.name
+    for (const task of t.tasks) {
+      yield * traverseFilteredTestNames(parentName, filter, task)
+    }
+  }
+}
+
+function* getFilteredTestNames(pattern: string, suite: File[]): Generator<FilterType> {
+  try {
+    const reg = new RegExp(pattern)
+    // TODO: we cannot run tests per workspace yet: filtering files
+    const files = new Set<string>()
+    for (const file of suite) {
+      if (!files.has(file.name)) {
+        files.add(file.name)
+        yield * traverseFilteredTestNames('', reg, file)
+      }
+    }
+  }
+  catch {
+    // `new RegExp` may throw error when input is invalid regexp
+  }
+}
+
 export function registerConsoleShortcuts(
   ctx: Vitest,
   stdin: NodeJS.ReadStream = process.stdin,
@@ -126,24 +161,13 @@ export function registerConsoleShortcuts(
 
   async function inputNamePattern() {
     off()
-    const watchFilter = new WatchFilter(
+    const watchFilter = new WatchFilter<'object'>(
       'Input test name pattern (RegExp)',
       stdin,
       stdout,
     )
     const filter = await watchFilter.filter((str: string) => {
-      const files = ctx.state.getFiles()
-      const tests = getTests(files)
-      try {
-        const reg = new RegExp(str)
-        return tests
-          .map(test => test.name)
-          .filter(testName => testName.match(reg))
-      }
-      catch {
-        // `new RegExp` may throw error when input is invalid regexp
-        return []
-      }
+      return [...getFilteredTestNames(str, ctx.state.getFiles())]
     })
 
     on()

packages/vitest/src/node/watch-filter.ts

@@ -9,13 +9,18 @@ const MAX_RESULT_COUNT = 10
 const SELECTION_MAX_INDEX = 7
 const ESC = '\u001B['
 
-type FilterFunc = (keyword: string) => Promise<string[]> | string[]
+export interface FilterType {
+  name: string
+  toString: () => string
+}
+type FilterItemType<T extends 'string' | 'object' = 'string'> = T extends 'string' ? string : FilterType
+type FilterFuncType<T extends 'string' | 'object' = 'string'> = (keyword: string) => Promise<FilterItemType<T>[]> | FilterItemType<T>[]
 
-export class WatchFilter {
+export class WatchFilter<T extends 'string' | 'object' = 'string'> {
   private filterRL: readline.Interface
   private currentKeyword: string | undefined = undefined
   private message: string
-  private results: string[] = []
+  private results: FilterItemType<T>[] = []
   private selectionIndex = -1
   private onKeyPress?: (str: string, key: any) => void
   private stdin: NodeJS.ReadStream
@@ -40,7 +45,7 @@ export class WatchFilter {
     }
   }
 
-  public async filter(filterFunc: FilterFunc): Promise<string | undefined> {
+  public async filter(filterFunc: FilterFuncType<T>): Promise<string | undefined> {
     this.write(this.promptLine())
 
     const resultPromise = createDefer<string | undefined>()
@@ -58,7 +63,7 @@ export class WatchFilter {
   }
 
   private filterHandler(
-    filterFunc: FilterFunc,
+    filterFunc: FilterFuncType<T>,
     onSubmit: (result?: string) => void,
   ) {
     return async (str: string | undefined, key: any) => {
@@ -78,12 +83,17 @@ export class WatchFilter {
           onSubmit(undefined)
           break
         case key?.name === 'enter':
-        case key?.name === 'return':
+        case key?.name === 'return': {
+          const selection = this.results[this.selectionIndex]
+          const result = typeof selection === 'string'
+            ? selection
+            : selection?.name
           onSubmit(
-            this.results[this.selectionIndex] || this.currentKeyword || '',
+            result || this.currentKeyword || '',
           )
           this.currentKeyword = undefined
           break
+        }
         case key?.name === 'up':
           if (this.selectionIndex && this.selectionIndex > 0) {
             this.selectionIndex--