refactor(tinybench-plugin): call setupCore no matter the codspeed mode if not disabled

Author: GuillaumeLagrange

packages/tinybench-plugin/src/index.ts

@@ -3,6 +3,7 @@ import {
   getGitDir,
   InstrumentHooks,
   mongoMeasurement,
+  setupCore,
   SetupInstrumentsRequestBody,
   SetupInstrumentsResponse,
   tryIntrospect,
@@ -22,6 +23,7 @@ export function withCodSpeed(bench: Bench): Bench {
   if (codspeedRunnerMode === "disabled") {
     return bench;
   }
+  setupCore();
 
   const rootCallingFile = getCallingFile();
 

packages/tinybench-plugin/src/instrumented.ts

@@ -2,7 +2,6 @@ import {
   InstrumentHooks,
   mongoMeasurement,
   optimizeFunction,
-  setupCore,
   teardownCore,
 } from "@codspeed/core";
 import { Bench, Fn, FnOptions } from "tinybench";
@@ -18,7 +17,6 @@ export function runInstrumentedBench(
     console.log(
       `[CodSpeed] running with @codspeed/tinybench v${__VERSION__} (instrumented mode)`
     );
-    setupCore();
 
     for (const task of bench.tasks) {
       const uri = getTaskUri(bench, task.name, rootCallingFile);

packages/tinybench-plugin/tests/index.integ.test.ts

@@ -205,7 +205,7 @@ describe("Benchmark.Suite", () => {
     expect(afterAll).toHaveBeenCalledTimes(2);
   });
 
-  it("should call setupCore and teardownCore only once after run()", async () => {
+  it("should call setupCore and teardownCore only once", async () => {
     mockCore.InstrumentHooks.isInstrumented.mockReturnValue(true);
     const bench = withCodSpeed(new Bench())
       .add("RegExp", function () {
@@ -215,7 +215,7 @@ describe("Benchmark.Suite", () => {
         /o/.test("Hello World!");
       });
 
-    expect(mockCore.setupCore).not.toHaveBeenCalled();
+    expect(mockCore.setupCore).toHaveBeenCalledTimes(1);
     expect(mockCore.teardownCore).not.toHaveBeenCalled();
 
     await bench.run();