LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-04-27 16:33:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Google: move runtime tests to provider contracts

Browse Source
This commit is contained in:
Vincent Koc
2026-03-16 00:10:08 -07:00
parent 62de7e02ea
commit 182a00cc49
1 changed files with 0 additions and 197 deletions
Download Patch File Download Diff File Expand all files Collapse all files

197
extensions/google/gemini-cli-provider.test.ts
View File

@@ -1,197 +0,0 @@
import { describe, expect, it } from "vitest";
import type { ProviderPlugin } from "../../src/plugins/types.js";
import { createCapturedPluginRegistration } from "../../src/test-utils/plugin-registration.js";
import {
createProviderUsageFetch,
makeResponse,
} from "../../src/test-utils/provider-usage-fetch.js";
import googlePlugin from "./index.js";
function findProvider(providers: ProviderPlugin[], id: string): ProviderPlugin {
const provider = providers.find((candidate) => candidate.id === id);
if (!provider) {
throw new Error(`provider ${id} missing`);
}
return provider;
}
function registerGooglePlugin(): {
providers: ProviderPlugin[];
webSearchProvider: {
id: string;
envVars: string[];
label: string;
} | null;
webSearchProviderRegistered: boolean;
} {
const captured = createCapturedPluginRegistration();
googlePlugin.register(captured.api);
if (captured.providers.length === 0) {
throw new Error("provider registration missing");
}
const webSearchProvider = captured.webSearchProviders[0] ?? null;
return {
providers: captured.providers,
webSearchProviderRegistered: webSearchProvider !== null,
webSearchProvider:
webSearchProvider === null
? null
: {
id: webSearchProvider.id,
envVars: webSearchProvider.envVars,
label: webSearchProvider.label,
},
};
}
describe("google plugin", () => {
it("registers Google direct, Gemini CLI auth, and Gemini web search", () => {
const result = registerGooglePlugin();
expect(result.providers.map((provider) => provider.id)).toEqual([
"google",
"google-gemini-cli",
]);
expect(result.webSearchProviderRegistered).toBe(true);
expect(result.webSearchProvider).toMatchObject({
id: "gemini",
label: "Gemini (Google Search)",
envVars: ["GEMINI_API_KEY"],
});
});
it("owns google direct gemini 3.1 forward-compat resolution", () => {
const { providers } = registerGooglePlugin();
const provider = findProvider(providers, "google");
const model = provider.resolveDynamicModel?.({
provider: "google",
modelId: "gemini-3.1-pro-preview",
modelRegistry: {
find: (_provider: string, id: string) =>
id === "gemini-3-pro-preview"
? {
id,
name: id,
api: "google-generative-ai",
provider: "google",
baseUrl: "https://generativelanguage.googleapis.com",
reasoning: false,
input: ["text"],
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
contextWindow: 1_048_576,
maxTokens: 65_536,
}
: null,
} as never,
});
expect(model).toMatchObject({
id: "gemini-3.1-pro-preview",
provider: "google",
api: "google-generative-ai",
baseUrl: "https://generativelanguage.googleapis.com",
reasoning: true,
});
});
it("owns gemini cli 3.1 forward-compat resolution", () => {
const { providers } = registerGooglePlugin();
const provider = findProvider(providers, "google-gemini-cli");
const model = provider.resolveDynamicModel?.({
provider: "google-gemini-cli",
modelId: "gemini-3.1-pro-preview",
modelRegistry: {
find: (_provider: string, id: string) =>
id === "gemini-3-pro-preview"
? {
id,
name: id,
api: "google-gemini-cli",
provider: "google-gemini-cli",
baseUrl: "https://cloudcode-pa.googleapis.com",
reasoning: false,
input: ["text"],
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
contextWindow: 1_048_576,
maxTokens: 65_536,
}
: null,
} as never,
});
expect(model).toMatchObject({
id: "gemini-3.1-pro-preview",
provider: "google-gemini-cli",
reasoning: true,
});
});
it("owns usage-token parsing", async () => {
const { providers } = registerGooglePlugin();
const provider = findProvider(providers, "google-gemini-cli");
await expect(
provider.resolveUsageAuth?.({
config: {} as never,
env: {} as NodeJS.ProcessEnv,
provider: "google-gemini-cli",
resolveApiKeyFromConfigAndStore: () => undefined,
resolveOAuthToken: async () => ({
token: '{"token":"google-oauth-token"}',
accountId: "google-account",
}),
}),
).resolves.toEqual({
token: "google-oauth-token",
accountId: "google-account",
});
});
it("owns OAuth auth-profile formatting", () => {
const { providers } = registerGooglePlugin();
const provider = findProvider(providers, "google-gemini-cli");
expect(
provider.formatApiKey?.({
type: "oauth",
provider: "google-gemini-cli",
access: "google-oauth-token",
refresh: "refresh-token",
expires: Date.now() + 60_000,
projectId: "proj-123",
}),
).toBe('{"token":"google-oauth-token","projectId":"proj-123"}');
});
it("owns usage snapshot fetching", async () => {
const { providers } = registerGooglePlugin();
const provider = findProvider(providers, "google-gemini-cli");
const mockFetch = createProviderUsageFetch(async (url) => {
if (url.includes("cloudcode-pa.googleapis.com/v1internal:retrieveUserQuota")) {
return makeResponse(200, {
buckets: [
{ modelId: "gemini-3.1-pro-preview", remainingFraction: 0.4 },
{ modelId: "gemini-3.1-flash-preview", remainingFraction: 0.8 },
],
});
}
return makeResponse(404, "not found");
});
const snapshot = await provider.fetchUsageSnapshot?.({
config: {} as never,
env: {} as NodeJS.ProcessEnv,
provider: "google-gemini-cli",
token: "google-oauth-token",
timeoutMs: 5_000,
fetchFn: mockFetch as unknown as typeof fetch,
});
expect(snapshot).toMatchObject({
provider: "google-gemini-cli",
displayName: "Gemini",
});
expect(snapshot?.windows[0]).toEqual({ label: "Pro", usedPercent: 60 });
expect(snapshot?.windows[1]?.label).toBe("Flash");
expect(snapshot?.windows[1]?.usedPercent).toBeCloseTo(20);
});
});