Centralize Logger mock in vitest-setup.ts, remove per-file duplicates

Move the Logger vi.mock into a single setupFiles entry so all test files
get it automatically. Remove the duplicated mock from 6 test files and
add vi.unmock in logger.test.ts which needs the real implementation.

https://claude.ai/code/session_01P16MVHHxCnhKCYdTCNayv3

Author: claude

src/__tests__/cross-file-prepass.test.ts

@@ -1,4 +1,4 @@
-import { afterEach, beforeEach, describe, it, expect, vi } from "vitest";
+import { afterEach, beforeEach, describe, it, expect } from "vitest";
 import { join, resolve as pathResolve } from "node:path";
 import { fileURLToPath } from "node:url";
 import { dirname } from "node:path";
@@ -25,16 +25,6 @@ import {
   patchConsumerFile,
 } from "../internal/bridge-consumer-patcher.js";
 
-// Suppress codemod logs in tests
-vi.mock("../internal/logger.js", () => ({
-  Logger: {
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    logWarnings: vi.fn(),
-  },
-}));
-
 const __dirname = dirname(fileURLToPath(import.meta.url));
 const fixturesDir = join(__dirname, "fixtures", "cross-file");
 const fixture = (name: string) => join(fixturesDir, name);

src/__tests__/extract-external-interface.test.ts

@@ -9,22 +9,12 @@ import {
 } from "node:fs";
 import { tmpdir } from "node:os";
 import path from "node:path";
-import { describe, it, expect, beforeAll, afterAll, vi } from "vitest";
+import { describe, it, expect, beforeAll, afterAll } from "vitest";
 
 import type { ExternalInterfaceResult } from "../adapter.js";
 import { createModuleResolver } from "../internal/prepass/resolve-imports.js";
 import { runPrepass } from "../internal/prepass/run-prepass.js";
 
-// Suppress codemod logs in tests
-vi.mock("../internal/logger.js", () => ({
-  Logger: {
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    logWarnings: vi.fn(),
-  },
-}));
-
 /** Recursively collect all .tsx/.ts/.jsx files in a directory. */
 function collectFiles(dir: string): string[] {
   const files: string[] = [];

src/__tests__/forwarded-as-patcher.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect, vi } from "vitest";
+import { describe, it, expect } from "vitest";
 import { join } from "node:path";
 import { fileURLToPath } from "node:url";
 import { dirname } from "node:path";
@@ -10,16 +10,6 @@ import {
   patchConsumerForwardedAs,
 } from "../internal/forwarded-as-consumer-patcher.js";
 
-// Suppress codemod logs in tests
-vi.mock("../internal/logger.js", () => ({
-  Logger: {
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    logWarnings: vi.fn(),
-  },
-}));
-
 const __dirname = dirname(fileURLToPath(import.meta.url));
 const fixturesDir = join(__dirname, "fixtures", "cross-file");
 const fixture = (name: string) => join(fixturesDir, name);

src/__tests__/logger.test.ts

@@ -1,4 +1,6 @@
 import { describe, it, expect, beforeEach, vi } from "vitest";
+
+vi.unmock("../internal/logger.js");
 import { Logger } from "../internal/logger.js";
 
 describe("Logger", () => {

src/__tests__/run-prepass-rg-prefilter.test.ts

@@ -3,16 +3,6 @@ import { tmpdir } from "node:os";
 import path from "node:path";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 
-// Suppress codemod logs in tests
-vi.mock("../internal/logger.js", () => ({
-  Logger: {
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    logWarnings: vi.fn(),
-  },
-}));
-
 interface FixturePaths {
   root: string;
   componentFile: string;

src/__tests__/run-prepass-selector-prefilter.test.ts

@@ -1,19 +1,9 @@
-import { describe, expect, it, vi } from "vitest";
+import { describe, expect, it } from "vitest";
 import { dirname, join, resolve as pathResolve } from "node:path";
 import { fileURLToPath } from "node:url";
 
 import { createModuleResolver } from "../internal/prepass/resolve-imports.js";
 import { runPrepass } from "../internal/prepass/run-prepass.js";
-
-// Suppress codemod logs in tests
-vi.mock("../internal/logger.js", () => ({
-  Logger: {
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    logWarnings: vi.fn(),
-  },
-}));
 import {
   scanCrossFileSelectors,
   type CrossFileInfo,

src/__tests__/run.e2e.test.ts

@@ -1,32 +1,11 @@
-import { describe, it, expect, vi } from "vitest";
+import { describe, it, expect } from "vitest";
 import { mkdtemp, copyFile, mkdir, readFile } from "node:fs/promises";
 import { tmpdir } from "node:os";
 import { join, dirname } from "node:path";
 import { fileURLToPath } from "node:url";
 import { execFileSync } from "node:child_process";
 import { format } from "oxfmt";
 import { runTransform } from "../run.js";
-
-// Suppress codemod logs in tests
-vi.mock("../internal/logger.js", () => ({
-  Logger: {
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    logError: vi.fn(),
-    logWarnings: vi.fn(),
-    createReport: vi.fn(() => ({
-      toString: () => "",
-      print: vi.fn(),
-      getWarnings: vi.fn(() => []),
-    })),
-    setFileCount: vi.fn(),
-    setMaxExamples: vi.fn(),
-    markErrorAsLogged: vi.fn(),
-    isErrorLogged: vi.fn(() => false),
-    _clearCollected: vi.fn(),
-  },
-}));
 import {
   runTransform as runTransformFromIndex,
   defineAdapter as defineAdapterFromIndex,

src/__tests__/transform.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect, vi } from "vitest";
+import { describe, it, expect } from "vitest";
 import { applyTransform } from "jscodeshift/src/testUtils.js";
 import jscodeshift from "jscodeshift";
 import { readdirSync, readFileSync, existsSync } from "node:fs";
@@ -23,16 +23,6 @@ function isBailOutFixture(filename: string): boolean {
   return BAIL_OUT_PREFIXES.some((prefix) => filename.startsWith(prefix));
 }
 
-// Suppress codemod logs in tests
-vi.mock("../internal/logger.js", () => ({
-  Logger: {
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    logWarnings: vi.fn(),
-  },
-}));
-
 const __dirname = dirname(fileURLToPath(import.meta.url));
 
 const testCasesDir = join(__dirname, "..", "..", "test-cases");

src/__tests__/vitest-setup.ts

@@ -0,0 +1,28 @@
+/**
+ * Vitest global setup: mock Logger to suppress codemod diagnostic output
+ * during test runs so that the agent reporter output stays clean.
+ *
+ * Tests that need the real Logger (e.g. logger.test.ts) should call
+ * vi.unmock("../internal/logger.js") at the top of the file.
+ */
+import { vi } from "vitest";
+
+vi.mock("../internal/logger.js", () => ({
+  Logger: {
+    info: vi.fn(),
+    warn: vi.fn(),
+    error: vi.fn(),
+    logError: vi.fn(),
+    logWarnings: vi.fn(),
+    createReport: vi.fn(() => ({
+      toString: () => "",
+      print: vi.fn(),
+      getWarnings: vi.fn(() => []),
+    })),
+    setFileCount: vi.fn(),
+    setMaxExamples: vi.fn(),
+    markErrorAsLogged: vi.fn(),
+    isErrorLogged: vi.fn(() => false),
+    _clearCollected: vi.fn(),
+  },
+}));

vitest.config.ts

@@ -8,6 +8,7 @@ export default defineConfig({
     testTimeout: 20000,
     hookTimeout: 10000,
     reporters: ["agent"],
+    setupFiles: ["src/__tests__/vitest-setup.ts"],
     coverage: {
       provider: "v8",
       reporter: ["text", "json", "html"],