mem0/integrations/openclaw/tests/handoff.test.ts

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.resetAllMocks(); dir = await mkdtemp(join(tmpdir(), "openclaw-handoff-"));});
afterEach(async () => {await rm(dir, {recursive: true, force: true});});
function setup(context: {workspaceDir?: string} = {workspaceDir: "/tmp/native-project"}) {
  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](context), 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");
});

it("lists resources from the transcript project without queuing prompt context", async () => {
  const {cmd, hooks} = setup();
  const sessionFile = join(dir, "native.jsonl");
  await writeFile(sessionFile, JSON.stringify({type: "session", cwd: dir}) + "\n");
  vi.mocked(runHandoffAction).mockResolvedValue("Available shared resource");
  expect(await cmd.handler({args: "list", sessionFile, sessionId: "native"})).toEqual({text: "Available shared resource"});
  expect(runHandoffAction).toHaveBeenCalledWith(expect.any(URL), "list", dir, undefined);
  expect(hooks.get("before_prompt_build")({}, {sessionId: "native"})).toBeUndefined();
});
it.each([
  ["empty transcript", ""],
  ["non-session header", JSON.stringify({type: "message", cwd: "/tmp"})],
  ["missing cwd", JSON.stringify({type: "session"})],
  ["relative cwd", JSON.stringify({type: "session", cwd: "relative/project"})],
])("rejects %s before accessing shared resources", async (_description, content) => {
  const {cmd, hooks} = setup();
  const sessionFile = join(dir, "invalid.jsonl");
  await writeFile(sessionFile, content);
  const result = await cmd.handler({args: "resume /tmp/shared.json", sessionFile, sessionId: "native"});
  expect(result.text).toContain("project directory is unavailable");
  expect(runHandoffAction).not.toHaveBeenCalled();
  expect(hooks.get("before_prompt_build")({}, {sessionId: "native"})).toBeUndefined();
});
it("reports a missing transcript rather than guessing a project", async () => {
  const {cmd} = setup();
  expect((await cmd.handler({args: "list", sessionFile: join(dir, "missing.jsonl")})).text).toContain("ENOENT");
  expect(runHandoffAction).not.toHaveBeenCalled();
});
it("rejects tools without a native workspace and unsupported actions", async () => {
  const missing = setup({}).tool;
  expect(await missing.execute("call", {action: "list"})).toMatchObject({isError: true, content: [{text: expect.stringContaining("project directory is unavailable")}]});
  const {tool} = setup();
  expect(await tool.execute("call", {action: "save"})).toMatchObject({isError: true, content: [{text: expect.stringContaining("Choose list or resume")}]});
  expect(runHandoffAction).not.toHaveBeenCalled();
});
it("lists through the tool without requiring a resource path", async () => {
  const {tool} = setup();
  vi.mocked(runHandoffAction).mockResolvedValue("No shared resources");
  expect(await tool.execute("call", {action: "list"})).toEqual({content: [{type: "text", text: "No shared resources"}]});
  expect(runHandoffAction).toHaveBeenCalledWith(expect.any(URL), "list", "/tmp/native-project", undefined);
});
it.each([new Error("Resource cannot be read"), "Resource cannot be read"])("surfaces resource failures as tool errors: %s", async (error) => {
  const {tool} = setup();
  vi.mocked(runHandoffAction).mockRejectedValue(error);
  expect(await tool.execute("call", {action: "resume", resource: "/tmp/shared.json"})).toEqual({isError: true, content: [{type: "text", text: "Session handoff failed: Resource cannot be read"}]});
});
it("reports a non-Error command failure without injecting context", async () => {
  const {cmd, hooks} = setup();
  const sessionFile = join(dir, "native.jsonl");
  await writeFile(sessionFile, JSON.stringify({type: "session", cwd: dir}) + "\n");
  vi.mocked(runHandoffAction).mockRejectedValue("Resource cannot be read");
  expect(await cmd.handler({args: "resume /tmp/shared.json", sessionFile, sessionId: "native"})).toEqual({text: "Session handoff failed: Resource cannot be read"});
  expect(hooks.get("before_prompt_build")({}, {sessionId: "native"})).toBeUndefined();
});