Test useCPED when on Node 24

Author: szegedi

package.json

@@ -57,6 +57,7 @@
     "mocha": "^10.2.0",
     "nan": "^2.22.2",
     "nyc": "^15.1.0",
+    "semver": "^7.7.2",
     "sinon": "^15.2.0",
     "source-map-support": "^0.5.21",
     "tmp": "0.2.1",

ts/test/test-time-profiler.ts

@@ -23,9 +23,15 @@ import {hrtime} from 'process';
 import {Label, Profile} from 'pprof-format';
 import {AssertionError} from 'assert';
 import {GenerateTimeLabelsArgs, LabelSet} from '../src/v8-types';
+import {AsyncLocalStorage} from 'async_hooks';
+import {satisfies} from 'semver';
 
 const assert = require('assert');
 
+const useCPED =
+  satisfies(process.versions.node, '>=24.0.0') &&
+  !process.execArgv.includes('--no-async-context-frame');
+
 const PROFILE_OPTIONS = {
   durationMillis: 500,
   intervalMicros: 1000,
@@ -44,11 +50,16 @@ describe('Time Profiler', () => {
         this.skip();
       }
       const startTime = BigInt(Date.now()) * 1000n;
+      if (useCPED) {
+        // Ensure an async context frame is created to hold the profiler context.
+        new AsyncLocalStorage().enterWith(1);
+      }
       time.start({
         intervalMicros: 20 * 1_000,
         durationMillis: PROFILE_OPTIONS.durationMillis,
         withContexts: true,
         lineNumbers: false,
+        useCPED,
       });
       const initialContext: {[key: string]: string} = {};
       time.setContext(initialContext);
@@ -97,11 +108,16 @@ describe('Time Profiler', () => {
       this.timeout(3000);
 
       const intervalNanos = PROFILE_OPTIONS.intervalMicros * 1_000;
+      if (useCPED) {
+        // Ensure an async context frame is created to hold the profiler context.
+        new AsyncLocalStorage().enterWith(1);
+      }
       time.start({
         intervalMicros: PROFILE_OPTIONS.intervalMicros,
         durationMillis: PROFILE_OPTIONS.durationMillis,
         withContexts: true,
         lineNumbers: false,
+        useCPED,
       });
       // By repeating the test few times, we also exercise the profiler
       // start-stop overlap behavior.

ts/test/worker.ts

@@ -3,13 +3,18 @@ import {pbkdf2} from 'crypto';
 import {time} from '../src/index';
 import {Profile, ValueType} from 'pprof-format';
 import {getAndVerifyPresence, getAndVerifyString} from './profiles-for-tests';
+import {satisfies} from 'semver';
 
 import assert from 'assert';
+import {AsyncLocalStorage} from 'async_hooks';
 
 const DURATION_MILLIS = 1000;
 const intervalMicros = 10000;
 const withContexts =
   process.platform === 'darwin' || process.platform === 'linux';
+const useCPED =
+  satisfies(process.versions.node, '>=24.0.0') &&
+  !process.execArgv.includes('--no-async-context-frame');
 
 function createWorker(durationMs: number): Promise<Profile[]> {
   return new Promise((resolve, reject) => {
@@ -52,15 +57,20 @@ function getCpuUsage() {
 }
 
 async function main(durationMs: number) {
+  if (useCPED) new AsyncLocalStorage().enterWith(1);
   time.start({
     durationMillis: durationMs * 3,
     intervalMicros,
     withContexts,
     collectCpuTime: withContexts,
+    useCPED: useCPED,
   });
+  if (withContexts) {
+    time.setContext({});
+  }
 
   const cpu0 = getCpuUsage();
-  const nbWorkers = Number(process.argv[2] || 2);
+  const nbWorkers = Number(process.argv[2] ?? 2);
 
   // start workers
   const workers = executeWorkers(nbWorkers, durationMs);
@@ -91,12 +101,17 @@ async function main(durationMs: number) {
 }
 
 async function worker(durationMs: number) {
+  if (useCPED) new AsyncLocalStorage().enterWith(1);
   time.start({
     durationMillis: durationMs,
     intervalMicros,
     withContexts,
     collectCpuTime: withContexts,
+    useCPED: useCPED,
   });
+  if (withContexts) {
+    time.setContext({});
+  }
 
   const deadline = Date.now() + durationMs;
   await Promise.all([bar(deadline), foo(deadline)]);