76 lines
2.7 KiB
TypeScript
76 lines
2.7 KiB
TypeScript
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import { CURRENT_SESSION_VERSION } from "@mariozechner/pi-coding-agent";
|
|
import { describe, expect, it, vi } from "vitest";
|
|
import type { GatewayRequestContext } from "./types.js";
|
|
|
|
// Guardrail: Ensure gateway "injected" assistant transcript messages are appended via SessionManager,
|
|
// so they are attached to the current leaf with a `parentId` and do not sever compaction history.
|
|
describe("gateway chat.inject transcript writes", () => {
|
|
it("appends a Pi session entry that includes parentId", async () => {
|
|
const dir = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-chat-inject-"));
|
|
const transcriptPath = path.join(dir, "sess.jsonl");
|
|
|
|
// Minimal Pi session header so SessionManager can open/append safely.
|
|
fs.writeFileSync(
|
|
transcriptPath,
|
|
`${JSON.stringify({
|
|
type: "session",
|
|
version: CURRENT_SESSION_VERSION,
|
|
id: "sess-1",
|
|
timestamp: new Date(0).toISOString(),
|
|
cwd: "/tmp",
|
|
})}\n`,
|
|
"utf-8",
|
|
);
|
|
|
|
vi.doMock("../session-utils.js", async (importOriginal) => {
|
|
const original = await importOriginal<typeof import("../session-utils.js")>();
|
|
return {
|
|
...original,
|
|
loadSessionEntry: () => ({
|
|
storePath: path.join(dir, "sessions.json"),
|
|
entry: {
|
|
sessionId: "sess-1",
|
|
sessionFile: transcriptPath,
|
|
},
|
|
}),
|
|
};
|
|
});
|
|
|
|
const { chatHandlers } = await import("./chat.js");
|
|
|
|
const respond = vi.fn();
|
|
type InjectCtx = Pick<GatewayRequestContext, "broadcast" | "nodeSendToSession">;
|
|
const context: InjectCtx = {
|
|
broadcast: vi.fn() as unknown as InjectCtx["broadcast"],
|
|
nodeSendToSession: vi.fn() as unknown as InjectCtx["nodeSendToSession"],
|
|
};
|
|
await chatHandlers["chat.inject"]({
|
|
params: { sessionKey: "k1", message: "hello" },
|
|
respond,
|
|
req: {} as never,
|
|
client: null as never,
|
|
isWebchatConnect: () => false,
|
|
context: context as unknown as GatewayRequestContext,
|
|
});
|
|
|
|
expect(respond).toHaveBeenCalled();
|
|
const [, payload, error] = respond.mock.calls.at(-1) ?? [];
|
|
expect(error).toBeUndefined();
|
|
expect(payload).toMatchObject({ ok: true });
|
|
|
|
const lines = fs.readFileSync(transcriptPath, "utf-8").split(/\r?\n/).filter(Boolean);
|
|
expect(lines.length).toBeGreaterThanOrEqual(2);
|
|
|
|
const last = JSON.parse(lines.at(-1) as string) as Record<string, unknown>;
|
|
expect(last.type).toBe("message");
|
|
|
|
// The regression we saw: raw jsonl appends omitted this field entirely.
|
|
expect(Object.prototype.hasOwnProperty.call(last, "parentId")).toBe(true);
|
|
expect(last).toHaveProperty("id");
|
|
expect(last).toHaveProperty("message");
|
|
});
|
|
});
|