wehub-resource-sync
103 lines
4.0 KiB
TypeScript
103 lines
4.0 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
const { VertexExecutor, isExpressApiKey, looksLikeServiceAccountJson } = await import(
|
|
"../../open-sse/executors/vertex.ts"
|
|
);
|
|
|
|
test("looksLikeServiceAccountJson is true only for a JSON object credential", () => {
|
|
assert.equal(looksLikeServiceAccountJson(JSON.stringify({ project_id: "p" })), true);
|
|
assert.equal(looksLikeServiceAccountJson("express-opaque-key"), false);
|
|
assert.equal(looksLikeServiceAccountJson(JSON.stringify([1, 2, 3])), false);
|
|
assert.equal(looksLikeServiceAccountJson(""), false);
|
|
});
|
|
|
|
test("isExpressApiKey is true for a non-empty, non-JSON credential", () => {
|
|
assert.equal(isExpressApiKey("AIzaSyExpressKey"), true);
|
|
assert.equal(isExpressApiKey(" "), false);
|
|
assert.equal(isExpressApiKey(""), false);
|
|
assert.equal(isExpressApiKey(null), false);
|
|
assert.equal(isExpressApiKey(undefined), false);
|
|
assert.equal(isExpressApiKey(JSON.stringify({ project_id: "p" })), false);
|
|
});
|
|
|
|
test("buildUrl Express: streaming google model uses streamGenerateContent + ?alt=sse&key=", () => {
|
|
const executor = new VertexExecutor();
|
|
const url = executor.buildUrl("gemini-3-flash-preview", true, 0, { apiKey: "k-express" });
|
|
assert.equal(
|
|
url,
|
|
"https://aiplatform.googleapis.com/v1/publishers/google/models/gemini-3-flash-preview:streamGenerateContent?alt=sse&key=k-express"
|
|
);
|
|
});
|
|
|
|
test("buildUrl Express: non-streaming google model uses generateContent?key=", () => {
|
|
const executor = new VertexExecutor();
|
|
const url = executor.buildUrl("gemini-3-flash-preview", false, 0, { apiKey: "k-express" });
|
|
assert.equal(
|
|
url,
|
|
"https://aiplatform.googleapis.com/v1/publishers/google/models/gemini-3-flash-preview:generateContent?key=k-express"
|
|
);
|
|
});
|
|
|
|
test("buildUrl Express: the API key is URL-encoded and trimmed", () => {
|
|
const executor = new VertexExecutor();
|
|
const url = executor.buildUrl("gemini-2.5-flash", false, 0, { apiKey: " a/b+c=d " });
|
|
assert.equal(
|
|
url,
|
|
"https://aiplatform.googleapis.com/v1/publishers/google/models/gemini-2.5-flash:generateContent?key=a%2Fb%2Bc%3Dd"
|
|
);
|
|
});
|
|
|
|
test("buildUrl Express: a present accessToken takes the Service Account path, not Express", () => {
|
|
const executor = new VertexExecutor();
|
|
const url = executor.buildUrl("gemini-2.5-flash", false, 0, {
|
|
apiKey: "k-express",
|
|
accessToken: "ya29.token",
|
|
providerSpecificData: { region: "us-central1" },
|
|
});
|
|
assert.equal(
|
|
url,
|
|
"https://aiplatform.googleapis.com/v1/projects/unknown-project/locations/us-central1/publishers/google/models/gemini-2.5-flash:generateContent"
|
|
);
|
|
});
|
|
|
|
test("buildHeaders for an Express key (no accessToken) omits the Authorization header", () => {
|
|
const executor = new VertexExecutor();
|
|
const headers = executor.buildHeaders({ apiKey: "k-express" }, false);
|
|
assert.equal(headers["Content-Type"], "application/json");
|
|
assert.equal(headers.Authorization, undefined);
|
|
});
|
|
|
|
test("execute with an Express key calls the publisher endpoint directly (no OAuth token exchange)", async () => {
|
|
const executor = new VertexExecutor();
|
|
const originalFetch = globalThis.fetch;
|
|
const calls: string[] = [];
|
|
|
|
globalThis.fetch = async (url: any) => {
|
|
calls.push(String(url));
|
|
return new Response(JSON.stringify({ ok: true }), {
|
|
status: 200,
|
|
headers: { "Content-Type": "application/json" },
|
|
});
|
|
};
|
|
|
|
try {
|
|
const result = await executor.execute({
|
|
model: "gemini-3-flash-preview",
|
|
body: { contents: [{ role: "user", parts: [{ text: "hi" }] }] },
|
|
stream: false,
|
|
credentials: { apiKey: "AIzaSyExpressKey" },
|
|
} as any);
|
|
|
|
assert.equal(result.response.status, 200);
|
|
// Exactly one call — straight to Vertex, no oauth2 token mint.
|
|
assert.equal(calls.length, 1);
|
|
assert.equal(
|
|
calls[0],
|
|
"https://aiplatform.googleapis.com/v1/publishers/google/models/gemini-3-flash-preview:generateContent?key=AIzaSyExpressKey"
|
|
);
|
|
} finally {
|
|
globalThis.fetch = originalFetch;
|
|
}
|
|
});
|