claude-code/packages/remote-control-server/src/__tests__/client-payload.test.ts

import { describe, test, expect } from "bun:test";
import { toClientPayload } from "../transport/client-payload";
import type { SessionEvent } from "../transport/event-bus";

function makeEvent(overrides: Partial<SessionEvent> & Pick<SessionEvent, "type" | "sessionId">): SessionEvent {
  return {
    id: "evt-1",
    payload: null,
    direction: "inbound",
    seqNum: 1,
    createdAt: Date.now(),
    ...overrides,
  };
}

// =============================================================================
// user / user_message
// =============================================================================

describe("toClientPayload — user message", () => {
  test("maps user type with content", () => {
    const event = makeEvent({
      type: "user",
      sessionId: "sess-1",
      payload: { content: "hello" },
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("user");
    expect(result.session_id).toBe("sess-1");
    expect((result as any).message.role).toBe("user");
    expect((result as any).message.content).toBe("hello");
  });

  test("maps user_message type same as user", () => {
    const event = makeEvent({
      type: "user_message",
      sessionId: "sess-2",
      payload: { content: "world" },
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("user");
    expect(result.session_id).toBe("sess-2");
  });

  test("falls back to message field when content is missing", () => {
    const event = makeEvent({
      type: "user",
      sessionId: "sess-3",
      payload: { message: "fallback msg" },
    });
    const result = toClientPayload(event);
    expect((result as any).message.content).toBe("fallback msg");
  });

  test("falls back to empty string when both content and message missing", () => {
    const event = makeEvent({
      type: "user",
      sessionId: "sess-4",
      payload: {},
    });
    const result = toClientPayload(event);
    expect((result as any).message.content).toBe("");
  });

  test("includes isSynthetic when true", () => {
    const event = makeEvent({
      type: "user",
      sessionId: "sess-5",
      payload: { content: "auto", isSynthetic: true },
    });
    const result = toClientPayload(event);
    expect((result as any).isSynthetic).toBe(true);
  });

  test("does not include isSynthetic when false", () => {
    const event = makeEvent({
      type: "user",
      sessionId: "sess-6",
      payload: { content: "manual", isSynthetic: false },
    });
    const result = toClientPayload(event);
    expect((result as any).isSynthetic).toBeUndefined();
  });

  test("uses payload.uuid when present", () => {
    const event = makeEvent({
      type: "user",
      sessionId: "sess-7",
      payload: { content: "hi", uuid: "custom-uuid" },
    });
    const result = toClientPayload(event);
    expect(result.uuid).toBe("custom-uuid");
  });

  test("falls back to event.id when payload.uuid is missing", () => {
    const event = makeEvent({
      type: "user",
      sessionId: "sess-8",
      payload: { content: "hi" },
    });
    const result = toClientPayload(event);
    expect(result.uuid).toBe("evt-1");
  });
});

// =============================================================================
// permission_response / control_response
// =============================================================================

describe("toClientPayload — permission response", () => {
  test("approved=true maps to allow behavior", () => {
    const event = makeEvent({
      type: "permission_response",
      sessionId: "sess-1",
      payload: { approved: true, request_id: "req-1" },
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("control_response");
    const resp = (result as any).response;
    expect(resp.subtype).toBe("success");
    expect(resp.request_id).toBe("req-1");
    expect(resp.response.behavior).toBe("allow");
  });

  test("approved=false maps to deny behavior with error", () => {
    const event = makeEvent({
      type: "permission_response",
      sessionId: "sess-2",
      payload: { approved: false, request_id: "req-2" },
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("control_response");
    const resp = (result as any).response;
    expect(resp.subtype).toBe("error");
    expect(resp.error).toBe("Permission denied by user");
    expect(resp.response.behavior).toBe("deny");
  });

  test("approved=false includes feedback message when provided", () => {
    const event = makeEvent({
      type: "permission_response",
      sessionId: "sess-3",
      payload: { approved: false, request_id: "req-3", message: "please revise" },
    });
    const result = toClientPayload(event);
    expect((result as any).response.message).toBe("please revise");
  });

  test("passes through existingResponse directly", () => {
    const existingResponse = { subtype: "success", custom: true };
    const event = makeEvent({
      type: "control_response",
      sessionId: "sess-4",
      payload: { approved: true, response: existingResponse },
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("control_response");
    expect((result as any).response).toBe(existingResponse);
  });

  test("includes updatedInput when approved with updated_input", () => {
    const updatedInput = { file_path: "/new/path" };
    const event = makeEvent({
      type: "permission_response",
      sessionId: "sess-5",
      payload: { approved: true, request_id: "req-5", updated_input: updatedInput },
    });
    const result = toClientPayload(event);
    expect((result as any).response.response.updatedInput).toEqual(updatedInput);
  });

  test("includes updatedPermissions when approved with updated_permissions", () => {
    const perms = [{ type: "allow", tool: "bash" }];
    const event = makeEvent({
      type: "permission_response",
      sessionId: "sess-6",
      payload: { approved: true, request_id: "req-6", updated_permissions: perms },
    });
    const result = toClientPayload(event);
    expect((result as any).response.response.updatedPermissions).toEqual(perms);
  });
});

// =============================================================================
// interrupt
// =============================================================================

describe("toClientPayload — interrupt", () => {
  test("maps interrupt to control_request with subtype interrupt", () => {
    const event = makeEvent({
      type: "interrupt",
      sessionId: "sess-1",
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("control_request");
    expect((result as any).request_id).toBe("evt-1");
    expect((result as any).request.subtype).toBe("interrupt");
  });
});

// =============================================================================
// control_request
// =============================================================================

describe("toClientPayload — control_request", () => {
  test("passes through request_id and request from payload", () => {
    const event = makeEvent({
      type: "control_request",
      sessionId: "sess-1",
      payload: { request_id: "req-99", request: { subtype: "permission", tool: "bash" } },
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("control_request");
    expect((result as any).request_id).toBe("req-99");
    expect((result as any).request.subtype).toBe("permission");
  });

  test("falls back request to payload when no request field", () => {
    const event = makeEvent({
      type: "control_request",
      sessionId: "sess-2",
      payload: { request_id: "req-10", custom: "data" },
    });
    const result = toClientPayload(event);
    expect((result as any).request).toEqual({ request_id: "req-10", custom: "data" });
  });

  test("falls back request_id to event.id when missing", () => {
    const event = makeEvent({
      type: "control_request",
      sessionId: "sess-3",
      payload: { request: { subtype: "test" } },
    });
    const result = toClientPayload(event);
    expect((result as any).request_id).toBe("evt-1");
  });
});

// =============================================================================
// default fallback
// =============================================================================

describe("toClientPayload — default types", () => {
  test("passes through unknown type with type/uuid/session_id/message", () => {
    const event = makeEvent({
      type: "assistant",
      sessionId: "sess-1",
      payload: { uuid: "u-1", content: "response text" },
    });
    const result = toClientPayload(event);
    expect(result.type).toBe("assistant");
    expect(result.uuid).toBe("u-1");
    expect(result.session_id).toBe("sess-1");
    expect(result.message).toEqual({ uuid: "u-1", content: "response text" });
  });
});