fix(pool): init VITEST_POOL_ID before environment setup

Author: AriPerkkio

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

@@ -73,6 +73,9 @@ export class Pool {
       let isMemoryLimitReached = false
       const runner = this.getPoolRunner(task, method)
 
+      const poolId = runner.poolId ?? this.getWorkerId()
+      runner.poolId = poolId
+
       const activeTask = { task, resolver, method, cancelTask }
       this.activeTasks.push(activeTask)
 
@@ -113,9 +116,6 @@ export class Pool {
         await runner.start().finally(() => clearTimeout(id))
       }
 
-      const poolId = runner.poolId ?? this.getWorkerId()
-      runner.poolId = poolId
-
       const span = runner.startTracesSpan(`vitest.worker.${method}`)
       // Start running the test in the worker
       runner.request(method, task.context)

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

@@ -127,7 +127,6 @@ export class PoolRunner {
       __vitest_worker_request__: true,
       type: method,
       context,
-      poolId: this.poolId!,
       otelCarrier: this.getOTELCarrier(),
     } satisfies WorkerRequest)
   }
@@ -182,6 +181,7 @@ export class PoolRunner {
 
       this.postMessage({
         type: 'start',
+        poolId: this.poolId!,
         __vitest_worker_request__: true,
         options: {
           reportMemory: this.worker.reportMemory ?? false,

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

@@ -70,6 +70,7 @@ export type WorkerRequest
   = { __vitest_worker_request__: true } & (
     | {
       type: 'start'
+      poolId: number
       options: { reportMemory: boolean }
       context: {
         environment: WorkerTestEnvironment
@@ -89,13 +90,11 @@ export type WorkerRequest
     | {
       type: 'run'
       context: WorkerExecuteContext
-      poolId: number
       otelCarrier?: OTELCarrier
     }
     | {
       type: 'collect'
       context: WorkerExecuteContext
-      poolId: number
       otelCarrier?: OTELCarrier
     }
     | { type: 'cancel' }

packages/vitest/src/runtime/workers/init.ts

@@ -41,6 +41,7 @@ export function init(worker: Options): void {
 
     switch (message.type) {
       case 'start': {
+        process.env.VITEST_POOL_ID = String(message.poolId)
         reportMemory = message.options.reportMemory
 
         const tracesStart = performance.now()
@@ -96,7 +97,6 @@ export function init(worker: Options): void {
         }
 
         try {
-          process.env.VITEST_POOL_ID = String(message.poolId)
           process.env.VITEST_WORKER_ID = String(message.context.workerId)
         }
         catch (error) {
@@ -155,7 +155,6 @@ export function init(worker: Options): void {
         }
 
         try {
-          process.env.VITEST_POOL_ID = String(message.poolId)
           process.env.VITEST_WORKER_ID = String(message.context.workerId)
         }
         catch (error) {