test: stabilize gateway suites

src/gateway/server-discovery-runtime.ts

@@ -17,7 +17,12 @@ export async function startGatewayDiscovery(params: {
   logDiscovery: { info: (msg: string) => void; warn: (msg: string) => void };
 }) {
   let bonjourStop: (() => Promise<void>) | null = null;
-  const tailnetDns = await resolveTailnetDnsHint();
+  const bonjourEnabled =
+    process.env.CLAWDBOT_DISABLE_BONJOUR !== "1" &&
+    process.env.NODE_ENV !== "test" &&
+    !process.env.VITEST;
+  const needsTailnetDns = bonjourEnabled || params.wideAreaDiscoveryEnabled;
+  const tailnetDns = needsTailnetDns ? await resolveTailnetDnsHint() : undefined;
   const sshPortEnv = process.env.CLAWDBOT_SSH_PORT?.trim();
   const sshPortParsed = sshPortEnv ? Number.parseInt(sshPortEnv, 10) : NaN;
   const sshPort = Number.isFinite(sshPortParsed) && sshPortParsed > 0 ? sshPortParsed : undefined;

src/gateway/test-helpers.server.ts

@@ -29,6 +29,10 @@ import {
   testTailnetIPv4,
 } from "./test-helpers.mocks.js";
 
+// Preload the gateway server module once per worker.
+// Important: `test-helpers.mocks` must run before importing the server so vi.mock hooks apply.
+const serverModulePromise = import("./server.js");
+
 let previousHome: string | undefined;
 let previousUserProfile: string | undefined;
 let previousStateDir: string | undefined;
@@ -105,7 +109,7 @@ export function installGatewayTestHooks() {
     embeddedRunMock.waitResults.clear();
     drainSystemEvents(resolveMainSessionKeyFromConfig());
     resetAgentRunContextForTest();
-    const mod = await import("./server.js");
+    const mod = await serverModulePromise;
     mod.__resetModelCatalogCacheForTest();
     piSdkMock.enabled = false;
     piSdkMock.discoverCalls = 0;
@@ -184,7 +188,7 @@ export function onceMessage<T = unknown>(
 }
 
 export async function startGatewayServer(port: number, opts?: GatewayServerOptions) {
-  const mod = await import("./server.js");
+  const mod = await serverModulePromise;
   return await mod.startGatewayServer(port, opts);
 }
 

src/infra/machine-name.ts

@@ -31,6 +31,9 @@ function fallbackHostName() {
 export async function getMachineDisplayName(): Promise<string> {
   if (cachedPromise) return cachedPromise;
   cachedPromise = (async () => {
+    if (process.env.VITEST || process.env.NODE_ENV === "test") {
+      return fallbackHostName();
+    }
     if (process.platform === "darwin") {
       const computerName = await tryScutil("ComputerName");
       if (computerName) return computerName;

src/test-utils/ports.ts

@@ -1,4 +1,5 @@
 import { type AddressInfo, createServer } from "node:net";
+import { isMainThread, threadId } from "node:worker_threads";
 
 async function isPortFree(port: number): Promise<boolean> {
   if (!Number.isFinite(port) || port <= 0 || port > 65535) return false;
@@ -45,7 +46,11 @@ export async function getDeterministicFreePortBlock(params?: {
 
   const workerIdRaw = process.env.VITEST_WORKER_ID ?? process.env.VITEST_POOL_ID ?? "";
   const workerId = Number.parseInt(workerIdRaw, 10);
-  const shard = Number.isFinite(workerId) ? Math.max(0, workerId) : Math.abs(process.pid);
+  const shard = Number.isFinite(workerId)
+    ? Math.max(0, workerId)
+    : isMainThread
+      ? Math.abs(process.pid)
+      : Math.abs(threadId);
 
   const rangeSize = 1000;
   const shardCount = 30;
@@ -79,4 +84,3 @@ export async function getDeterministicFreePortBlock(params?: {
 
   throw new Error("failed to acquire a free port block");
 }
-