chore: test

Author: GuillaumeLagrange

packages/tinybench-plugin/src/instrumented.ts

@@ -48,10 +48,10 @@ class InstrumentedBenchRunner extends BaseBenchRunner {
     await mongoMeasurement.start(uri);
 
     global.gc?.();
-    await this.wrapWithInstrumentHooksAsync(
-      this.wrapFunctionWithFrame(fn, true),
-      uri
-    );
+    await this.wrapFunctionWithFrame(
+      this.wrapWithInstrumentHooksAsync(fn, uri),
+      true
+    )();
 
     await mongoMeasurement.stop(uri);
     await fnOpts?.afterEach?.call(task, "run");
@@ -66,7 +66,8 @@ class InstrumentedBenchRunner extends BaseBenchRunner {
     fnOpts?.beforeAll?.call(task, "run");
     fnOpts?.beforeEach?.call(task, "run");
 
-    this.wrapWithInstrumentHooks(this.wrapFunctionWithFrame(fn, false), uri);
+    global.gc?.();
+    this.wrapFunctionWithFrame(this.wrapWithInstrumentHooks(fn, uri), false)();
 
     fnOpts?.afterEach?.call(task, "run");
     fnOpts?.afterAll?.call(task, "run");

packages/tinybench-plugin/src/shared.ts

@@ -38,23 +38,24 @@ export abstract class BaseBenchRunner {
     return this.bench.tasks;
   }
 
-  protected wrapWithInstrumentHooks<T>(fn: () => T, uri: string): T {
-    InstrumentHooks.startBenchmark();
-    const result = fn();
-    InstrumentHooks.stopBenchmark();
-    InstrumentHooks.setExecutedBenchmark(process.pid, uri);
-    return result;
+  protected wrapWithInstrumentHooks(fn: Fn, uri: string): Fn {
+    return () => {
+      InstrumentHooks.startBenchmark();
+      const result = fn();
+      InstrumentHooks.stopBenchmark();
+      InstrumentHooks.setExecutedBenchmark(process.pid, uri);
+      return result;
+    };
   }
 
-  protected async wrapWithInstrumentHooksAsync(
-    fn: Fn,
-    uri: string
-  ): Promise<unknown> {
-    InstrumentHooks.startBenchmark();
-    const result = await fn();
-    InstrumentHooks.stopBenchmark();
-    InstrumentHooks.setExecutedBenchmark(process.pid, uri);
-    return result;
+  protected wrapWithInstrumentHooksAsync(fn: Fn, uri: string): Fn {
+    return async () => {
+      InstrumentHooks.startBenchmark();
+      const result = await fn();
+      InstrumentHooks.stopBenchmark();
+      InstrumentHooks.setExecutedBenchmark(process.pid, uri);
+      return result;
+    };
   }
 
   protected abstract getModeName(): string;

packages/tinybench-plugin/src/walltime.ts

@@ -35,7 +35,7 @@ class WalltimeBenchRunner extends BaseBenchRunner {
     }
 
     await mongoMeasurement.start(uri);
-    await this.wrapWithInstrumentHooksAsync(() => task.run(), uri);
+    await this.wrapWithInstrumentHooksAsync(task.run, uri)();
     await mongoMeasurement.stop(uri);
 
     this.registerCodspeedBenchmarkFromTask(task);
@@ -49,7 +49,7 @@ class WalltimeBenchRunner extends BaseBenchRunner {
       task.warmup();
     }
 
-    this.wrapWithInstrumentHooks(() => task.runSync(), uri);
+    this.wrapWithInstrumentHooks(task.runSync, uri)();
 
     this.registerCodspeedBenchmarkFromTask(task);
   }