wehub-resource-sync
96 lines
3.4 KiB
TypeScript
96 lines
3.4 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { QoderExecutor, __test__ } from "../../open-sse/executors/qoder.ts";
|
|
|
|
const { unwrapQoderEnvelope } = __test__;
|
|
|
|
function sseResponse(body: string, status = 200): Response {
|
|
return new Response(body, {
|
|
status,
|
|
headers: { "Content-Type": "text/event-stream" },
|
|
});
|
|
}
|
|
|
|
test("unwrapQoderEnvelope: surfaces an embedded non-200 statusCodeValue as a real HTTP error", async () => {
|
|
// Qoder wraps an upstream 429 inside a 200 SSE envelope. Before the fix this
|
|
// passed straight through as a 200, so combo/account fallback never fired.
|
|
const wrapped = sseResponse(
|
|
'data: {"statusCodeValue":429,"body":"rate limit exceeded"}\n\ndata: [DONE]\n\n'
|
|
);
|
|
|
|
const result = await unwrapQoderEnvelope(wrapped);
|
|
|
|
assert.equal(result.status, 429, "embedded 429 must become a real HTTP 429");
|
|
const payload = (await result.json()) as any;
|
|
assert.match(payload.error.message, /qoder error 429/);
|
|
assert.match(payload.error.message, /rate limit exceeded/);
|
|
});
|
|
|
|
test("unwrapQoderEnvelope: maps a sub-400 embedded status to 502", async () => {
|
|
const wrapped = sseResponse('data: {"statusCodeValue":302,"body":"redirect"}\n\n');
|
|
|
|
const result = await unwrapQoderEnvelope(wrapped);
|
|
|
|
assert.equal(result.status, 502);
|
|
});
|
|
|
|
test("unwrapQoderEnvelope: classifies embedded 401 as an authentication_error", async () => {
|
|
const wrapped = sseResponse('data: {"statusCodeValue":401,"body":"invalid token"}\n\n');
|
|
|
|
const result = await unwrapQoderEnvelope(wrapped);
|
|
|
|
assert.equal(result.status, 401);
|
|
const payload = (await result.json()) as any;
|
|
assert.equal(payload.error.type, "authentication_error");
|
|
});
|
|
|
|
test("unwrapQoderEnvelope: passes a successful stream through with the first chunk intact", async () => {
|
|
const wrapped = sseResponse(
|
|
'data: {"choices":[{"delta":{"content":"O"}}]}\n\ndata: {"choices":[{"delta":{"content":"K"}}]}\n\ndata: [DONE]\n\n'
|
|
);
|
|
|
|
const result = await unwrapQoderEnvelope(wrapped);
|
|
|
|
assert.equal(result.status, 200);
|
|
const body = await result.text();
|
|
// The first chunk must not be swallowed by the peek.
|
|
assert.match(body, /"content":"O"/);
|
|
assert.match(body, /"content":"K"/);
|
|
assert.match(body, /\[DONE\]/);
|
|
});
|
|
|
|
test("unwrapQoderEnvelope: an empty stream becomes a 502 error", async () => {
|
|
const result = await unwrapQoderEnvelope(sseResponse(""));
|
|
assert.equal(result.status, 502);
|
|
});
|
|
|
|
test("unwrapQoderEnvelope: a non-ok response is returned unchanged", async () => {
|
|
const errResp = sseResponse("nope", 500);
|
|
const result = await unwrapQoderEnvelope(errResp);
|
|
assert.equal(result, errResp);
|
|
});
|
|
|
|
test("QoderExecutor: stream call surfaces an embedded error envelope as a real HTTP status", async () => {
|
|
const executor = new QoderExecutor();
|
|
const originalFetch = globalThis.fetch;
|
|
globalThis.fetch = async () =>
|
|
sseResponse('data: {"statusCodeValue":429,"body":"quota exceeded"}\n\ndata: [DONE]\n\n');
|
|
|
|
try {
|
|
const { response } = await executor.execute({
|
|
model: "qoder-rome-30ba3b",
|
|
body: { messages: [{ role: "user", content: "hi" }] },
|
|
stream: true,
|
|
credentials: { apiKey: "pat_test" },
|
|
});
|
|
|
|
// Before the port this was a 200 — fallback could never trigger.
|
|
assert.equal(response.status, 429);
|
|
const payload = (await response.json()) as any;
|
|
assert.match(payload.error.message, /qoder error 429/);
|
|
} finally {
|
|
globalThis.fetch = originalFetch;
|
|
}
|
|
});
|