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)
31 lines
1.0 KiB
JavaScript
31 lines
1.0 KiB
JavaScript
import { describe, it, expect, 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: {} },
|
|
text: async () => "{}",
|
|
}));
|
|
|
|
vi.mock("../../open-sse/utils/proxyFetch.js", () => ({
|
|
proxyAwareFetch,
|
|
}));
|
|
|
|
describe("Antigravity usage headers", () => {
|
|
beforeEach(() => proxyAwareFetch.mockClear());
|
|
|
|
it("uses the official IDE user agent and omits router-only source headers", async () => {
|
|
const { getAntigravityUsage } = await import("../../open-sse/services/usage/google.js");
|
|
|
|
await getAntigravityUsage("access-token", {});
|
|
|
|
expect(proxyAwareFetch).toHaveBeenCalledTimes(2);
|
|
for (const [, options] of proxyAwareFetch.mock.calls) {
|
|
expect(options.headers["User-Agent"]).toBe("antigravity/ide/2.1.1 darwin/arm64");
|
|
expect(options.headers).not.toHaveProperty("x-request-source");
|
|
}
|
|
});
|
|
});
|