wehub-resource-sync
46 lines
1.8 KiB
TypeScript
46 lines
1.8 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import { createRequire } from "node:module";
|
|
|
|
const require = createRequire(import.meta.url);
|
|
const { resolveForwardTarget, isCloudcodeEnvelope, CHAT_PATH, ANTIGRAVITY_PATH } = require(
|
|
"../../src/mitm/_internal/forwardTarget.cjs"
|
|
);
|
|
|
|
const BASE = "http://localhost:20128";
|
|
|
|
test("cloudcode envelope routes to the antigravity endpoint", () => {
|
|
const body = {
|
|
model: "gemini-2.5-pro",
|
|
project: "projects/test",
|
|
request: { contents: [{ role: "user", parts: [{ text: "oi" }] }] },
|
|
};
|
|
assert.equal(isCloudcodeEnvelope(body), true);
|
|
const t = resolveForwardTarget(BASE, body);
|
|
assert.equal(t.url, `${BASE}${ANTIGRAVITY_PATH}`);
|
|
assert.equal(t.format, "antigravity");
|
|
});
|
|
|
|
test("plain OpenAI body routes to chat/completions", () => {
|
|
const body = { model: "gemini-2.5-pro", messages: [{ role: "user", content: "oi" }] };
|
|
assert.equal(isCloudcodeEnvelope(body), false);
|
|
const t = resolveForwardTarget(BASE, body);
|
|
assert.equal(t.url, `${BASE}${CHAT_PATH}`);
|
|
assert.equal(t.format, "openai");
|
|
});
|
|
|
|
test("base url trailing slash is trimmed", () => {
|
|
const t = resolveForwardTarget("http://localhost:20128/", { messages: [] });
|
|
assert.equal(t.url, `http://localhost:20128${CHAT_PATH}`);
|
|
});
|
|
|
|
test("non-envelope shapes are not misclassified as cloudcode", () => {
|
|
// request present but contents missing / wrong type
|
|
assert.equal(isCloudcodeEnvelope({ request: {} }), false);
|
|
assert.equal(isCloudcodeEnvelope({ request: { contents: "nope" } }), false);
|
|
assert.equal(isCloudcodeEnvelope({ contents: [{ parts: [] }] }), false); // bare gemini, no envelope
|
|
assert.equal(isCloudcodeEnvelope(null), false);
|
|
assert.equal(isCloudcodeEnvelope([]), false);
|
|
assert.equal(isCloudcodeEnvelope("string"), false);
|
|
});
|