test: align pi-ai oauth mocks

src/agents/pi-tools.workspace-only-false.test.ts

@@ -3,10 +3,14 @@ import os from "node:os";
 import path from "node:path";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 
-vi.mock("@mariozechner/pi-ai/oauth", () => ({
+vi.mock("@mariozechner/pi-ai", async (importOriginal) => {
-  getOAuthApiKey: () => undefined,
+  const original = await importOriginal<typeof import("@mariozechner/pi-ai")>();
-  getOAuthProviders: () => [],
+  return {
-}));
+    ...original,
+    getOAuthApiKey: () => undefined,
+    getOAuthProviders: () => [],
+  };
+});
 
 import { createOpenClawCodingTools } from "./pi-tools.js";
 

src/tts/tts.test.ts

@@ -7,15 +7,16 @@ import type { OpenClawConfig } from "../config/config.js";
 import { withEnv } from "../test-utils/env.js";
 import * as tts from "./tts.js";
 
-vi.mock("@mariozechner/pi-ai", () => ({
+vi.mock("@mariozechner/pi-ai", async (importOriginal) => {
-  completeSimple: vi.fn(),
+  const original = await importOriginal<typeof import("@mariozechner/pi-ai")>();
-}));
+  return {
-
+    ...original,
-vi.mock("@mariozechner/pi-ai/oauth", () => ({
+    completeSimple: vi.fn(),
-  // Some auth helpers import oauth provider metadata at module load time.
+    // Some auth helpers import oauth provider metadata at module load time.
-  getOAuthProviders: () => [],
+    getOAuthProviders: () => [],
-  getOAuthApiKey: vi.fn(async () => null),
+    getOAuthApiKey: vi.fn(async () => null),
-}));
+  };
+});
 
 vi.mock("../agents/pi-embedded-runner/model.js", () => ({
   resolveModel: vi.fn((provider: string, modelId: string) => ({

test/setup.ts

@@ -1,5 +1,15 @@
 import { afterAll, afterEach, beforeAll, vi } from "vitest";
 
+vi.mock("@mariozechner/pi-ai", async (importOriginal) => {
+  const original = await importOriginal<typeof import("@mariozechner/pi-ai")>();
+  return {
+    ...original,
+    getOAuthApiKey: () => undefined,
+    getOAuthProviders: () => [],
+    loginOpenAICodex: vi.fn(),
+  };
+});
+
 vi.mock("@mariozechner/pi-ai/oauth", () => ({
   getOAuthApiKey: () => undefined,
   getOAuthProviders: () => [],