refactor: forks config wrapping and unwrapping

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

@@ -8,6 +8,7 @@ import type { ContextRPC, ContextTestEnvironment, ResolvedConfig, RunnerRPC, Run
 import type { PoolProcessOptions, ProcessPool, RunWithFiles } from '../pool'
 import type { WorkspaceProject } from '../workspace'
 import { envsOrder, groupFilesByEnv } from '../../utils/test-helpers'
+import { wrapSerializableConfig } from '../../utils/config-helpers'
 import { groupBy, resolve } from '../../utils'
 import { createMethodsRPC } from './rpc'
 
@@ -44,12 +45,6 @@ function createChildProcessChannel(project: WorkspaceProject) {
   return { channel, cleanup }
 }
 
-function stringifyRegex(input: RegExp | string): string {
-  if (typeof input === 'string')
-    return input
-  return `$$vitest:${input.toString()}`
-}
-
 export function createForksPool(ctx: Vitest, { execArgv, env }: PoolProcessOptions): ProcessPool {
   const numCpus
     = typeof nodeos.availableParallelism === 'function'
@@ -144,14 +139,7 @@ export function createForksPool(ctx: Vitest, { execArgv, env }: PoolProcessOptio
           return configs.get(project)!
 
         const _config = project.getSerializableConfig()
-
-        const config = {
-          ..._config,
-          // v8 serialize does not support regex
-          testNamePattern: _config.testNamePattern
-            ? stringifyRegex(_config.testNamePattern)
-            : undefined,
-        } as ResolvedConfig
+        const config = wrapSerializableConfig(_config)
 
         configs.set(project, config)
         return config

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

@@ -12,6 +12,7 @@ import { groupFilesByEnv } from '../../utils/test-helpers'
 import { AggregateError } from '../../utils/base'
 import type { WorkspaceProject } from '../workspace'
 import { getWorkerMemoryLimit, stringToBytes } from '../../utils/memory-limit'
+import { wrapSerializableConfig } from '../../utils/config-helpers'
 import { createMethodsRPC } from './rpc'
 
 const suppressWarningsPath = resolve(rootDir, './suppress-warnings.cjs')
@@ -49,12 +50,6 @@ function createChildProcessChannel(project: WorkspaceProject) {
   return { channel, cleanup }
 }
 
-function stringifyRegex(input: RegExp | string): string {
-  if (typeof input === 'string')
-    return input
-  return `$$vitest:${input.toString()}`
-}
-
 export function createVmForksPool(ctx: Vitest, { execArgv, env }: PoolProcessOptions): ProcessPool {
   const numCpus
     = typeof nodeos.availableParallelism === 'function'
@@ -149,14 +144,8 @@ export function createVmForksPool(ctx: Vitest, { execArgv, env }: PoolProcessOpt
           return configs.get(project)!
 
         const _config = project.getSerializableConfig()
+        const config = wrapSerializableConfig(_config)
 
-        const config = {
-          ..._config,
-          // v8 serialize does not support regex
-          testNamePattern: _config.testNamePattern
-            ? stringifyRegex(_config.testNamePattern)
-            : undefined,
-        } as ResolvedConfig
         configs.set(project, config)
         return config
       }

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

@@ -1,6 +1,6 @@
 import v8 from 'node:v8'
 import type { WorkerGlobalState } from '../../types/worker'
-import { createForksRpcOptions, unwrapForksConfig } from './utils'
+import { createForksRpcOptions, unwrapSerializableConfig } from './utils'
 import { runBaseTests } from './base'
 import type { VitestWorker } from './types'
 
@@ -13,7 +13,7 @@ class ForksBaseWorker implements VitestWorker {
     // TODO: don't rely on reassigning process.exit
     // https://github.com/vitest-dev/vitest/pull/4441#discussion_r1443771486
     const exit = process.exit
-    state.ctx.config = unwrapForksConfig(state.ctx.config)
+    state.ctx.config = unwrapSerializableConfig(state.ctx.config)
 
     try {
       await runBaseTests(state)

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

@@ -4,6 +4,8 @@ import type { WorkerContext } from '../../types/worker'
 import type { ResolvedConfig } from '../../types/config'
 import type { WorkerRpcOptions } from './types'
 
+const REGEXP_WRAP_PREFIX = '$$vitest:'
+
 export function createThreadsRpcOptions({ port }: WorkerContext): WorkerRpcOptions {
   return {
     post: (v) => { port.postMessage(v) },
@@ -28,15 +30,16 @@ export function createForksRpcOptions(nodeV8: typeof import('v8')): WorkerRpcOpt
   }
 }
 
-function parsePossibleRegexp(str: string | RegExp) {
-  const prefix = '$$vitest:'
-  if (typeof str === 'string' && str.startsWith(prefix))
-    return parseRegexp(str.slice(prefix.length))
-  return str
-}
+/**
+ * Reverts the wrapping done by `utils/config-helpers.ts`'s `wrapSerializableConfig`
+ */
+export function unwrapSerializableConfig(config: ResolvedConfig) {
+  if (config.testNamePattern && typeof config.testNamePattern === 'string') {
+    const testNamePattern = config.testNamePattern as string
+
+    if (testNamePattern.startsWith(REGEXP_WRAP_PREFIX))
+      config.testNamePattern = parseRegexp(testNamePattern.slice(REGEXP_WRAP_PREFIX.length))
+  }
 
-export function unwrapForksConfig(config: ResolvedConfig) {
-  if (config.testNamePattern)
-    config.testNamePattern = parsePossibleRegexp(config.testNamePattern) as RegExp
   return config
 }

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

@@ -1,6 +1,6 @@
 import v8 from 'node:v8'
 import type { WorkerGlobalState } from '../../types/worker'
-import { createForksRpcOptions, unwrapForksConfig } from './utils'
+import { createForksRpcOptions, unwrapSerializableConfig } from './utils'
 import type { VitestWorker } from './types'
 import { runVmTests } from './vm'
 
@@ -11,7 +11,7 @@ class ForksVmWorker implements VitestWorker {
 
   async runTests(state: WorkerGlobalState) {
     const exit = process.exit
-    state.ctx.config = unwrapForksConfig(state.ctx.config)
+    state.ctx.config = unwrapSerializableConfig(state.ctx.config)
 
     try {
       await runVmTests(state)

packages/vitest/src/utils/config-helpers.ts

@@ -1,5 +1,8 @@
+import type { ResolvedConfig } from '../types/config'
 import type { BenchmarkBuiltinReporters, BuiltinReporters } from '../node/reporters'
 
+const REGEXP_WRAP_PREFIX = '$$vitest:'
+
 interface PotentialConfig {
   outputFile?: string | Partial<Record<string, string>>
 }
@@ -13,3 +16,19 @@ export function getOutputFile(config: PotentialConfig | undefined, reporter: Bui
 
   return config.outputFile[reporter]
 }
+
+/**
+ * Prepares `ResolvedConfig` for serialization, e.g. `node:v8.serialize`
+ */
+export function wrapSerializableConfig(config: ResolvedConfig) {
+  let testNamePattern = config.testNamePattern
+
+  // v8 serialize does not support regex
+  if (testNamePattern && typeof testNamePattern !== 'string')
+    testNamePattern = `${REGEXP_WRAP_PREFIX}${testNamePattern.toString()}` as unknown as RegExp
+
+  return {
+    ...config,
+    testNamePattern,
+  } as ResolvedConfig
+}

packages/vitest/src/workers.ts

@@ -1,4 +1,4 @@
-export { createForksRpcOptions, createThreadsRpcOptions, unwrapForksConfig } from './runtime/workers/utils'
+export { createForksRpcOptions, createThreadsRpcOptions, unwrapSerializableConfig } from './runtime/workers/utils'
 export { provideWorkerState } from './utils/global'
 export { run as runVitestWorker } from './runtime/worker'
 export { runVmTests } from './runtime/workers/vm'