test(zoo-gateway): cover surfaceGatewayApiError UX branches for codecov patch

Adds vscode + i18n mocks and asserts the 401/402/403/429 paths in
surfaceGatewayApiError: token clear + sign-in URL on 401, add-credits
URL on 402 and budget-coded 429, support URL on 403, no-op on 429
without a budget code or on errors without a status. Also verifies the
helper still runs before completePrompt rewraps the upstream error.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: James Mtendamema

src/api/providers/__tests__/zoo-gateway.spec.ts

@@ -1,6 +1,19 @@
 // npx vitest run src/api/providers/__tests__/zoo-gateway.spec.ts
 
-vitest.mock("vscode", () => ({}))
+const { showErrorMessage, openExternal } = vitest.hoisted(() => ({
+	showErrorMessage: vitest.fn(async () => undefined as string | undefined),
+	openExternal: vitest.fn(async () => true),
+}))
+
+vitest.mock("vscode", () => ({
+	window: { showErrorMessage },
+	env: { openExternal, uriScheme: "vscode", appName: "VS Code" },
+	Uri: { parse: (value: string) => ({ toString: () => value }) },
+}))
+
+vitest.mock("../../../i18n", () => ({
+	t: (key: string) => key,
+}))
 
 import OpenAI from "openai"
 
@@ -9,6 +22,7 @@ import { zooGatewayDefaultModelId, ZOO_GATEWAY_DEFAULT_TEMPERATURE } from "@roo-
 import { ZooGatewayHandler } from "../zoo-gateway"
 import { ApiHandlerOptions } from "../../../shared/api"
 import { Package } from "../../../shared/package"
+import { clearZooCodeToken } from "../../../services/zoo-code-auth"
 
 vitest.mock("openai")
 vitest.mock("delay", () => ({ default: vitest.fn(() => Promise.resolve()) }))
@@ -78,9 +92,32 @@ describe("ZooGatewayHandler", () => {
 	beforeEach(() => {
 		vitest.clearAllMocks()
 		mockCreate.mockClear()
+		showErrorMessage.mockReset()
+		showErrorMessage.mockResolvedValue(undefined)
+		openExternal.mockReset()
+		openExternal.mockResolvedValue(true)
 		mockOpenAIClient()
 	})
 
+	function makeApiError(status: number, options: { code?: string; message?: string } = {}) {
+		const err = new Error(options.message ?? `HTTP ${status}`) as Error & {
+			status: number
+			code?: string
+		}
+		err.status = status
+		if (options.code) err.code = options.code
+		return err
+	}
+
+	async function drainCreateMessage(handler: ZooGatewayHandler) {
+		const stream = handler.createMessage("system", [{ role: "user", content: "hi" }])
+		const out: unknown[] = []
+		for await (const chunk of stream) {
+			out.push(chunk)
+		}
+		return out
+	}
+
 	describe("constructor", () => {
 		it("allows construction without a session token (auth is enforced at request time)", () => {
 			expect(() => new ZooGatewayHandler({})).not.toThrow()
@@ -336,4 +373,112 @@ describe("ZooGatewayHandler", () => {
 			await expect(handler.completePrompt("Test")).resolves.toBe("")
 		})
 	})
+
+	describe("surfaceGatewayApiError", () => {
+		it("clears the cached token and offers re-sign-in on 401", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw makeApiError(401)
+			})
+			showErrorMessage.mockResolvedValueOnce("common:zooAuth.buttons.sign_in")
+
+			await expect(drainCreateMessage(handler)).rejects.toThrow()
+			expect(clearZooCodeToken).toHaveBeenCalledTimes(1)
+			expect(showErrorMessage).toHaveBeenCalledWith(
+				"common:zooAuth.errors.session_expired",
+				"common:zooAuth.buttons.sign_in",
+			)
+			expect(openExternal).toHaveBeenCalledTimes(1)
+		})
+
+		it("does not open a URL on 401 when the user dismisses the prompt", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw makeApiError(401)
+			})
+			showErrorMessage.mockResolvedValueOnce(undefined)
+
+			await expect(drainCreateMessage(handler)).rejects.toThrow()
+			expect(clearZooCodeToken).toHaveBeenCalledTimes(1)
+			expect(openExternal).not.toHaveBeenCalled()
+		})
+
+		it("prompts to add credits on 402", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw makeApiError(402)
+			})
+			showErrorMessage.mockResolvedValueOnce("common:zooAuth.buttons.add_credits")
+
+			await expect(drainCreateMessage(handler)).rejects.toThrow()
+			expect(clearZooCodeToken).not.toHaveBeenCalled()
+			expect(showErrorMessage).toHaveBeenCalledWith(
+				"common:zooAuth.errors.out_of_credits",
+				"common:zooAuth.buttons.add_credits",
+			)
+			expect(openExternal).toHaveBeenCalledTimes(1)
+		})
+
+		it("shows the budget message on 429 with a budget code", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw makeApiError(429, { code: "monthly_budget_exceeded" })
+			})
+
+			await expect(drainCreateMessage(handler)).rejects.toThrow()
+			expect(showErrorMessage).toHaveBeenCalledWith(
+				"common:zooAuth.errors.budget_exceeded",
+				"common:zooAuth.buttons.add_credits",
+			)
+		})
+
+		it("does not surface a notification on 429 without a budget code", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw makeApiError(429, { code: "rate_limited" })
+			})
+
+			await expect(drainCreateMessage(handler)).rejects.toThrow()
+			expect(showErrorMessage).not.toHaveBeenCalled()
+		})
+
+		it("offers contact support on 403", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw makeApiError(403)
+			})
+			showErrorMessage.mockResolvedValueOnce("common:zooAuth.buttons.contact_support")
+
+			await expect(drainCreateMessage(handler)).rejects.toThrow()
+			expect(showErrorMessage).toHaveBeenCalledWith(
+				"common:zooAuth.errors.account_unavailable",
+				"common:zooAuth.buttons.contact_support",
+			)
+			expect(openExternal).toHaveBeenCalledTimes(1)
+		})
+
+		it("ignores errors without an HTTP status", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw new Error("network down")
+			})
+
+			await expect(drainCreateMessage(handler)).rejects.toThrow("network down")
+			expect(showErrorMessage).not.toHaveBeenCalled()
+			expect(clearZooCodeToken).not.toHaveBeenCalled()
+		})
+
+		it("surfaces the gateway error then wraps the message in completePrompt", async () => {
+			const handler = new ZooGatewayHandler(mockOptions)
+			mockCreate.mockImplementation(() => {
+				throw makeApiError(402, { message: "out of credits" })
+			})
+
+			await expect(handler.completePrompt("ping")).rejects.toThrow("Zoo Gateway completion error: out of credits")
+			expect(showErrorMessage).toHaveBeenCalledWith(
+				"common:zooAuth.errors.out_of_credits",
+				"common:zooAuth.buttons.add_credits",
+			)
+		})
+	})
 })