From 2b586b423afae22d8b394a23ec1d5032ecf1fa2f Mon Sep 17 00:00:00 2001
From: Tak Hoffman <781889+Takhoffman@users.noreply.github.com>
Date: Fri, 27 Mar 2026 00:51:33 -0500
Subject: [PATCH] test(google): spy on image-generation auth surface

---
 extensions/google/image-generation-provider.test.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/extensions/google/image-generation-provider.test.ts b/extensions/google/image-generation-provider.test.ts
index b10c41a3950..75716fe1382 100644
--- a/extensions/google/image-generation-provider.test.ts
+++ b/extensions/google/image-generation-provider.test.ts
@@ -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",