68 lines
2.3 KiB
TypeScript
68 lines
2.3 KiB
TypeScript
import fs from "node:fs/promises";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import { afterEach, describe, expect, it } from "vitest";
|
|
import {
|
|
readDiscordModelPickerRecentModels,
|
|
recordDiscordModelPickerRecentModel,
|
|
} from "./model-picker-preferences.js";
|
|
|
|
const tempDirs: string[] = [];
|
|
|
|
async function createStateEnv(): Promise<NodeJS.ProcessEnv> {
|
|
const dir = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-model-picker-"));
|
|
tempDirs.push(dir);
|
|
return { ...process.env, OPENCLAW_STATE_DIR: dir };
|
|
}
|
|
|
|
afterEach(async () => {
|
|
await Promise.all(
|
|
tempDirs.splice(0).map(async (dir) => {
|
|
await fs.rm(dir, { recursive: true, force: true });
|
|
}),
|
|
);
|
|
});
|
|
|
|
describe("discord model picker preferences", () => {
|
|
it("records recent models in recency order without duplicates", async () => {
|
|
const env = await createStateEnv();
|
|
const scope = { userId: "123" };
|
|
|
|
await recordDiscordModelPickerRecentModel({ env, scope, modelRef: "openai/gpt-4o" });
|
|
await recordDiscordModelPickerRecentModel({ env, scope, modelRef: "openai/gpt-4.1" });
|
|
await recordDiscordModelPickerRecentModel({ env, scope, modelRef: "openai/gpt-4o" });
|
|
|
|
const recent = await readDiscordModelPickerRecentModels({ env, scope });
|
|
expect(recent).toEqual(["openai/gpt-4o", "openai/gpt-4.1"]);
|
|
});
|
|
|
|
it("filters recent models using an allowlist", async () => {
|
|
const env = await createStateEnv();
|
|
const scope = { userId: "456" };
|
|
|
|
await recordDiscordModelPickerRecentModel({ env, scope, modelRef: "openai/gpt-4o" });
|
|
await recordDiscordModelPickerRecentModel({ env, scope, modelRef: "openai/gpt-4.1" });
|
|
|
|
const recent = await readDiscordModelPickerRecentModels({
|
|
env,
|
|
scope,
|
|
allowedModelRefs: new Set(["openai/gpt-4.1"]),
|
|
});
|
|
expect(recent).toEqual(["openai/gpt-4.1"]);
|
|
});
|
|
|
|
it("falls back to an empty store when the file is corrupt", async () => {
|
|
const env = await createStateEnv();
|
|
const stateDir = env.OPENCLAW_STATE_DIR as string;
|
|
const filePath = path.join(stateDir, "discord", "model-picker-preferences.json");
|
|
await fs.mkdir(path.dirname(filePath), { recursive: true });
|
|
await fs.writeFile(filePath, "{not-json", "utf-8");
|
|
|
|
const recent = await readDiscordModelPickerRecentModels({
|
|
env,
|
|
scope: { userId: "789" },
|
|
});
|
|
expect(recent).toEqual([]);
|
|
});
|
|
});
|