Merge remote-tracking branch 'origin/master' into gitea/feature/end
Resolved conflicts taking origin/master (v0.5.55) as canonical, with local features re-applied: - runtime log level (LOG_LEVEL env + dashboard Settings → Logging, applied immediately and persisted across restarts) - free/noAuth provider enable/disable toggle via providerStrategies.enabled - parallel model testing (Test All Models / Test Selected Keys)
This commit is contained in:
147
tests/unit/qoder-billing.test.js
Normal file
147
tests/unit/qoder-billing.test.js
Normal file
@@ -0,0 +1,147 @@
|
||||
/**
|
||||
* Unit tests for qoder billing error detection.
|
||||
*
|
||||
* Ensures that billing blocks (code 112, 10605, pricingUrl) are detected
|
||||
* on the first SSE frame and returned as 403 responses so chatCore can
|
||||
* mark the connection unavailable and trigger combo failover.
|
||||
*/
|
||||
|
||||
import { describe, it, expect } from "vitest";
|
||||
import { __test__ as qoderExecutorInternals } from "../../open-sse/executors/qoder.js";
|
||||
|
||||
describe("isBillingBlock", () => {
|
||||
const { isBillingBlock } = qoderExecutorInternals;
|
||||
|
||||
it("detects code 112 (quota exhausted)", () => {
|
||||
const msg = '{"code":"112","message":"Quota exhausted","pricingUrl":"..."}';
|
||||
expect(isBillingBlock(msg)).toBe(true);
|
||||
});
|
||||
|
||||
it("detects code 10605 (queue throttle)", () => {
|
||||
const msg = '{"code":"10605","message":"Queue limit"}';
|
||||
expect(isBillingBlock(msg)).toBe(true);
|
||||
});
|
||||
|
||||
it("detects pricingUrl field", () => {
|
||||
const msg = '{"message":"Upgrade required","pricingUrl":"https://..."}';
|
||||
expect(isBillingBlock(msg)).toBe(true);
|
||||
});
|
||||
|
||||
it("returns false for normal errors without billing markers", () => {
|
||||
const msg = '{"code":"500","message":"Internal error"}';
|
||||
expect(isBillingBlock(msg)).toBe(false);
|
||||
});
|
||||
|
||||
it("returns false for empty or non-string input", () => {
|
||||
expect(isBillingBlock("")).toBe(false);
|
||||
expect(isBillingBlock(null)).toBe(false);
|
||||
expect(isBillingBlock(undefined)).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
describe("wrapQoderSSE billing detection", () => {
|
||||
const { wrapQoderSSE } = qoderExecutorInternals;
|
||||
|
||||
function makeResponse(lines, { status = 200 } = {}) {
|
||||
const body = new ReadableStream({
|
||||
start(controller) {
|
||||
const encoder = new TextEncoder();
|
||||
for (const line of lines) controller.enqueue(encoder.encode(line));
|
||||
controller.close();
|
||||
},
|
||||
});
|
||||
return new Response(body, { status });
|
||||
}
|
||||
|
||||
it("returns 403 response when first frame is billing block (code 112)", async () => {
|
||||
const billingEnv = JSON.stringify({
|
||||
statusCodeValue: 403,
|
||||
body: '{"code":"112","message":"Quota exhausted","pricingUrl":"https://qoder.sh/pricing"}',
|
||||
});
|
||||
const upstream = `data: ${billingEnv}\n\n`;
|
||||
|
||||
const wrapped = await wrapQoderSSE(makeResponse([upstream]), "qoder/ultimate");
|
||||
|
||||
expect(wrapped.status).toBe(403);
|
||||
expect(wrapped.ok).toBe(false);
|
||||
const json = await wrapped.json();
|
||||
expect(json.error).toBeDefined();
|
||||
expect(json.error.message).toContain("112");
|
||||
});
|
||||
|
||||
it("returns 403 response when first frame is billing block (code 10605)", async () => {
|
||||
const billingEnv = JSON.stringify({
|
||||
statusCodeValue: 429,
|
||||
body: '{"code":"10605","message":"Queue limit"}',
|
||||
});
|
||||
const upstream = `data: ${billingEnv}\n\n`;
|
||||
|
||||
const wrapped = await wrapQoderSSE(makeResponse([upstream]), "qoder/ultimate");
|
||||
|
||||
expect(wrapped.status).toBe(403);
|
||||
expect(wrapped.ok).toBe(false);
|
||||
});
|
||||
|
||||
it("returns 403 response when first frame has pricingUrl", async () => {
|
||||
const billingEnv = JSON.stringify({
|
||||
statusCodeValue: 402,
|
||||
body: '{"message":"Payment required","pricingUrl":"https://..."}',
|
||||
});
|
||||
const upstream = `data: ${billingEnv}\n\n`;
|
||||
|
||||
const wrapped = await wrapQoderSSE(makeResponse([upstream]), "qoder/ultimate");
|
||||
|
||||
expect(wrapped.status).toBe(403);
|
||||
});
|
||||
|
||||
it("passes through normal errors (non-billing) as wrapped SSE", async () => {
|
||||
const errorEnv = JSON.stringify({
|
||||
statusCodeValue: 500,
|
||||
body: "Internal server error",
|
||||
});
|
||||
const upstream = `data: ${errorEnv}\n\n`;
|
||||
|
||||
const wrapped = await wrapQoderSSE(makeResponse([upstream]), "qoder/ultimate");
|
||||
|
||||
// Normal error: still 200 response, error text in SSE body
|
||||
expect(wrapped.status).toBe(200);
|
||||
expect(wrapped.ok).toBe(true);
|
||||
|
||||
const reader = wrapped.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buf = "";
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
buf += decoder.decode(value, { stream: true });
|
||||
}
|
||||
buf += decoder.decode();
|
||||
|
||||
expect(buf).toContain("[qoder error 500");
|
||||
expect(buf).toContain("data: [DONE]");
|
||||
});
|
||||
|
||||
it("passes through successful responses unchanged", async () => {
|
||||
const inner = JSON.stringify({ choices: [{ delta: { content: "hello" } }] });
|
||||
const successEnv = JSON.stringify({ statusCodeValue: 200, body: inner });
|
||||
const upstream = `data: ${successEnv}\n\n`;
|
||||
|
||||
const wrapped = await wrapQoderSSE(makeResponse([upstream]), "qoder/ultimate");
|
||||
|
||||
expect(wrapped.status).toBe(200);
|
||||
expect(wrapped.ok).toBe(true);
|
||||
|
||||
const reader = wrapped.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buf = "";
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
buf += decoder.decode(value, { stream: true });
|
||||
}
|
||||
buf += decoder.decode();
|
||||
|
||||
expect(buf).toContain(`data: ${inner}`);
|
||||
expect(buf).toContain("data: [DONE]");
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user