wehub-resource-sync
58 lines
2.5 KiB
TypeScript
58 lines
2.5 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { resolveImageBaseUrl } from "@omniroute/open-sse/handlers/imageGeneration.ts";
|
|
|
|
const GEMINI_FALLBACK =
|
|
"https://generativelanguage.googleapis.com/v1beta/openai/images/generations";
|
|
|
|
test("#3205: custom node baseUrl from providerSpecificData is used (not Gemini fallback)", () => {
|
|
const credentials = { providerSpecificData: { baseUrl: "https://example.com/v1" } };
|
|
const resolved = resolveImageBaseUrl(credentials, GEMINI_FALLBACK);
|
|
|
|
assert.ok(
|
|
resolved.startsWith("https://example.com/"),
|
|
`expected resolved URL to point to example.com, got: ${resolved}`
|
|
);
|
|
// The exact-equality assertion below already guarantees the Gemini fallback
|
|
// was not used (a substring `.includes` check trips CodeQL
|
|
// js/incomplete-url-substring-sanitization and is redundant here).
|
|
assert.equal(resolved, "https://example.com/v1/images/generations");
|
|
});
|
|
|
|
test("#3205: providerSpecificData.baseUrl wins over top-level credentials.baseUrl", () => {
|
|
const credentials = {
|
|
baseUrl: "https://toplevel.example/v1",
|
|
providerSpecificData: { baseUrl: "https://psd.example/v1" },
|
|
};
|
|
const resolved = resolveImageBaseUrl(credentials, GEMINI_FALLBACK);
|
|
assert.equal(resolved, "https://psd.example/v1/images/generations");
|
|
});
|
|
|
|
test("#3205: trailing slash on node baseUrl is normalized (no double slash)", () => {
|
|
const credentials = { providerSpecificData: { baseUrl: "https://example.com/v1/" } };
|
|
const resolved = resolveImageBaseUrl(credentials, GEMINI_FALLBACK);
|
|
assert.equal(resolved, "https://example.com/v1/images/generations");
|
|
});
|
|
|
|
test("#3205: an already-complete images URL is not double-appended", () => {
|
|
const credentials = {
|
|
providerSpecificData: { baseUrl: "https://example.com/v1/images/generations" },
|
|
};
|
|
const resolved = resolveImageBaseUrl(credentials, GEMINI_FALLBACK);
|
|
assert.equal(resolved, "https://example.com/v1/images/generations");
|
|
});
|
|
|
|
test("#3205: top-level credentials.baseUrl is honored when no providerSpecificData", () => {
|
|
const credentials = { baseUrl: "https://legacy.example/v1" };
|
|
const resolved = resolveImageBaseUrl(credentials, GEMINI_FALLBACK);
|
|
assert.equal(resolved, "https://legacy.example/v1/images/generations");
|
|
});
|
|
|
|
test("#3205: falls back to provided default when no node baseUrl present", () => {
|
|
const resolved = resolveImageBaseUrl({}, GEMINI_FALLBACK);
|
|
assert.equal(resolved, GEMINI_FALLBACK);
|
|
const resolvedNull = resolveImageBaseUrl(null, GEMINI_FALLBACK);
|
|
assert.equal(resolvedNull, GEMINI_FALLBACK);
|
|
});
|