Files
diegosouzapw--omniroute/tests/unit/chatcore-extracted-modules-3821.test.ts
2026-07-13 13:39:12 +08:00

111 lines
4.4 KiB
TypeScript

/**
* LEDGER-6 (#3821-review) — the chatCore modularization (#3598) relocated ~300 lines into
* open-sse/handlers/chatCore/{idempotency,sanitization,semanticCache,memorySkillsInjection}.ts
* with no direct tests at the new seam. The extraction shipped a real `ReferenceError`
* (idempotencyKey) that no test caught. These tests pin the pure/extractable pieces:
* - sanitizeChatRequestBody (token-field normalization, empty-name stripping, tool filter)
* - checkIdempotencyCache now returns { hit, idempotencyKey } so the save site reuses the
* single derivation (no dual getIdempotencyKey call).
*/
import test from "node:test";
import assert from "node:assert/strict";
import { sanitizeChatRequestBody } from "../../open-sse/handlers/chatCore/sanitization.ts";
import { checkIdempotencyCache } from "../../open-sse/handlers/chatCore/idempotency.ts";
import { FORMATS } from "../../open-sse/translator/formats.ts";
import { saveIdempotency } from "../../src/lib/idempotencyLayer.ts";
test("sanitizeChatRequestBody: Chat Completions target maps max_output_tokens → max_tokens", () => {
const out = sanitizeChatRequestBody({ max_output_tokens: 256 }, FORMATS.OPENAI, FORMATS.OPENAI);
assert.equal(out.max_tokens, 256);
assert.equal(out.max_output_tokens, undefined);
});
test("sanitizeChatRequestBody: Responses target maps max_completion_tokens → max_output_tokens", () => {
const out = sanitizeChatRequestBody(
{ max_completion_tokens: 512 },
FORMATS.OPENAI,
FORMATS.OPENAI_RESPONSES
);
assert.equal(out.max_output_tokens, 512);
assert.equal(out.max_completion_tokens, undefined);
});
test("sanitizeChatRequestBody: Responses target maps max_tokens → max_output_tokens", () => {
const out = sanitizeChatRequestBody({ max_tokens: 128 }, FORMATS.OPENAI_RESPONSES, FORMATS.OPENAI);
assert.equal(out.max_output_tokens, 128);
assert.equal(out.max_tokens, undefined);
});
test("sanitizeChatRequestBody: strips empty message name and filters nameless tools", () => {
const out = sanitizeChatRequestBody(
{
messages: [
{ role: "user", content: "hi", name: "" },
{ role: "assistant", content: "yo", name: "keepme" },
],
tools: [
{ type: "function", function: { name: "real_tool", parameters: {} } },
{ type: "function", function: { name: "" } }, // dropped — empty name
{ type: "function", function: {} }, // dropped — no name
],
},
FORMATS.OPENAI,
FORMATS.OPENAI
);
const messages = out.messages as Array<Record<string, unknown>>;
assert.ok(!("name" in messages[0]), "empty name stripped");
assert.equal(messages[1].name, "keepme", "non-empty name kept");
const tools = out.tools as Array<Record<string, unknown>>;
assert.equal(tools.length, 1, "only the named tool survives");
assert.equal((tools[0].function as Record<string, unknown>).name, "real_tool");
});
test("checkIdempotencyCache returns { hit:null, idempotencyKey } on a miss", async () => {
const headers = new Headers({ "idempotency-key": "idem-miss-3821" });
const result = await checkIdempotencyCache({
clientRawRequest: { headers },
provider: "openai",
model: "gpt-4.1",
effectiveServiceTier: undefined,
startTime: 0,
log: undefined,
});
assert.equal(result.hit, null);
assert.equal(result.idempotencyKey, "idem-miss-3821");
});
test("checkIdempotencyCache returns a hit Response reusing the same key after a save", async () => {
const key = "idem-hit-3821";
saveIdempotency(key, { object: "chat.completion", choices: [], usage: {} }, 200);
const headers = new Headers({ "idempotency-key": key });
const result = await checkIdempotencyCache({
clientRawRequest: { headers },
provider: "openai",
model: "gpt-4.1",
effectiveServiceTier: undefined,
startTime: 0,
log: undefined,
});
assert.equal(result.idempotencyKey, key, "the resolved key is returned for the save site to reuse");
assert.ok(result.hit, "a cached entry produces a hit");
assert.equal(result.hit!.response.headers.get("X-OmniRoute-Idempotent"), "true");
});
test("checkIdempotencyCache resolves a null key when no idempotency headers are present", async () => {
const result = await checkIdempotencyCache({
clientRawRequest: { headers: new Headers() },
provider: "openai",
model: "gpt-4.1",
effectiveServiceTier: undefined,
startTime: 0,
log: undefined,
});
assert.equal(result.hit, null);
assert.equal(result.idempotencyKey, null);
});