From c665b01e8979fdd9f68da97a1d0af8394dafb437 Mon Sep 17 00:00:00 2001
From: diegosouzapw <diegosouzapw@users.noreply.github.com>
Date: Wed, 1 Apr 2026 21:34:39 -0300
Subject: [PATCH] test: add usage service and response translator test scripts

---
 test-antigrav.mjs   | 43 +++++++++++++++++++++++++++++++++++++++++++
 test_translator.mjs |  9 +++++++++
 2 files changed, 52 insertions(+)
 create mode 100644 test-antigrav.mjs
 create mode 100644 test_translator.mjs

diff --git a/test-antigrav.mjs b/test-antigrav.mjs
new file mode 100644
index 0000000000..3578ee5bb6
--- /dev/null
+++ b/test-antigrav.mjs
@@ -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;
+  }
+});
diff --git a/test_translator.mjs b/test_translator.mjs
new file mode 100644
index 0000000000..e6f1d7fc4f
--- /dev/null
+++ b/test_translator.mjs
@@ -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
+  )
+);