Initial commit

tests/session-persist.test.ts

@@ -0,0 +1,115 @@
+import { describe, expect, it, beforeAll } from "vitest";
+import { mkdtempSync } from "node:fs";
+import { tmpdir } from "node:os";
+import path from "node:path";
+import { randomUUID } from "node:crypto";
+import { createSession } from "../src/runtime/phase-machine.js";
+import type { PersistedBookSession } from "../src/types/book-session.js";
+
+describe("session persistence", () => {
+  beforeAll(() => {
+    process.env.WRITING_AGENT_DATA_DIR = mkdtempSync(
+      path.join(tmpdir(), "wa-session-persist-"),
+    );
+  });
+
+  it("saveBookSession / loadBookSession roundtrip", async () => {
+    const { createBook } = await import("../src/book/store.js");
+    const { saveBookSession, loadBookSession } = await import(
+      "../src/book/session-store.js"
+    );
+
+    const book = createBook({ title: "持久化测试" });
+    const sessionId = randomUUID();
+    const runtimeSession = createSession("default");
+    runtimeSession.id = sessionId;
+    runtimeSession.phase = "running";
+    runtimeSession.slots = { startupCompleted: true, userGoal: "测试目标" };
+
+    const snapshot: PersistedBookSession = {
+      version: 1,
+      sessionId,
+      bookId: book.id,
+      orchestratorId: "basic",
+      runtimeSession,
+      blackboardItems: [
+        {
+          id: randomUUID(),
+          tag: "测试.条目",
+          content: "hello",
+          source: "worker",
+          createdAt: new Date().toISOString(),
+          updatedAt: new Date().toISOString(),
+        },
+      ],
+      messages: [
+        {
+          id: randomUUID(),
+          role: "user",
+          text: "用户消息",
+          createdAt: new Date().toISOString(),
+        },
+      ],
+      savedAt: new Date().toISOString(),
+    };
+
+    saveBookSession(snapshot);
+    const loaded = loadBookSession(book.id);
+    expect(loaded).not.toBeNull();
+    expect(loaded!.sessionId).toBe(sessionId);
+    expect(loaded!.runtimeSession.phase).toBe("running");
+    expect(loaded!.runtimeSession.slots.userGoal).toBe("测试目标");
+    expect(loaded!.blackboardItems).toHaveLength(1);
+    expect(loaded!.messages[0].text).toBe("用户消息");
+  });
+
+  it("openBook restores persisted session after restart", async () => {
+    const { createBook, getBook } = await import("../src/book/store.js");
+    const { SessionManager } = await import("../src/server/session-manager.js");
+
+    const book = createBook({ title: "续作测试" });
+
+    const mgr1 = new SessionManager();
+    const view1 = await mgr1.createForBook(book.id, "basic");
+    await mgr1.sendMessage(view1.id, "测试输入");
+
+    const mgr2 = new SessionManager();
+    const view2 = await mgr2.openBook(book.id);
+
+    expect(view2.id).toBe(view1.id);
+    expect(view2.resumed).toBe(true);
+    expect(view2.hints.some((h) => h.includes("恢复"))).toBe(true);
+    expect(view2.messages.some((m) => m.text === "测试输入")).toBe(true);
+    expect(getBook(book.id)?.activeSessionId).toBe(view1.id);
+  });
+
+  it("deleteBook removes session snapshot", async () => {
+    const { createBook, deleteBook } = await import("../src/book/store.js");
+    const { hasBookSession } = await import("../src/book/session-store.js");
+    const { SessionManager } = await import("../src/server/session-manager.js");
+
+    const book = createBook({ title: "删除测试" });
+
+    const mgr = new SessionManager();
+    await mgr.createForBook(book.id, "basic");
+    expect(hasBookSession(book.id)).toBe(true);
+
+    deleteBook(book.id);
+    expect(hasBookSession(book.id)).toBe(false);
+  });
+
+  it("createForBook captures startup inquiry in messages", async () => {
+    const { createBook, deleteBook } = await import("../src/book/store.js");
+    const { SessionManager } = await import("../src/server/session-manager.js");
+
+    const book = createBook({ title: "启动询问" });
+
+    const mgr = new SessionManager();
+    const view = await mgr.createForBook(book.id, "roleplay-game-theory");
+    expect(view.waitingReason?.kind).toBe("intake");
+    expect(view.messages.some((m) => m.text?.includes("角色扮演博弈"))).toBe(true);
+    expect(view.messages.some((m) => m.kind === "orchestrator_prompt")).toBe(true);
+
+    deleteBook(book.id);
+  });
+});