源码快照 @ 2026-09-01 (source-only, 剔除非源码素材)

packages/agent/gui/vitest.shared.setup.ts

@@ -0,0 +1,66 @@
+import { afterEach, beforeEach } from "vitest";
+import {
+  resetAgentHostApiForTests,
+  setAgentHostApiForTests
+} from "./agentActivityHost";
+import type { AgentHostRuntimeApi } from "./host/agentHostApi";
+import { installReactRenderLoopConsoleTrap } from "./test/reactRenderLoopConsoleTrap";
+
+// Vitest 4 runs with NODE_ENV=development; agent test harnesses gate on "test".
+process.env.NODE_ENV = "test";
+
+const originalConsoleInfo = console.info.bind(console);
+console.info = (...args: unknown[]) => {
+  if (isSuppressedAgentGuiDiagnostic(args)) {
+    return;
+  }
+  originalConsoleInfo(...args);
+};
+
+let restoreReactRenderLoopConsoleTrap: (() => void) | null = null;
+
+beforeEach(() => {
+  restoreReactRenderLoopConsoleTrap?.();
+  restoreReactRenderLoopConsoleTrap = installReactRenderLoopConsoleTrap({
+    console
+  });
+  resetAgentHostApiForTests();
+  resetMentionSearchBrowseCacheForTests();
+  setAgentHostApiForTests(createTestAgentHostApi());
+});
+
+afterEach(() => {
+  try {
+    resetAgentHostApiForTests();
+    resetMentionSearchBrowseCacheForTests();
+  } finally {
+    restoreReactRenderLoopConsoleTrap?.();
+    restoreReactRenderLoopConsoleTrap = null;
+  }
+});
+
+export function createTestAgentHostApi(): AgentHostRuntimeApi {
+  return {
+    account: {},
+    clipboard: {},
+    debug: {},
+    filesystem: {},
+    workspace: {}
+  } as unknown as AgentHostRuntimeApi;
+}
+
+function resetMentionSearchBrowseCacheForTests(): void {
+  (
+    globalThis as typeof globalThis & {
+      __tuttiResetAgentMentionSearchBrowseCacheForTests?: () => void;
+    }
+  ).__tuttiResetAgentMentionSearchBrowseCacheForTests?.();
+}
+
+function isSuppressedAgentGuiDiagnostic(args: readonly unknown[]): boolean {
+  const [prefix] = args;
+  return (
+    prefix === "[agent-gui] mention-lifecycle" ||
+    prefix === "[agent-gui] mention-search"
+  );
+}