claude-code/src/utils/__tests__/collapseHookSummaries.test.ts

import { describe, expect, test } from "bun:test";
import { collapseHookSummaries } from "../collapseHookSummaries";

function makeHookSummary(overrides: Partial<{
  hookLabel: string;
  hookCount: number;
  hookInfos: any[];
  hookErrors: any[];
  preventedContinuation: boolean;
  hasOutput: boolean;
  totalDurationMs: number;
}> = {}): any {
  return {
    type: "system",
    subtype: "stop_hook_summary",
    hookLabel: overrides.hookLabel ?? "PostToolUse",
    hookCount: overrides.hookCount ?? 1,
    hookInfos: overrides.hookInfos ?? [],
    hookErrors: overrides.hookErrors ?? [],
    preventedContinuation: overrides.preventedContinuation ?? false,
    hasOutput: overrides.hasOutput ?? false,
    totalDurationMs: overrides.totalDurationMs ?? 10,
  };
}

function makeNonHookMessage(): any {
  return { type: "user", message: { content: "hello" } };
}

describe("collapseHookSummaries", () => {
  test("returns same messages when no hook summaries", () => {
    const messages = [makeNonHookMessage(), makeNonHookMessage()];
    expect(collapseHookSummaries(messages)).toEqual(messages);
  });

  test("collapses consecutive messages with same hookLabel", () => {
    const messages = [
      makeHookSummary({ hookLabel: "PostToolUse", hookCount: 1 }),
      makeHookSummary({ hookLabel: "PostToolUse", hookCount: 2 }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result).toHaveLength(1);
    expect(result[0].hookCount).toBe(3);
  });

  test("does not collapse messages with different hookLabels", () => {
    const messages = [
      makeHookSummary({ hookLabel: "PostToolUse" }),
      makeHookSummary({ hookLabel: "PreToolUse" }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result).toHaveLength(2);
  });

  test("aggregates hookCount across collapsed messages", () => {
    const messages = [
      makeHookSummary({ hookLabel: "A", hookCount: 3 }),
      makeHookSummary({ hookLabel: "A", hookCount: 5 }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result[0].hookCount).toBe(8);
  });

  test("merges hookInfos arrays", () => {
    const info1 = { tool: "Read" };
    const info2 = { tool: "Write" };
    const messages = [
      makeHookSummary({ hookLabel: "A", hookInfos: [info1] }),
      makeHookSummary({ hookLabel: "A", hookInfos: [info2] }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result[0].hookInfos).toEqual([info1, info2]);
  });

  test("merges hookErrors arrays", () => {
    const err1 = new Error("e1");
    const err2 = new Error("e2");
    const messages = [
      makeHookSummary({ hookLabel: "A", hookErrors: [err1] }),
      makeHookSummary({ hookLabel: "A", hookErrors: [err2] }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result[0].hookErrors).toHaveLength(2);
  });

  test("takes max totalDurationMs", () => {
    const messages = [
      makeHookSummary({ hookLabel: "A", totalDurationMs: 50 }),
      makeHookSummary({ hookLabel: "A", totalDurationMs: 100 }),
      makeHookSummary({ hookLabel: "A", totalDurationMs: 75 }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result[0].totalDurationMs).toBe(100);
  });

  test("takes any truthy preventContinuation", () => {
    const messages = [
      makeHookSummary({ hookLabel: "A", preventedContinuation: false }),
      makeHookSummary({ hookLabel: "A", preventedContinuation: true }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result[0].preventedContinuation).toBe(true);
  });

  test("leaves single hook summary unchanged", () => {
    const msg = makeHookSummary({ hookLabel: "PostToolUse", hookCount: 5 });
    const result = collapseHookSummaries([msg]);
    expect(result).toHaveLength(1);
    expect(result[0].hookCount).toBe(5);
  });

  test("handles three consecutive same-label summaries", () => {
    const messages = [
      makeHookSummary({ hookLabel: "X", hookCount: 1 }),
      makeHookSummary({ hookLabel: "X", hookCount: 1 }),
      makeHookSummary({ hookLabel: "X", hookCount: 1 }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result).toHaveLength(1);
    expect(result[0].hookCount).toBe(3);
  });

  test("preserves non-hook messages in between", () => {
    const messages = [
      makeHookSummary({ hookLabel: "A" }),
      makeNonHookMessage(),
      makeHookSummary({ hookLabel: "A" }),
    ];
    const result = collapseHookSummaries(messages);
    expect(result).toHaveLength(3);
  });

  test("returns empty array for empty input", () => {
    expect(collapseHookSummaries([])).toEqual([]);
  });
});