wehub-resource-sync
82 lines
2.4 KiB
TypeScript
82 lines
2.4 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
const { handleImageGeneration } = await import(
|
|
"../../open-sse/handlers/imageGeneration.ts"
|
|
);
|
|
|
|
function restore<T>(fn: () => T): T {
|
|
const originalFetch = globalThis.fetch;
|
|
try {
|
|
return fn();
|
|
} finally {
|
|
globalThis.fetch = originalFetch;
|
|
}
|
|
}
|
|
|
|
function makeAbortError() {
|
|
return Object.create(Error.prototype, {
|
|
message: { value: "The operation was aborted", writable: true, configurable: true },
|
|
name: { value: "AbortError", writable: true, configurable: true },
|
|
});
|
|
}
|
|
|
|
test("fetch timeout in OpenAI provider path returns 504 and sanitized error", () =>
|
|
restore(async () => {
|
|
globalThis.fetch = async () => {
|
|
throw makeAbortError();
|
|
};
|
|
|
|
const result = await handleImageGeneration({
|
|
body: { model: "openai/gpt-image-2", prompt: "timeout test" },
|
|
credentials: { apiKey: "test-key" },
|
|
log: null,
|
|
});
|
|
|
|
assert.equal(result.success, false);
|
|
assert.equal(result.status, 504);
|
|
assert.match(result.error, /Image provider error:/);
|
|
}));
|
|
|
|
test("non-timeout fetch error still returns 502", () =>
|
|
restore(async () => {
|
|
globalThis.fetch = async () => {
|
|
throw new Error("network down");
|
|
};
|
|
|
|
const result = await handleImageGeneration({
|
|
body: { model: "openai/gpt-image-2", prompt: "network error test" },
|
|
credentials: { apiKey: "test-key" },
|
|
log: null,
|
|
});
|
|
|
|
assert.equal(result.success, false);
|
|
assert.equal(result.status, 502);
|
|
}));
|
|
|
|
test("successful image gen passes AbortSignal and returns URL", () =>
|
|
restore(async () => {
|
|
let seenSignal: AbortSignal | null = null;
|
|
|
|
globalThis.fetch = async (url, options) => {
|
|
seenSignal = (options as RequestInit).signal ?? null;
|
|
return new Response(
|
|
JSON.stringify({
|
|
created: 999,
|
|
data: [{ url: "https://cdn.example.com/timeout-test.png" }],
|
|
}),
|
|
{ status: 200, headers: { "content-type": "application/json" } }
|
|
);
|
|
};
|
|
|
|
const result = await handleImageGeneration({
|
|
body: { model: "openai/gpt-image-2", prompt: "success test" },
|
|
credentials: { apiKey: "test-key" },
|
|
log: null,
|
|
});
|
|
|
|
assert.equal(result.success, true);
|
|
assert.ok(seenSignal, "AbortSignal should be passed through fetchWithTimeout");
|
|
assert.equal(result.data.data[0].url, "https://cdn.example.com/timeout-test.png");
|
|
}));
|