Files
OmniRoute/tests/unit/chatcore-upstream-execute-headers.test.ts
Diego Rodrigues de Sa e Souza ee24eb52d4 Release v3.8.33 (#4515)
Release v3.8.33 — full CHANGELOG in the PR body. Blocking gates green (Build, Lint, Unit Tests 8/8, Package Artifact, Quality Gates, Quality Ratchet, Docs Sync, PR Test Policy, test-vitest). Admin-merged over a Node 26 future-compat timer flake (1/4) + an E2E UI flake (3/9) — both verified non-deterministic; full test:unit validated locally (16936 pass).
2026-06-22 03:17:02 -03:00

71 lines
2.9 KiB
TypeScript

// tests/unit/chatcore-upstream-execute-headers.test.ts
// Characterization of buildUpstreamHeadersForExecute — the per-model upstream extra-header builder
// extracted from handleChatCore (chatCore god-file decomposition, #3501). Locks: the
// connection custom User-Agent override, the Claude Fast Mode opt-in (claude provider + enabled
// settings + supported model only), and the modelToCall===effectiveModel vs alias branches.
import { test } from "node:test";
import assert from "node:assert/strict";
import { buildUpstreamHeadersForExecute } from "../../open-sse/handlers/chatCore/upstreamExecuteHeaders.ts";
import { CPA_FORCE_FAST_MODE_HEADER } from "../../src/lib/providers/claudeFastMode.ts";
const base = {
modelToCall: "some-model",
effectiveModel: "some-model",
provider: "openai",
model: "some-model",
resolvedModel: "some-model",
sourceFormat: "openai",
connectionCustomUserAgent: "",
settings: {},
};
test("no custom UA / non-claude / no fast settings → no UA and no fast-mode header", () => {
const h = buildUpstreamHeadersForExecute({ ...base });
assert.equal(h["User-Agent"], undefined);
assert.equal(h[CPA_FORCE_FAST_MODE_HEADER], undefined);
});
test("connection custom User-Agent overrides the upstream User-Agent", () => {
const h = buildUpstreamHeadersForExecute({ ...base, connectionCustomUserAgent: "MyAgent/1.0" });
assert.equal(h["User-Agent"], "MyAgent/1.0");
});
test("claude provider + enabled fast-mode settings + supported model → CPA header set", () => {
const h = buildUpstreamHeadersForExecute({
...base,
provider: "claude",
modelToCall: "claude-fast-x",
effectiveModel: "claude-fast-x",
settings: { claudeFastMode: { enabled: true, supportedModels: ["claude-fast-x"] } },
});
assert.equal(h[CPA_FORCE_FAST_MODE_HEADER], "1");
});
test("fast-mode header is NOT set when the model is not in the supported list", () => {
const h = buildUpstreamHeadersForExecute({
...base,
provider: "claude",
modelToCall: "claude-other",
effectiveModel: "claude-other",
settings: { claudeFastMode: { enabled: true, supportedModels: ["claude-fast-x"] } },
});
assert.equal(h[CPA_FORCE_FAST_MODE_HEADER], undefined);
});
test("fast-mode header is NOT set for non-claude providers even with fast settings", () => {
const h = buildUpstreamHeadersForExecute({
...base,
provider: "openai",
modelToCall: "claude-fast-x",
effectiveModel: "claude-fast-x",
settings: { claudeFastMode: { enabled: true, supportedModels: ["claude-fast-x"] } },
});
assert.equal(h[CPA_FORCE_FAST_MODE_HEADER], undefined);
});
test("returns a plain object (no per-model extra headers configured for unknown models)", () => {
const h = buildUpstreamHeadersForExecute({ ...base, modelToCall: "totally-unknown", effectiveModel: "x" });
assert.equal(typeof h, "object");
assert.equal(h[CPA_FORCE_FAST_MODE_HEADER], undefined);
});