mac
62 lines
2.6 KiB
JavaScript
62 lines
2.6 KiB
JavaScript
const test = require("node:test");
|
|
const assert = require("node:assert/strict");
|
|
const fs = require("node:fs");
|
|
const os = require("node:os");
|
|
const path = require("node:path");
|
|
const Module = require("node:module");
|
|
|
|
const DATA_DIR_MODULE = path.resolve(__dirname, "../electron/lib/codex_data_dir.js");
|
|
const SESSIONS_MODULE = path.resolve(__dirname, "../electron/lib/sessions.js");
|
|
|
|
function loadModules(userData) {
|
|
const originalLoad = Module._load;
|
|
Module._load = function (request, parent, isMain) {
|
|
if (request === "electron") return { app: { getPath: () => userData } };
|
|
return originalLoad.call(this, request, parent, isMain);
|
|
};
|
|
delete require.cache[DATA_DIR_MODULE];
|
|
delete require.cache[SESSIONS_MODULE];
|
|
try {
|
|
return { dataDir: require(DATA_DIR_MODULE), sessions: require(SESSIONS_MODULE) };
|
|
} finally {
|
|
Module._load = originalLoad;
|
|
}
|
|
}
|
|
|
|
function writeRollout(root, id, title) {
|
|
const dir = path.join(root, "sessions", "2026", "08", "06");
|
|
fs.mkdirSync(dir, { recursive: true });
|
|
fs.writeFileSync(path.join(dir, "rollout-test.jsonl"), [
|
|
JSON.stringify({ type: "session_meta", payload: { session_id: id, cwd: "/work/demo", timestamp: "2026-08-06T10:00:00.000Z" } }),
|
|
JSON.stringify({ type: "event_msg", payload: { type: "user_message", message: title } }),
|
|
].join("\n") + "\n");
|
|
}
|
|
|
|
test("会话数据目录切换后列表读取新目录，配置可持久化", () => {
|
|
const root = fs.mkdtempSync(path.join(os.tmpdir(), "codex-market-data-dir-"));
|
|
const userData = path.join(root, "user-data");
|
|
const first = path.join(root, "first-codex");
|
|
const second = path.join(root, "second-codex");
|
|
fs.mkdirSync(userData, { recursive: true });
|
|
writeRollout(first, "11111111-1111-4111-8111-111111111111", "第一份记录");
|
|
writeRollout(second, "22222222-2222-4222-8222-222222222222", "第二份记录");
|
|
|
|
try {
|
|
let { dataDir, sessions } = loadModules(userData);
|
|
assert.throws(() => dataDir.set(path.join(root, "not-codex")), /Codex 数据/);
|
|
assert.equal(dataDir.set(first).path, first);
|
|
assert.equal(sessions.getCodexHome(), first);
|
|
assert.deepEqual(sessions.listConversations().map((row) => row.title), ["第一份记录"]);
|
|
|
|
assert.equal(dataDir.set(second).path, second);
|
|
assert.equal(sessions.getCodexHome(), second);
|
|
assert.deepEqual(sessions.listConversations().map((row) => row.title), ["第二份记录"]);
|
|
|
|
({ dataDir, sessions } = loadModules(userData));
|
|
assert.equal(dataDir.info().path, second);
|
|
assert.deepEqual(sessions.listConversations().map((row) => row.title), ["第二份记录"]);
|
|
} finally {
|
|
fs.rmSync(root, { recursive: true, force: true });
|
|
}
|
|
});
|