File size: 2,547 Bytes
88c4c60 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | import { beforeEach, describe, expect, it, vi } from "vitest";
vi.mock("../../open-sse/utils/proxyFetch.js", () => ({
proxyAwareFetch: vi.fn(),
}));
import { proxyAwareFetch } from "../../open-sse/utils/proxyFetch.js";
import { getUsageForProvider } from "../../open-sse/services/usage.js";
function jsonResponse(body, status = 200) {
return new Response(JSON.stringify(body), {
status,
headers: { "Content-Type": "application/json" },
});
}
describe("Gemini CLI usage project id resolution", () => {
beforeEach(() => {
vi.clearAllMocks();
});
it("uses the projectId stored on the provider connection", async () => {
proxyAwareFetch.mockResolvedValueOnce(jsonResponse({
buckets: [
{
modelId: "gemini-3-flash-preview",
remainingFraction: 0.75,
resetTime: "2026-05-25T12:00:00Z",
},
],
}));
const usage = await getUsageForProvider({
provider: "gemini-cli",
accessToken: "token",
projectId: "cloud-code-project",
});
expect(proxyAwareFetch).toHaveBeenCalledTimes(1);
expect(proxyAwareFetch).toHaveBeenCalledWith(
"https://cloudcode-pa.googleapis.com/v1internal:retrieveUserQuota",
expect.objectContaining({
body: JSON.stringify({ project: "cloud-code-project" }),
}),
null
);
expect(usage.quotas["gemini-3-flash-preview"]).toMatchObject({
used: 250,
total: 1000,
remainingPercentage: 75,
});
});
it("normalizes project objects returned by loadCodeAssist", async () => {
proxyAwareFetch
.mockResolvedValueOnce(jsonResponse({
cloudaicompanionProject: { id: "project-from-load" },
currentTier: { name: "Free" },
}))
.mockResolvedValueOnce(jsonResponse({ buckets: [] }));
await getUsageForProvider({
provider: "gemini-cli",
accessToken: "token",
});
expect(proxyAwareFetch).toHaveBeenLastCalledWith(
"https://cloudcode-pa.googleapis.com/v1internal:retrieveUserQuota",
expect.objectContaining({
body: JSON.stringify({ project: "project-from-load" }),
}),
null
);
});
it("returns actionable guidance when no project id is available", async () => {
proxyAwareFetch.mockResolvedValueOnce(jsonResponse({}));
const usage = await getUsageForProvider({
provider: "gemini-cli",
accessToken: "token",
});
expect(usage.message).toContain("Reconnect Gemini CLI");
expect(usage.message).toContain("Gemini Code Assist");
});
});
|