From bdd9ed238aabce86dc5c8a6041302765ee5a9295 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 11 Mar 2026 20:11:38 +0000
Subject: [PATCH] test: align pi-ai oauth mocks

---
 .../pi-tools.workspace-only-false.test.ts     | 12 ++++++++----
 src/tts/tts.test.ts                           | 19 ++++++++++---------
 test/setup.ts                                 | 10 ++++++++++
 3 files changed, 28 insertions(+), 13 deletions(-)

diff --git a/src/agents/pi-tools.workspace-only-false.test.ts b/src/agents/pi-tools.workspace-only-false.test.ts
index fb18260db09..146eb943c49 100644
--- a/src/agents/pi-tools.workspace-only-false.test.ts
+++ b/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", () => ({
-  getOAuthApiKey: () => undefined,
-  getOAuthProviders: () => [],
-}));
+vi.mock("@mariozechner/pi-ai", async (importOriginal) => {
+  const original = await importOriginal<typeof import("@mariozechner/pi-ai")>();
+  return {
+    ...original,
+    getOAuthApiKey: () => undefined,
+    getOAuthProviders: () => [],
+  };
+});
 
 import { createOpenClawCodingTools } from "./pi-tools.js";
 
diff --git a/src/tts/tts.test.ts b/src/tts/tts.test.ts
index 642e403ec7b..d11190a21d4 100644
--- a/src/tts/tts.test.ts
+++ b/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", () => ({
-  completeSimple: vi.fn(),
-}));
-
-vi.mock("@mariozechner/pi-ai/oauth", () => ({
-  // Some auth helpers import oauth provider metadata at module load time.
-  getOAuthProviders: () => [],
-  getOAuthApiKey: vi.fn(async () => null),
-}));
+vi.mock("@mariozechner/pi-ai", async (importOriginal) => {
+  const original = await importOriginal<typeof import("@mariozechner/pi-ai")>();
+  return {
+    ...original,
+    completeSimple: vi.fn(),
+    // Some auth helpers import oauth provider metadata at module load time.
+    getOAuthProviders: () => [],
+    getOAuthApiKey: vi.fn(async () => null),
+  };
+});
 
 vi.mock("../agents/pi-embedded-runner/model.js", () => ({
   resolveModel: vi.fn((provider: string, modelId: string) => ({
diff --git a/test/setup.ts b/test/setup.ts
index f232e5fc2d0..a6f902cb90f 100644
--- a/test/setup.ts
+++ b/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: () => [],