perf: use advanced serialization

Author: claneo

.vscode/tasks.json

@@ -2,20 +2,6 @@
 // for the documentation about the tasks.json format
 {
   "version": "2.0.0",
-  "problemMatchers": [
-    {
-      "name": "rslib-watch",
-      "owner": "rslib",
-      "background": {
-        "activeOnStart": true,
-        "beginsPattern": "build started...",
-        "endsPattern": "build completed, watching for changes..."
-      },
-      "pattern": {
-        "regexp": "build failed in"
-      }
-    }
-  ],
   "tasks": [
     {
       "label": "core dev",

packages/browser/src/protocol.ts

@@ -1,8 +1,8 @@
 import type { DevicePreset } from '@rstest/core/browser';
 import type {
   RuntimeConfig,
-  Test,
   TestFileResult,
+  TestInfo,
   TestResult,
 } from '@rstest/core/browser-runtime';
 import type { SnapshotUpdateState } from '@vitest/snapshot';
@@ -96,7 +96,7 @@ export type BrowserClientMessage =
   // Collect mode messages
   | {
       type: 'collect-result';
-      payload: { testPath: string; project: string; tests: Test[] };
+      payload: { testPath: string; project: string; tests: TestInfo[] };
     }
   | { type: 'collect-complete' }
   // Unified RPC envelope for all runner -> container/host capability calls.

packages/core/src/browserRuntime.ts

@@ -20,6 +20,7 @@ export type {
   RuntimeConfig,
   Test,
   TestFileResult,
+  TestInfo,
   TestResult,
   WorkerState,
 } from './types';

packages/core/src/core/listTests.ts

@@ -8,7 +8,7 @@ import type {
   Location,
   ProjectContext,
   RstestContext,
-  Test,
+  TestInfo,
 } from '../types';
 import {
   bgColor,
@@ -348,7 +348,7 @@ export async function listTests(
     type: 'file' | 'suite' | 'case';
   }[] = [];
 
-  const traverseTests = (test: Test) => {
+  const traverseTests = (test: TestInfo) => {
     if (['skip', 'todo'].includes(test.runMode)) {
       return;
     }

packages/core/src/pool/forks.ts

@@ -1,6 +1,5 @@
 import EventEmitter from 'node:events';
 import { fileURLToPath } from 'node:url';
-import v8 from 'node:v8';
 import { createBirpc } from 'birpc';
 import { dirname, resolve } from 'pathe';
 import { type Options, Tinypool } from 'tinypool';
@@ -50,8 +49,6 @@ export function createForksChannel(
   };
 
   const rpc = createBirpcImpl<ServerRPC, RuntimeRPC>(rpcMethods, {
-    serialize: v8.serialize,
-    deserialize: (v) => v8.deserialize(Buffer.from(v)),
     timeout: -1,
     post(v) {
       emitter.emit(events.message, v);
@@ -103,6 +100,7 @@ export const createForksPool = (poolOptions: {
     minThreads,
     concurrentTasksPerWorker: 1,
     isolateWorkers: isolate,
+    serialization: 'advanced',
   };
 
   const pool = new Tinypool(options);

packages/core/src/pool/index.ts

@@ -9,10 +9,10 @@ import type {
   RstestContext,
   RuntimeConfig,
   RuntimeRPC,
-  Test,
   TestCaseInfo,
   TestFileInfo,
   TestFileResult,
+  TestInfo,
   TestResult,
   TestSuiteInfo,
   UserConsoleLog,
@@ -22,7 +22,6 @@ import {
   getForceColorEnv,
   isDeno,
   needFlagExperimentalDetectModule,
-  serializableConfig,
 } from '../utils';
 import { isMemorySufficient } from '../utils/memory';
 import { createForksPool } from './forks';
@@ -158,7 +157,7 @@ export const createPool = async ({
     project: ProjectContext;
   }) => Promise<
     {
-      tests: Test[];
+      tests: TestInfo[];
       testPath: string;
       errors?: FormattedError[];
       project: string;
@@ -309,7 +308,7 @@ export const createPool = async ({
                   project: projectName,
                   rootPath: context.rootPath,
                   projectRoot: project.rootPath,
-                  runtimeConfig: serializableConfig(runtimeConfig),
+                  runtimeConfig,
                 },
                 type: 'run',
                 setupEntries,
@@ -417,7 +416,7 @@ export const createPool = async ({
                   outputModule: project.outputModule,
                   rootPath: context.rootPath,
                   projectRoot: project.rootPath,
-                  runtimeConfig: serializableConfig(runtimeConfig),
+                  runtimeConfig,
                 },
                 type: 'collect',
                 setupEntries,

packages/core/src/runtime/api/index.ts

@@ -2,9 +2,9 @@ import type {
   Rstest,
   RstestExpect,
   RunnerHooks,
-  Test,
   TestCase,
   TestFileResult,
+  TestInfo,
   WorkerState,
 } from '../../types';
 import { createRunner } from '../runner';
@@ -20,7 +20,7 @@ export const createRstestRuntime = async (
       hooks: RunnerHooks,
       api: Rstest,
     ) => Promise<TestFileResult>;
-    collectTests: () => Promise<Test[]>;
+    collectTests: () => Promise<TestInfo[]>;
     getCurrentTest: () => TestCase | undefined;
   };
   api: Rstest;

packages/core/src/runtime/runner/index.ts

@@ -19,7 +19,7 @@ export function createRunner({ workerState }: { workerState: WorkerState }): {
       hooks: RunnerHooks,
       api: Rstest,
     ) => Promise<TestFileResult>;
-    collectTests: () => Promise<Test[]>;
+    collectTests: () => Promise<TestInfo[]>;
     getCurrentTest: TestRunner['getCurrentTest'];
   };
 } {
@@ -55,18 +55,7 @@ export function createRunner({ workerState }: { workerState: WorkerState }): {
         traverseUpdateTest(tests, testNamePattern);
         hooks.onTestFileReady?.({
           testPath,
-          tests: tests.map(function toTestInfo(test: Test): TestInfo {
-            return {
-              testId: test.testId,
-              name: test.name,
-              parentNames: test.parentNames,
-              testPath: test.testPath,
-              project: test.project,
-              type: test.type,
-              location: test.location,
-              tests: test.type === 'suite' ? test.tests.map(toTestInfo) : [],
-            };
-          }),
+          tests: tests.map(toTestInfo),
         });
         runtime.instance.updateStatus('running');
 
@@ -85,9 +74,23 @@ export function createRunner({ workerState }: { workerState: WorkerState }): {
         const tests = await runtime.instance.getTests();
         traverseUpdateTest(tests, testNamePattern);
 
-        return tests;
+        return tests.map(toTestInfo);
       },
       getCurrentTest: () => testRunner.getCurrentTest(),
     },
   };
 }
+
+function toTestInfo(test: Test): TestInfo {
+  return {
+    testId: test.testId,
+    name: test.name,
+    parentNames: test.parentNames,
+    testPath: test.testPath,
+    project: test.project,
+    type: test.type,
+    location: test.location,
+    tests: test.type === 'suite' ? test.tests.map(toTestInfo) : [],
+    runMode: test.runMode,
+  };
+}

packages/core/src/runtime/runner/runner.ts

@@ -302,6 +302,7 @@ export class TestRunner {
           testId: test.testId,
           type: 'suite',
           location: test.location,
+          runMode: test.runMode,
         });
 
         if (test.tests.length === 0) {
@@ -393,6 +394,7 @@ export class TestRunner {
           project: test.project,
           type: 'case',
           location: test.location,
+          runMode: test.runMode,
         });
 
         do {

packages/core/src/runtime/worker/index.ts

@@ -2,16 +2,16 @@ import type {
   MaybePromise,
   Rstest,
   RunWorkerOptions,
-  Test,
   TestFileResult,
+  TestInfo,
   WorkerState,
 } from '../../types';
 import './setup';
+import type { FileCoverageData } from 'istanbul-lib-coverage';
 import { install } from 'source-map-support';
 import { createCoverageProvider } from '../../coverage';
 import { createWorkerMetaMessage } from '../../pool/workerMeta';
 import { globalApis } from '../../utils/constants';
-import { undoSerializableConfig } from '../../utils/helper';
 import { color } from '../../utils/logger';
 import { formatTestError, getRealTimers, setRealTimers } from '../util';
 import { createForksRpcOptions, createRuntimeRpc } from './rpc';
@@ -65,7 +65,6 @@ const preparePool = async ({
   globalCleanups.length = 0;
 
   setRealTimers();
-  context.runtimeConfig = undoSerializableConfig(context.runtimeConfig);
 
   // Prefer public env var from tinypool, fallback to context.taskId
   process.env.RSTEST_WORKER_ID = String(
@@ -273,7 +272,7 @@ const runInPool = async (
   options: RunWorkerOptions['options'],
 ): Promise<
   | {
-      tests: Test[];
+      tests: TestInfo[];
       testPath: string;
     }
   | TestFileResult
@@ -462,7 +461,12 @@ const runInPool = async (
       const coverageMap = coverageProvider.collect();
       if (coverageMap) {
         // Attach coverage data to test result
-        results.coverage = coverageMap.toJSON();
+        results.coverage = {};
+        Object.entries(coverageMap.toJSON()).forEach(([key, value]) => {
+          if ('toJSON' in value)
+            results.coverage![key] = value.toJSON() as FileCoverageData;
+          else results.coverage![key] = value;
+        });
       }
       // Cleanup
       coverageProvider.cleanup();