110 lines
4.2 KiB
TypeScript
110 lines
4.2 KiB
TypeScript
import { test, expect } from "vitest";
|
|
import { spawn } from "node:child_process";
|
|
import { mkdtempSync, writeFileSync, rmSync } from "node:fs";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import path from "node:path";
|
|
import { createPiModelLister } from "../src/pi/list-models.js";
|
|
import { loadConfig } from "../src/config.js";
|
|
|
|
const FAKE = path.resolve("../harness/tests/fake_pi/fake_pi_rpc.mjs");
|
|
|
|
function scriptWith(models: unknown[]): string {
|
|
const dir = mkdtempSync(join(tmpdir(), "tht-models-"));
|
|
const file = join(dir, "models.json");
|
|
writeFileSync(file, JSON.stringify({ available_models: models }));
|
|
return file;
|
|
}
|
|
|
|
test("createPiModelLister returns mapped PiModel[] from get_available_models", async () => {
|
|
const script = scriptWith([
|
|
{ provider: "zai", id: "glm-5.2", name: "GLM 5.2", reasoning: true, extra: "ignored" },
|
|
{ provider: "anthropic", id: "claude-opus-4-8", name: "Claude Opus 4.8", reasoning: true },
|
|
]);
|
|
try {
|
|
const lister = createPiModelLister(loadConfig({ THT_HARNESS_DIR: "../harness" }), {
|
|
spawnFn: () => spawn("node", [FAKE, script]) as any,
|
|
});
|
|
const models = await lister();
|
|
expect(models).toEqual([
|
|
{ provider: "zai", id: "glm-5.2", name: "GLM 5.2", reasoning: true },
|
|
{ provider: "anthropic", id: "claude-opus-4-8", name: "Claude Opus 4.8", reasoning: true },
|
|
]);
|
|
} finally {
|
|
rmSync(path.dirname(script), { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("createPiModelLister caches within ttl (spawns once for two calls)", async () => {
|
|
const script = scriptWith([{ provider: "zai", id: "glm-5.2", name: "GLM 5.2", reasoning: true }]);
|
|
try {
|
|
let spawns = 0;
|
|
const lister = createPiModelLister(loadConfig({ THT_HARNESS_DIR: "../harness" }), {
|
|
spawnFn: () => { spawns++; return spawn("node", [FAKE, script]) as any; },
|
|
ttlMs: 10_000,
|
|
nowMs: () => 1000,
|
|
});
|
|
await lister();
|
|
await lister();
|
|
expect(spawns).toBe(1);
|
|
} finally {
|
|
rmSync(path.dirname(script), { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("production model-list spawn preserves PATH and passes the portable data root", async () => {
|
|
const script = scriptWith([]);
|
|
const calls: any[][] = [];
|
|
const previousPath = process.env.PATH;
|
|
process.env.PATH = "/usr/local/bin:/usr/bin";
|
|
try {
|
|
const lister = createPiModelLister(loadConfig({
|
|
THT_HARNESS_DIR: "/app/harness",
|
|
PI_BIN: "/usr/local/bin/pi",
|
|
THT_DATA_ROOT: "/data",
|
|
}), {
|
|
spawnFn: (...args: any[]) => {
|
|
calls.push(args);
|
|
return spawn("node", [FAKE, script]) as any;
|
|
},
|
|
});
|
|
await lister();
|
|
expect(calls[0][0]).toBe("/usr/local/bin/pi");
|
|
expect(calls[0][1]).toEqual(["--mode", "rpc"]);
|
|
expect(calls[0][2]).toMatchObject({
|
|
cwd: "/app/harness",
|
|
env: expect.objectContaining({ PATH: "/usr/local/bin:/usr/bin", THT_DATA_ROOT: "/data" }),
|
|
});
|
|
} finally {
|
|
if (previousPath === undefined) delete process.env.PATH;
|
|
else process.env.PATH = previousPath;
|
|
rmSync(path.dirname(script), { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("model-list spawn omits ambient THT_DATA_ROOT when config does not provide one", async () => {
|
|
const script = scriptWith([]);
|
|
const calls: any[][] = [];
|
|
const previousDataRoot = process.env.THT_DATA_ROOT;
|
|
const previousCredential = process.env.PI_PROVIDER_API_KEY;
|
|
process.env.THT_DATA_ROOT = "/ambient-must-not-leak";
|
|
process.env.PI_PROVIDER_API_KEY = "still-inherited";
|
|
try {
|
|
const lister = createPiModelLister(loadConfig({ PI_BIN: "/usr/local/bin/pi" }), {
|
|
spawnFn: (...args: any[]) => {
|
|
calls.push(args);
|
|
return spawn("node", [FAKE, script]) as any;
|
|
},
|
|
});
|
|
await lister();
|
|
expect(calls[0][2].env).not.toHaveProperty("THT_DATA_ROOT");
|
|
expect(calls[0][2].env.PI_PROVIDER_API_KEY).toBe("still-inherited");
|
|
} finally {
|
|
if (previousDataRoot === undefined) delete process.env.THT_DATA_ROOT;
|
|
else process.env.THT_DATA_ROOT = previousDataRoot;
|
|
if (previousCredential === undefined) delete process.env.PI_PROVIDER_API_KEY;
|
|
else process.env.PI_PROVIDER_API_KEY = previousCredential;
|
|
rmSync(path.dirname(script), { recursive: true, force: true });
|
|
}
|
|
});
|