test: repair rebase fallout

pgondhi987 · pgondhi987 · commit c8dfd0b633d4 · 2026-05-11T06:23:07.000Z
diff --git a/src/agents/auth-profiles/oauth.mirror-refresh.test.ts b/src/agents/auth-profiles/oauth.mirror-refresh.test.ts
@@ -42,6 +42,14 @@ function expectPersistedOpenAICodexProfileWithoutInlineTokens(
   expect(credential).not.toHaveProperty("idToken");
 }
 
+function requireOAuthCredential(store: AuthProfileStore, profileId: string): OAuthCredential {
+  const profile = store.profiles[profileId];
+  if (!profile || profile.type !== "oauth") {
+    throw new Error(`expected OAuth credential for ${profileId}`);
+  }
+  return profile;
+}
+
 vi.mock("@mariozechner/pi-ai/oauth", () => ({
   getOAuthProviders: () => [{ id: "anthropic" }, { id: "openai-codex" }],
   getOAuthApiKey: vi.fn(async (provider: string, credentials: Record<string, OAuthCredential>) => {
diff --git a/src/agents/auth-profiles/profiles.test.ts b/src/agents/auth-profiles/profiles.test.ts
@@ -1,7 +1,7 @@
 import fs from "node:fs";
 import os from "node:os";
 import path from "node:path";
-import { describe, expect, it } from "vitest";
+import { describe, expect, it, vi } from "vitest";
 import { resolveOAuthDir } from "../../config/paths.js";
 import { AUTH_STORE_VERSION } from "./constants.js";
 import { resolveAuthStorePath } from "./paths.js";
@@ -1261,11 +1261,11 @@ describe("promoteAuthProfileInOrder", () => {
         agentDir,
       );
 
-vi.mock("./store.js", () => ({
-  ensureAuthProfileStoreForLocalUpdate: vi.fn(() => ({ version: 1, profiles: {} })),
-  saveAuthProfileStore: storeMocks.saveAuthProfileStore,
-  updateAuthProfileStoreWithLock: storeMocks.updateAuthProfileStoreWithLock,
-}));
+      const updated = await promoteAuthProfileInOrder({
+        agentDir,
+        provider: "openai-codex",
+        profileId: newProfileId,
+      });
 
       expect(updated?.order?.["openai-codex"]).toEqual([newProfileId, staleProfileId]);
       expect(loadAuthProfileStoreForRuntime(agentDir).order?.["openai-codex"]).toEqual([
diff --git a/src/commands/status-all/diagnosis.test.ts b/src/commands/status-all/diagnosis.test.ts
@@ -1,16 +1,26 @@
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import type { ProgressReporter } from "../../cli/progress.js";
 
+type GatewayLogPaths = {
+  logDir: string;
+  stdoutPath: string;
+  stderrPath: string;
+};
+
 const restartLogMocks = vi.hoisted(() => ({
-  resolveGatewayLogPaths: vi.fn(() => {
+  resolveGatewayLogPaths: vi.fn<() => GatewayLogPaths>(() => {
     throw new Error("skip log tail");
   }),
-  resolveGatewayRestartLogPath: vi.fn(() => "/tmp/gateway-restart.log"),
+  resolveGatewayRestartLogPath: vi.fn<() => string>(() => "/tmp/gateway-restart.log"),
 }));
 
 const gatewayMocks = vi.hoisted(() => ({
-  readFileTailLines: vi.fn(async () => [] as string[]),
-  summarizeLogTail: vi.fn((lines: string[]) => lines),
+  readFileTailLines: vi.fn<(filePath: string, maxLines: number) => Promise<string[]>>(
+    async () => [],
+  ),
+  summarizeLogTail: vi.fn<(lines: string[], opts?: { maxLines?: number }) => string[]>(
+    (lines) => lines,
+  ),
 }));
 
 vi.mock("../../daemon/restart-logs.js", () => ({
