42c691b3ea
getAntigravityUsage filtered the fetchAvailableModels response through a hardcoded importantModels list that only contained 3.5 Flash, silently dropping the 3.6 Flash quota buckets so no usage bar rendered.
79 lines
2.6 KiB
JavaScript
79 lines
2.6 KiB
JavaScript
import { describe, expect, it, vi, beforeEach } from "vitest";
|
|
|
|
const proxyAwareFetch = vi.fn(async (url) => ({
|
|
ok: true,
|
|
status: 200,
|
|
json: async () => url.includes(":loadCodeAssist")
|
|
? { cloudaicompanionProject: "project-1", currentTier: { name: "Pro" } }
|
|
: {
|
|
models: {
|
|
"gemini-3.6-flash-high": {
|
|
displayName: "Gemini 3.6 Flash (High)",
|
|
quotaInfo: { remainingFraction: 0.8, resetTime: "2026-07-25T12:00:00Z" },
|
|
},
|
|
"gemini-3.6-flash-medium": {
|
|
displayName: "Gemini 3.6 Flash (Medium)",
|
|
quotaInfo: { remainingFraction: 0.5, resetTime: "2026-07-25T12:00:00Z" },
|
|
},
|
|
"gemini-3.6-flash-low": {
|
|
displayName: "Gemini 3.6 Flash (Low)",
|
|
quotaInfo: { remainingFraction: 0.2, resetTime: "2026-07-25T12:00:00Z" },
|
|
},
|
|
"gemini-3.5-flash-low": {
|
|
displayName: "Gemini 3.5 Flash (Medium)",
|
|
quotaInfo: { remainingFraction: 0.9, resetTime: "2026-07-25T12:00:00Z" },
|
|
},
|
|
"internal-model": {
|
|
displayName: "Internal",
|
|
isInternal: true,
|
|
quotaInfo: { remainingFraction: 0.5 },
|
|
},
|
|
},
|
|
},
|
|
text: async () => "{}",
|
|
}));
|
|
|
|
vi.mock("../../open-sse/utils/proxyFetch.js", () => ({
|
|
proxyAwareFetch,
|
|
}));
|
|
|
|
describe("Antigravity quota tracker: Gemini 3.6 Flash usage bars", () => {
|
|
beforeEach(() => proxyAwareFetch.mockClear());
|
|
|
|
it("returns Gemini 3.6 Flash tier quotas so the dashboard can render usage bars", async () => {
|
|
const { getAntigravityUsage } = await import("../../open-sse/services/usage/google.js");
|
|
|
|
const usage = await getAntigravityUsage("access-token", {});
|
|
|
|
expect(usage.quotas["gemini-3.6-flash-high"]).toMatchObject({
|
|
used: 200,
|
|
total: 1000,
|
|
remainingPercentage: 80,
|
|
displayName: "Gemini 3.6 Flash (High)",
|
|
});
|
|
expect(usage.quotas["gemini-3.6-flash-medium"]).toMatchObject({
|
|
used: 500,
|
|
total: 1000,
|
|
remainingPercentage: 50,
|
|
});
|
|
expect(usage.quotas["gemini-3.6-flash-low"]).toMatchObject({
|
|
used: 800,
|
|
total: 1000,
|
|
remainingPercentage: 20,
|
|
});
|
|
});
|
|
|
|
it("still filters out internal and non-important models", async () => {
|
|
const { getAntigravityUsage } = await import("../../open-sse/services/usage/google.js");
|
|
|
|
const usage = await getAntigravityUsage("access-token", {});
|
|
|
|
expect(usage.quotas).not.toHaveProperty("internal-model");
|
|
expect(Object.keys(usage.quotas)).toEqual([
|
|
"gemini-3.6-flash-high",
|
|
"gemini-3.6-flash-medium",
|
|
"gemini-3.6-flash-low",
|
|
"gemini-3.5-flash-low",
|
|
]);
|
|
});
|
|
});
|