feat: process timeout to log names of stuck test files (#3031)

Author: AriPerkkio

packages/vitest/package.json

@@ -137,7 +137,7 @@
     "std-env": "^3.3.1",
     "strip-literal": "^1.0.0",
     "tinybench": "^2.3.1",
-    "tinypool": "^0.3.1",
+    "tinypool": "^0.4.0",
     "tinyspy": "^1.0.2",
     "vite": "^3.0.0 || ^4.0.0",
     "vite-node": "workspace:*",

packages/vitest/src/node/core.ts

@@ -597,6 +597,7 @@ export class Vitest {
     setTimeout(() => {
       this.report('onProcessTimeout').then(() => {
         console.warn(`close timed out after ${this.config.teardownTimeout}ms`)
+        this.state.getProcessTimeoutCauses().forEach(cause => console.warn(cause))
         process.exit()
       })
     }, this.config.teardownTimeout).unref()

packages/vitest/src/node/pools/threads.ts

@@ -54,6 +54,8 @@ export function createThreadsPool(ctx: Vitest, { execArgv, env }: PoolProcessOpt
 
     env,
     execArgv,
+
+    terminateTimeout: ctx.config.teardownTimeout,
   }
 
   if (ctx.config.isolate) {
@@ -87,6 +89,13 @@ export function createThreadsPool(ctx: Vitest, { execArgv, env }: PoolProcessOpt
       try {
         await pool.run(data, { transferList: [workerPort], name })
       }
+      catch (error) {
+        // Worker got stuck and won't terminate - this may cause process to hang
+        if (error instanceof Error && /Failed to terminate worker/.test(error.message))
+          ctx.state.addProcessTimeoutCause(`Failed to terminate worker while running ${files.join(', ')}.`)
+        else
+          throw error
+      }
       finally {
         port.close()
         workerPort.close()

packages/vitest/src/node/state.ts

@@ -22,6 +22,7 @@ export class StateManager {
   idMap = new Map<string, Task>()
   taskFileMap = new WeakMap<Task, File>()
   errorsSet = new Set<unknown>()
+  processTimeoutCauses = new Set<string>()
 
   catchError(err: unknown, type: string): void {
     if (isAggregateError(err))
@@ -39,6 +40,14 @@ export class StateManager {
     return Array.from(this.errorsSet.values())
   }
 
+  addProcessTimeoutCause(cause: string) {
+    this.processTimeoutCauses.add(cause)
+  }
+
+  getProcessTimeoutCauses() {
+    return Array.from(this.processTimeoutCauses.values())
+  }
+
   getPaths() {
     return Array.from(this.pathsSet)
   }