sscoble/pi-mono
1
0
Fork 0
mirror of https://github.com/badlogic/pi-mono.git synced 2026-11-05 23:37:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix(coding-agent): make model reload test deterministic

Browse source
This commit is contained in:
Christian Klotz 2026-08-01 22:53:41 +03:00
parent 6e48c10f5b
commit 6374a95fb2
No known key found for this signature in database
1 changed files with 18 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

30
packages/coding-agent/test/suite/regressions/6999-models-json-hot-reload.test.ts
View file

@ -1,12 +1,15 @@
import { writeFileSync } from "node:fs";
import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { setKeybindings, type TUI } from "@earendil-works/pi-tui";
import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
import { AuthStorage } from "../../../src/core/auth-storage.ts";
import { KeybindingsManager } from "../../../src/core/keybindings.ts";
import { SettingsManager } from "../../../src/core/settings-manager.ts";
import { ModelSelectorComponent } from "../../../src/modes/interactive/components/model-selector.ts";
import { initTheme } from "../../../src/modes/interactive/theme/theme.ts";
import { stripAnsi } from "../../../src/utils/ansi.ts";
import { createHarness, type Harness } from "../harness.ts";
import { createModelRegistry, getModelRuntime } from "../../model-runtime-test-utils.ts";
function createFakeTui(): TUI {
return {
@ -28,7 +31,7 @@ function modelsJson(provider: string, model: string): Record<string, unknown> {
}
describe("issue #6999 models.json hot reload", () => {
let harness: Harness | undefined;
let tempDir: string | undefined;
beforeAll(() => {
initTheme("dark");
@ -39,20 +42,23 @@ describe("issue #6999 models.json hot reload", () => {
});
afterEach(() => {
harness?.cleanup();
harness = undefined;
if (tempDir) rmSync(tempDir, { recursive: true, force: true });
tempDir = undefined;
});
it("reloads models.json when opening /model", async () => {
harness = await createHarness({ modelsJson: modelsJson("old-provider", "old-model") });
expect(harness.session.modelRuntime.getModel("old-provider", "old-model")).toBeDefined();
tempDir = mkdtempSync(join(tmpdir(), "pi-models-json-hot-reload-"));
const modelsPath = join(tempDir, "models.json");
writeFileSync(modelsPath, JSON.stringify(modelsJson("old-provider", "old-model")));
const modelRuntime = getModelRuntime(await createModelRegistry(AuthStorage.inMemory(), modelsPath));
expect(modelRuntime.getModel("old-provider", "old-model")).toBeDefined();
writeFileSync(join(harness.tempDir, "models.json"), JSON.stringify(modelsJson("new-provider", "new-model")));
writeFileSync(modelsPath, JSON.stringify(modelsJson("new-provider", "new-model")));
const selector = new ModelSelectorComponent(
createFakeTui(),
harness.getModel(),
harness.settingsManager,
harness.session.modelRuntime,
undefined,
SettingsManager.inMemory(),
modelRuntime,
[],
() => {},
() => {},
@ -63,6 +69,6 @@ describe("issue #6999 models.json hot reload", () => {
expect(rendered).toContain("new-model [new-provider]");
expect(rendered).toContain("Model catalogs refreshed.");
});
expect(harness.session.modelRuntime.getModel("old-provider", "old-model")).toBeUndefined();
expect(modelRuntime.getModel("old-provider", "old-model")).toBeUndefined();
});
});