fix(forks): --cpu-prof and --heap-prof not working

vitest-dev · Sep 28, 2024 · 2e4d894 · 2e4d894
1 parent 1800167
commit 2e4d894
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/packages/vitest/src/runtime/worker.ts b/packages/vitest/src/runtime/worker.ts
@@ -11,6 +11,21 @@ import { disposeInternalListeners } from './workers/utils'
 
 if (isChildProcess()) {
   setProcessTitle(`vitest ${poolId}`)
+
+  const isProfiling = process.execArgv.some(
+    execArg =>
+      execArg.startsWith('--prof')
+      || execArg.startsWith('--cpu-prof')
+      || execArg.startsWith('--heap-prof')
+      || execArg.startsWith('--diagnostic-dir'),
+  )
+
+  if (isProfiling) {
+    // Work-around for nodejs/node#55094
+    process.on('SIGTERM', () => {
+      process.exit()
+    })
+  }
 }
 
 // this is what every pool executes when running tests