import { mkdirSync, mkdtempSync, readdirSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { describe, expect, it } from "vitest";
import {
type RlmSubagentDisplayEntry,
readRlmSubagentDisplayEntry,
rlmSubagentDisplayPath,
writeRlmSubagentDisplayEntry,
} from "../src/modes/daemon/rlm-subagent-display.js";
function makeEntry(sessionDir: string, overrides: Partial<RlmSubagentDisplayEntry> = {}): RlmSubagentDisplayEntry {
return {
type: "rlm_subagent",
childId: "sub-1234abcd",
sessionName: "worker",
sessionDir,
sessionFile: join(sessionDir, "01a0-child.jsonl"),
rlmMaxDepth: 4,
rlmParentNodeId: "sub-1234abcd",
prompt: "do the work",
spawnCode: "await rlm('do the work')",
model: { provider: "test", modelId: "model" },
status: "running",
createdAt: 1,
updatedAt: "2026-01-01T00:00:00.000Z",
...overrides,
};
}
describe("rlm subagent display files", () => {
it("round-trips an entry and replaces it atomically without temp-file residue", async () => {
const tempDir = mkdtempSync(join(tmpdir(), "prime-rlm-display-"));
try {
const sessionDir = join(tempDir, "sub-1234abcd");
const entry = makeEntry(sessionDir);
writeRlmSubagentDisplayEntry(entry);
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toEqual(entry);
const failed = makeEntry(sessionDir, { status: "error", updatedAt: "2026-01-01T00:00:01.000Z" });
writeRlmSubagentDisplayEntry(failed);
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toEqual(failed);
const deleted = makeEntry(sessionDir, { status: "deleted", updatedAt: "2026-01-01T00:00:02.000Z" });
writeRlmSubagentDisplayEntry(deleted);
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toEqual(deleted);
expect(readdirSync(sessionDir)).toEqual(["rlm-subagent.json"]);
} finally {
rmSync(tempDir, { recursive: true, force: true });
}
});
it("reads tolerantly: missing, malformed, and invalid files are undefined", async () => {
const tempDir = mkdtempSync(join(tmpdir(), "prime-rlm-display-tolerant-"));
try {
const sessionDir = join(tempDir, "sub-1234abcd");
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toBeUndefined();
mkdirSync(sessionDir, { recursive: true });
writeFileSync(rlmSubagentDisplayPath(sessionDir), "{not json");
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toBeUndefined();
writeFileSync(rlmSubagentDisplayPath(sessionDir), JSON.stringify({ type: "rlm_subagent", childId: 42 }));
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toBeUndefined();
writeFileSync(
rlmSubagentDisplayPath(sessionDir),
JSON.stringify(makeEntry(sessionDir, { status: "exploded" as RlmSubagentDisplayEntry["status"] })),
);
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toBeUndefined();
} finally {
rmSync(tempDir, { recursive: true, force: true });
}
});
it("accepts unknown extra fields from newer writers", async () => {
const tempDir = mkdtempSync(join(tmpdir(), "prime-rlm-display-forward-"));
try {
const sessionDir = join(tempDir, "sub-1234abcd");
const entry = makeEntry(sessionDir);
mkdirSync(sessionDir, { recursive: true });
writeFileSync(rlmSubagentDisplayPath(sessionDir), JSON.stringify({ ...entry, futureField: true }));
await expect(readRlmSubagentDisplayEntry(sessionDir)).resolves.toMatchObject({
childId: entry.childId,
status: "running",
});
} finally {
rmSync(tempDir, { recursive: true, force: true });
}
});
});