zenord
63 lines
3.3 KiB
TypeScript
63 lines
3.3 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import test from "node:test";
|
|
import { DurableSessionStore } from "../src/core/durable-session-store.js";
|
|
|
|
const identity = { botId: "test-bot", platform: "qq" };
|
|
|
|
test("persists state v2 binding across reopen with 0600 atomic file", async () => {
|
|
const dir = await fs.promises.mkdtemp(path.join(os.tmpdir(), "gori-store-"));
|
|
const file = path.join(dir, "state.json");
|
|
const first = new DurableSessionStore(file, identity);
|
|
await first.open();
|
|
await first.setBinding({ chatKey: "qq:chat", agentId: "kimi", nativeSessionId: "native-1", workspace: "/tmp", botFingerprint: "fp", createdAt: 1, updatedAt: 1 });
|
|
await first.close();
|
|
assert.equal((await fs.promises.readdir(dir)).some((name) => name.endsWith(".tmp")), false);
|
|
assert.equal((await fs.promises.stat(file)).mode & 0o777, 0o600);
|
|
const persisted = JSON.parse(await fs.promises.readFile(file, "utf8"));
|
|
assert.equal(persisted.version, 2); assert.equal(persisted.botId, "test-bot"); assert.equal(persisted.platform, "qq");
|
|
|
|
const second = new DurableSessionStore(file, identity);
|
|
await second.open();
|
|
assert.equal(second.getBinding("qq:chat")?.nativeSessionId, "native-1");
|
|
await second.close();
|
|
});
|
|
|
|
test("rejects old state and bot or platform identity mismatch without rewriting", async () => {
|
|
const dir = await fs.promises.mkdtemp(path.join(os.tmpdir(), "gori-store-id-"));
|
|
const file = path.join(dir, "state.json");
|
|
const old = '{"version":1,"bindings":{}}\n';
|
|
await fs.promises.writeFile(file, old);
|
|
await assert.rejects(new DurableSessionStore(file, identity).open(), /requires state v2/);
|
|
assert.equal(await fs.promises.readFile(file, "utf8"), old);
|
|
|
|
await fs.promises.writeFile(file, JSON.stringify({ version: 2, botId: "other", platform: "qq", bindings: {} }));
|
|
await assert.rejects(new DurableSessionStore(file, identity).open(), /identity mismatch/);
|
|
await fs.promises.writeFile(file, JSON.stringify({ version: 2, botId: "test-bot", platform: "feishu", bindings: {} }));
|
|
await assert.rejects(new DurableSessionStore(file, identity).open(), /identity mismatch/);
|
|
|
|
await fs.promises.writeFile(file, JSON.stringify({ version: 2, botId: "test-bot", platform: "qq", bindings: { "qq:chat": { chatKey: "qq:other" } } }));
|
|
await assert.rejects(new DurableSessionStore(file, identity).open(), /invalid state v2 binding/);
|
|
});
|
|
|
|
test("preserves corrupt state and refuses a second writer lock", async () => {
|
|
const dir = await fs.promises.mkdtemp(path.join(os.tmpdir(), "gori-store-bad-"));
|
|
const badFile = path.join(dir, "bad.json");
|
|
await fs.promises.writeFile(badFile, "not-json");
|
|
await assert.rejects(new DurableSessionStore(badFile, identity).open(), /original file was preserved/);
|
|
assert.equal(await fs.promises.readFile(badFile, "utf8"), "not-json");
|
|
|
|
const file = path.join(dir, "state.json");
|
|
const first = new DurableSessionStore(file, identity); await first.open();
|
|
await assert.rejects(new DurableSessionStore(file, identity).open(), /locked by another/);
|
|
await first.close();
|
|
|
|
await fs.promises.writeFile(`${file}.lock`, "2147483647\n", { mode: 0o600 });
|
|
const recovered = new DurableSessionStore(file, identity);
|
|
await recovered.open();
|
|
await recovered.close();
|
|
assert.equal(fs.existsSync(`${file}.lock`), false);
|
|
});
|