mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-05 14:52:09 +03:00
test: add usage service and response translator test scripts
This commit is contained in:
43
test-antigrav.mjs
Normal file
43
test-antigrav.mjs
Normal 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
9
test_translator.mjs
Normal 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
|
||||
)
|
||||
);
|
||||
Reference in New Issue
Block a user