test(google): spy on image-generation auth surface

This commit is contained in:
Tak Hoffman
2026-03-27 00:51:33 -05:00
parent 04d01984ef
commit 2b586b423a

View File

@@ -1,10 +1,10 @@
import * as providerAuth from "openclaw/plugin-sdk/provider-auth";
import * as imageGenerationCore from "openclaw/plugin-sdk/image-generation-core";
import { afterEach, describe, expect, it, vi } from "vitest";
import { buildGoogleImageGenerationProvider } from "./image-generation-provider.js";
import { __testing as geminiWebSearchTesting } from "./src/gemini-web-search-provider.js";
function mockGoogleApiKeyAuth() {
vi.spyOn(providerAuth, "resolveApiKeyForProvider").mockResolvedValue({
vi.spyOn(imageGenerationCore, "resolveApiKeyForProvider").mockResolvedValue({
apiKey: "google-test-key",
source: "env",
mode: "api-key",
@@ -48,7 +48,7 @@ describe("Google image-generation provider", () => {
});
it("generates image buffers from the Gemini generateContent API", async () => {
vi.spyOn(providerAuth, "resolveApiKeyForProvider").mockResolvedValue({
vi.spyOn(imageGenerationCore, "resolveApiKeyForProvider").mockResolvedValue({
apiKey: "google-test-key",
source: "env",
mode: "api-key",
@@ -118,7 +118,7 @@ describe("Google image-generation provider", () => {
});
it("accepts OAuth JSON auth and inline_data responses", async () => {
vi.spyOn(providerAuth, "resolveApiKeyForProvider").mockResolvedValue({
vi.spyOn(imageGenerationCore, "resolveApiKeyForProvider").mockResolvedValue({
apiKey: JSON.stringify({ token: "oauth-token" }),
source: "profile",
mode: "token",