test: add usage service and response translator test scripts

This commit is contained in:
diegosouzapw
2026-04-01 21:34:39 -03:00
parent 5c2db0134f
commit c665b01e89
2 changed files with 52 additions and 0 deletions

43
test-antigrav.mjs Normal file
View File

@@ -0,0 +1,43 @@
import test from "node:test";
import assert from "node:assert/strict";
const usageService = await import("./open-sse/services/usage.ts");
test("antigravity fraction logic", async () => {
const originalFetch = globalThis.fetch;
globalThis.fetch = async (url) => {
if (url.includes("loadCodeAssist")) {
return new Response(
JSON.stringify({
currentTier: { id: "pro", name: "Pro" },
}),
{ status: 200 }
);
}
if (url.includes("fetchAvailableModels")) {
return new Response(
JSON.stringify({
models: {
"gemini-1.5-pro": {
quotaInfo: {
remainingFraction: 0,
resetTime: "2025-05-01T00:00:00Z",
},
},
},
}),
{ status: 200 }
);
}
};
try {
const res = await usageService.getUsageForProvider({
provider: "antigravity",
accessToken: "test",
});
console.log(JSON.stringify(res, null, 2));
} finally {
globalThis.fetch = originalFetch;
}
});

9
test_translator.mjs Normal file
View File

@@ -0,0 +1,9 @@
import { translateNonStreamingResponse } from "./open-sse/handlers/responseTranslator.ts";
import { FORMATS } from "./open-sse/translator/formats.ts";
console.log(
translateNonStreamingResponse(
{ object: "chat.completion", choices: [] },
FORMATS.CLAUDE,
FORMATS.OPENAI
)
);