wehub-resource-sync
47 lines
1.9 KiB
TypeScript
47 lines
1.9 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
|
|
const tokenizedRequest = await import("../../src/app/api/v1/vscode/[token]/tokenizedRequest.ts");
|
|
const rawTokenizedRequest =
|
|
await import("../../src/app/api/v1/vscode/raw/[token]/tokenizedRequest.ts");
|
|
|
|
test("vscode tokenized request helper infers grouped path token", () => {
|
|
const request = new Request("http://localhost/api/v1/vscode/sk-grouped/models");
|
|
const rewritten = tokenizedRequest.withPathTokenApiKey(request);
|
|
|
|
assert.equal(rewritten.headers.get("x-api-key"), "sk-grouped");
|
|
assert.equal(rewritten.headers.get("authorization"), "Bearer sk-grouped");
|
|
});
|
|
|
|
test("vscode tokenized request helper infers raw path token", () => {
|
|
const request = new Request("http://localhost/api/v1/vscode/raw/sk-raw/models");
|
|
const rewritten = rawTokenizedRequest.withPathTokenApiKey(request);
|
|
|
|
assert.equal(rewritten.headers.get("x-api-key"), "sk-raw");
|
|
assert.equal(rewritten.headers.get("authorization"), "Bearer sk-raw");
|
|
});
|
|
|
|
test("vscode tokenized request helper preserves existing auth headers", () => {
|
|
const request = new Request("http://localhost/api/v1/vscode/sk-path/models", {
|
|
headers: {
|
|
authorization: "Bearer existing",
|
|
"x-api-key": "sk-existing",
|
|
},
|
|
});
|
|
const rewritten = tokenizedRequest.withPathTokenApiKey(request, "sk-explicit");
|
|
|
|
assert.equal(rewritten.headers.get("x-api-key"), "sk-existing");
|
|
assert.equal(rewritten.headers.get("authorization"), "Bearer existing");
|
|
});
|
|
|
|
test("vscode tokenized request helper carries POST bodies", async () => {
|
|
const request = new Request("http://localhost/api/v1/vscode/raw/sk-raw/chat/completions", {
|
|
method: "POST",
|
|
body: JSON.stringify({ model: "test" }),
|
|
});
|
|
const rewritten = rawTokenizedRequest.withPathTokenApiKey(request);
|
|
|
|
assert.equal(rewritten.headers.get("x-api-key"), "sk-raw");
|
|
assert.deepEqual(await rewritten.json(), { model: "test" });
|
|
});
|