64 lines
4.3 KiB
TypeScript
64 lines
4.3 KiB
TypeScript
import {afterEach, beforeEach, expect, it, vi} from "vitest";
|
|
import {mkdtemp, rm, writeFile} from "node:fs/promises";
|
|
import {tmpdir} from "node:os";
|
|
import {join} from "node:path";
|
|
import {runNativeSession, runHandoffAction} from "../../agent-plugin-core/typescript/src/handoff.ts";
|
|
import {registerHandoffCommand} from "../tools/handoff.ts";
|
|
vi.mock("../../agent-plugin-core/typescript/src/handoff.ts", async original => ({...await original<typeof import("../../agent-plugin-core/typescript/src/handoff.ts")>(), runNativeSession: vi.fn(), runHandoffAction: vi.fn()}));
|
|
let dir: string;
|
|
beforeEach(async () => {vi.clearAllMocks(); dir = await mkdtemp(join(tmpdir(), "openclaw-handoff-"));});
|
|
afterEach(async () => {await rm(dir, {recursive: true, force: true});});
|
|
function setup() {
|
|
const registerCommand = vi.fn();
|
|
const registerTool = vi.fn();
|
|
const hooks = new Map<string, any>();
|
|
registerHandoffCommand({registerCommand, registerTool, on: (name: string, handler: any) => hooks.set(name, handler)} as any);
|
|
return {cmd: registerCommand.mock.calls[0][0], tool: registerTool.mock.calls[0][0]({workspaceDir: "/tmp/native-project"}), hooks};
|
|
}
|
|
it("saves the trusted current OpenClaw transcript from a user-only command", async () => {
|
|
const {cmd} = setup();
|
|
expect(cmd.name).toBe("mem0-handoff");
|
|
expect(cmd.requireAuth).toBe(true);
|
|
vi.mocked(runNativeSession).mockResolvedValue("Saved shared resource");
|
|
expect(await cmd.handler({sessionFile: "/tmp/native session.jsonl"})).toEqual({text: "Saved shared resource"});
|
|
expect(runNativeSession).toHaveBeenCalledWith(expect.any(URL), "openclaw", "/tmp/native session.jsonl");
|
|
});
|
|
it("loads resumed history only into the matching native session's next model prompt", async () => {
|
|
const {cmd, hooks} = setup();
|
|
const sessionFile = join(dir, "native.jsonl");
|
|
await writeFile(sessionFile, JSON.stringify({type: "session", cwd: "/tmp/native-project"}) + "\n");
|
|
vi.mocked(runHandoffAction).mockResolvedValue("Full historical context and tools");
|
|
expect((await cmd.handler({args: "resume /tmp/shared task.json", sessionFile, sessionId: "native"})).text).toContain("loaded");
|
|
expect(runHandoffAction).toHaveBeenCalledWith(expect.any(URL), "resume", "/tmp/native-project", "/tmp/shared task.json");
|
|
expect(hooks.get("before_prompt_build")({}, {sessionId: "other"})).toBeUndefined();
|
|
expect(hooks.get("before_prompt_build")({}, {sessionId: "native"})).toEqual({prependContext: "Full historical context and tools"});
|
|
expect(hooks.get("before_prompt_build")({}, {sessionId: "native"})).toBeUndefined();
|
|
});
|
|
it("clears queued context when the native session ends", async () => {
|
|
const {cmd, hooks} = setup();
|
|
const sessionFile = join(dir, "native.jsonl");
|
|
await writeFile(sessionFile, JSON.stringify({type: "session", cwd: "/tmp"}) + "\n");
|
|
vi.mocked(runHandoffAction).mockResolvedValue("Full historical context");
|
|
await cmd.handler({args: "resume /tmp/shared.json", sessionFile, sessionId: "native"});
|
|
hooks.get("session_end")({sessionId: "native"}, {});
|
|
expect(hooks.get("before_prompt_build")({}, {sessionId: "native"})).toBeUndefined();
|
|
});
|
|
it("provides a model-visible list/resume tool using only the native workspace", async () => {
|
|
const {tool} = setup();
|
|
vi.mocked(runHandoffAction).mockResolvedValue("Full historical context and tools");
|
|
for (const action of ["list", "resume"]) {
|
|
expect(await tool.execute("call", {action, resource: "/tmp/shared.json", cwd: "/untrusted"})).toEqual({content: [{type: "text", text: "Full historical context and tools"}]});
|
|
expect(runHandoffAction).toHaveBeenLastCalledWith(expect.any(URL), action, "/tmp/native-project", "/tmp/shared.json");
|
|
}
|
|
expect(runNativeSession).not.toHaveBeenCalled();
|
|
});
|
|
it("fails explicitly on unavailable context or resource errors", async () => {
|
|
const {cmd} = setup();
|
|
expect((await cmd.handler({})).text).toContain("unavailable");
|
|
expect((await cmd.handler({args: "unknown"})).text).toContain("Usage:");
|
|
expect(runNativeSession).not.toHaveBeenCalled();
|
|
vi.mocked(runNativeSession).mockRejectedValue(new Error("invalid native transcript"));
|
|
expect((await cmd.handler({sessionFile: "/tmp/native.jsonl"})).text).toContain("invalid native transcript");
|
|
expect((await cmd.handler({args: "resume /tmp/shared.json", sessionFile: "/tmp/native.jsonl"})).text).toContain("identity is unavailable");
|
|
});
|