fix(browser): wait for files scanning in browser (#1704)

* fix: wait for files scanning

* fix: update

* fix: update types and make them Promise

packages/vitest/src/api/setup.ts

@@ -50,8 +50,8 @@ export function setup(ctx: Vitest) {
         getFiles() {
           return ctx.state.getFiles()
         },
-        getPaths() {
-          return ctx.state.getPaths()
+        async getPaths() {
+          return await ctx.state.getPaths()
         },
         readFile(id) {
           return fs.readFile(id, 'utf-8')

packages/vitest/src/api/types.ts

@@ -11,7 +11,7 @@ export interface WebSocketHandlers {
   onCollected(files?: File[]): Promise<void>
   onTaskUpdate(packs: TaskResultPack[]): void
   getFiles(): File[]
-  getPaths(): string[]
+  getPaths(): Promise<string[]>
   getConfig(): ResolvedConfig
   getModuleGraph(id: string): Promise<ModuleGraphData>
   getTransformResult(id: string): Promise<TransformResultWithSource | undefined>

packages/vitest/src/node/core.ts

@@ -214,6 +214,7 @@ export class Vitest {
   async runFiles(paths: string[]) {
     // previous run
     await this.runningPromise
+    this.state.startCollectingPaths()
 
     // schedule the new run
     this.runningPromise = (async () => {
@@ -243,6 +244,7 @@ export class Vitest {
     })()
       .finally(() => {
         this.runningPromise = undefined
+        this.state.finishCollectingPaths()
       })
 
     return await this.runningPromise

packages/vitest/src/node/state.ts

@@ -1,9 +1,15 @@
 import type { ErrorWithDiff, File, Task, TaskResultPack, UserConsoleLog } from '../types'
 
+interface CollectingPromise {
+  promise: Promise<void>
+  resolve: () => void
+}
+
 // Note this file is shared for both node and browser, be aware to avoid node specific logic
 export class StateManager {
   filesMap = new Map<string, File>()
   pathsSet: Set<string> = new Set()
+  collectingPromise: CollectingPromise | undefined = undefined
   idMap = new Map<string, Task>()
   taskFileMap = new WeakMap<Task, File>()
   errorsSet = new Set<unknown>()
@@ -21,7 +27,21 @@ export class StateManager {
     return Array.from(this.errorsSet.values())
   }
 
-  getPaths() {
+  startCollectingPaths() {
+    let _resolve: CollectingPromise['resolve']
+    const promise = new Promise<void>((resolve) => {
+      _resolve = resolve
+    })
+    this.collectingPromise = { promise, resolve: _resolve! }
+  }
+
+  finishCollectingPaths() {
+    this.collectingPromise?.resolve()
+    this.collectingPromise = undefined
+  }
+
+  async getPaths() {
+    await this.collectingPromise?.promise
     return Array.from(this.pathsSet)
   }