wehub-resource-sync
99 lines
3.7 KiB
TypeScript
99 lines
3.7 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { PollinationsExecutor } from "../../open-sse/executors/pollinations.ts";
|
|
|
|
test("#2987 PollinationsExecutor.buildUrl uses the gen.pollinations.ai gateway (not the legacy text host)", () => {
|
|
const executor = new PollinationsExecutor();
|
|
// Legacy text.pollinations.ai now 404s ("legacy API"); gen.pollinations.ai/v1
|
|
// is the current OpenAI-compatible endpoint and must be the primary.
|
|
assert.equal(
|
|
executor.buildUrl("openai", true),
|
|
"https://gen.pollinations.ai/v1/chat/completions"
|
|
);
|
|
// No legacy text.pollinations.ai endpoint should remain in the rotation.
|
|
assert.equal(
|
|
executor.buildUrl("openai", true, 1),
|
|
"https://gen.pollinations.ai/v1/chat/completions"
|
|
);
|
|
});
|
|
|
|
test("PollinationsExecutor.buildHeaders supports anonymous access and optional SSE accept", () => {
|
|
const executor = new PollinationsExecutor();
|
|
assert.deepEqual(executor.buildHeaders({}, true), {
|
|
"Content-Type": "application/json",
|
|
Accept: "text/event-stream",
|
|
});
|
|
});
|
|
|
|
test("PollinationsExecutor.buildHeaders sends API auth for the key-backed tier when configured", () => {
|
|
const executor = new PollinationsExecutor();
|
|
assert.deepEqual(executor.buildHeaders({ apiKey: "poll-key" }, true), {
|
|
"Content-Type": "application/json",
|
|
Authorization: "Bearer poll-key",
|
|
Accept: "text/event-stream",
|
|
});
|
|
});
|
|
|
|
test("PollinationsExecutor.transformRequest is a passthrough for alias models", () => {
|
|
const executor = new PollinationsExecutor();
|
|
const body = { model: "claude", messages: [{ role: "user", content: "hello" }] };
|
|
assert.equal(executor.transformRequest("claude", body, true, {}), body);
|
|
});
|
|
|
|
test("PollinationsExecutor enhances 401 errors for premium models with actionable guidance", async () => {
|
|
const executor = new PollinationsExecutor();
|
|
|
|
// Mock super.execute (BaseExecutor.prototype.execute) to throw a 401
|
|
const origBaseExec = Object.getPrototypeOf(Object.getPrototypeOf(executor)).execute;
|
|
Object.getPrototypeOf(Object.getPrototypeOf(executor)).execute = async function () {
|
|
const err = new Error("Authentication required. Please provide an API key via Authorization header (Bearer token) or ?key= query parameter.");
|
|
(err as any).status = 401;
|
|
throw err;
|
|
};
|
|
|
|
try {
|
|
await executor.execute({
|
|
model: "claude",
|
|
body: { messages: [{ role: "user", content: "hi" }] },
|
|
stream: false,
|
|
credentials: {},
|
|
});
|
|
assert.fail("Should have thrown");
|
|
} catch (err: any) {
|
|
assert.equal(err.status, 401);
|
|
assert.match(err.message, /Pollinations model "claude" requires an API key/);
|
|
assert.match(err.message, /enter\.pollinations\.ai/);
|
|
assert.match(err.message, /Free keyless models/);
|
|
} finally {
|
|
Object.getPrototypeOf(Object.getPrototypeOf(executor)).execute = origBaseExec;
|
|
}
|
|
});
|
|
|
|
test("PollinationsExecutor passes through 401 errors for non-premium models", async () => {
|
|
const executor = new PollinationsExecutor();
|
|
|
|
const origBaseExec = Object.getPrototypeOf(Object.getPrototypeOf(executor)).execute;
|
|
Object.getPrototypeOf(Object.getPrototypeOf(executor)).execute = async function () {
|
|
const err = new Error("Authentication required");
|
|
(err as any).status = 401;
|
|
throw err;
|
|
};
|
|
|
|
try {
|
|
await executor.execute({
|
|
model: "openai",
|
|
body: { messages: [{ role: "user", content: "hi" }] },
|
|
stream: false,
|
|
credentials: {},
|
|
});
|
|
assert.fail("Should have thrown");
|
|
} catch (err: any) {
|
|
assert.equal(err.status, 401);
|
|
// Free models should NOT get the enhanced message
|
|
assert.doesNotMatch(err.message, /requires an API key/);
|
|
} finally {
|
|
Object.getPrototypeOf(Object.getPrototypeOf(executor)).execute = origBaseExec;
|
|
}
|
|
});
|