21 lines
1003 B
TypeScript
21 lines
1003 B
TypeScript
import { expect, it } from "bun:test";
|
|
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { SessionUiStateStore } from "../../src/sessions/uiStateStore.js";
|
|
|
|
it("restores plan history independently of chat messages after reconnecting", async () => {
|
|
const directory = await mkdtemp(join(tmpdir(), "tjwater-plan-state-"));
|
|
try {
|
|
const store = new SessionUiStateStore(directory);
|
|
await store.initialize();
|
|
const executionPlans = [{ revision: 1, nodes: [{ id: "sampling", status: "waiting" }] }];
|
|
await store.write({ sessionId: "session-1" }, { sessionId: "session-1", messages: [], executionPlans });
|
|
const reloaded = await new SessionUiStateStore(directory).read({ sessionId: "session-1" });
|
|
expect(reloaded?.executionPlans).toEqual(executionPlans);
|
|
expect(await store.read({ sessionId: "other-session" })).toBeNull();
|
|
} finally {
|
|
await rm(directory, { recursive: true, force: true });
|
|
}
|
|
});
|