luulam
7e45ead2ac
Resolved conflicts taking origin/master (v0.5.55) as canonical, with local features re-applied: - runtime log level (LOG_LEVEL env + dashboard Settings → Logging, applied immediately and persisted across restarts) - free/noAuth provider enable/disable toggle via providerStrategies.enabled - parallel model testing (Test All Models / Test Selected Keys)
80 lines
2.6 KiB
JavaScript
80 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",
|
|
]);
|
|
});
|
|
});
|