feat(vitest-runner): add mongo-measurement integration

Author: adriencaccia

packages/core/moon.yml

@@ -5,7 +5,8 @@ tasks:
       - generated/openapi
   build:
     deps:
-      - "build-native-addon"
+      - build-native-addon
+      - build-tracer-client
 
   build-native-addon:
     command: prebuildify --napi --strip

packages/vitest-plugin/src/__tests__/runner.test.ts

@@ -11,6 +11,10 @@ const coreMocks = vi.hoisted(() => {
     },
     setupCore: vi.fn(),
     teardownCore: vi.fn(),
+    mongoMeasurement: {
+      start: vi.fn(),
+      stop: vi.fn(),
+    },
   };
 });
 
@@ -44,7 +48,13 @@ describe("CodSpeedRunner", () => {
     );
 
     // run
+    expect(coreMocks.mongoMeasurement.start).toHaveBeenCalledWith(
+      "packages/vitest-plugin/src/__tests__/runner.test.ts::test bench"
+    );
+    expect(coreMocks.Measurement.startInstrumentation).toHaveBeenCalledTimes(1);
     expect(benchFn).toHaveBeenCalledTimes(2);
+    expect(coreMocks.Measurement.stopInstrumentation).toHaveBeenCalledTimes(1);
+    expect(coreMocks.mongoMeasurement.stop).toHaveBeenCalledTimes(1);
     expect(console.log).toHaveBeenCalledWith(
       "[CodSpeed] packages/vitest-plugin/src/__tests__/runner.test.ts::test bench done"
     );
@@ -89,7 +99,13 @@ describe("CodSpeedRunner", () => {
     );
 
     // run
+    expect(coreMocks.mongoMeasurement.start).toHaveBeenCalledWith(
+      "packages/vitest-plugin/src/__tests__/runner.test.ts::nested suite::test bench"
+    );
+    expect(coreMocks.Measurement.startInstrumentation).toHaveBeenCalledTimes(1);
     expect(benchFn).toHaveBeenCalledTimes(2);
+    expect(coreMocks.Measurement.stopInstrumentation).toHaveBeenCalledTimes(1);
+    expect(coreMocks.mongoMeasurement.stop).toHaveBeenCalledTimes(1);
     expect(console.log).toHaveBeenCalledWith(
       "[CodSpeed] packages/vitest-plugin/src/__tests__/runner.test.ts::nested suite::test bench done"
     );

packages/vitest-plugin/src/runner.ts

@@ -2,6 +2,7 @@ import {
   getGitDir,
   logDebug,
   Measurement,
+  mongoMeasurement,
   optimizeFunction,
   setupCore,
   teardownCore,
@@ -48,12 +49,14 @@ async function runBenchmarkSuite(suite: Suite, parentSuiteName?: string) {
     const fn = getBenchFn(benchmark);
 
     await optimizeFunction(fn);
+    await mongoMeasurement.start(uri);
     await (async function __codspeed_root_frame__() {
       Measurement.startInstrumentation();
       // @ts-expect-error we do not need to bind the function to an instance of tinybench's Bench
       await fn();
       Measurement.stopInstrumentation(uri);
     })();
+    await mongoMeasurement.stop(uri);
 
     logCodSpeed(`${uri} done`);
   }
@@ -82,7 +85,10 @@ class CodSpeedRunner extends NodeBenchmarkRunner {
     await runBenchmarkSuite(suite);
     logCodSpeed(`running suite ${suite.name} done`);
 
-    teardownCore();
+    const mongoAggregate = await teardownCore();
+    logCodSpeed(
+      `[CodSpeed] Mongo Aggregate: ${JSON.stringify(mongoAggregate, null, 2)}`
+    );
   }
 }