@lobu/worker 6.1.1 → 7.1.0

package/src/__tests__/processor-harden.test.ts

@@ -0,0 +1,259 @@
+/**
+ * Hardening tests for OpenClawProgressProcessor.
+ *
+ * Covers gaps in the existing processor.test.ts:
+ * - Malformed / missing fields on events (robustness, no throws)
+ * - getDelta when output was non-monotonically modified (rare guard)
+ * - reset() clears hasStreamedText so message_end re-extracts text
+ * - message_end with empty content blocks emits nothing
+ * - message_end with whitespace-only text emits nothing
+ * - Multiple consecutive getDelta() calls return only new content
+ * - tool_execution_start with non-object args does not throw
+ * - getOutputSnapshot reflects full output including already-sent content
+ */
+
+import { describe, expect, test } from "bun:test";
+import { OpenClawProgressProcessor } from "../openclaw/processor";
+
+function makeTextDelta(delta: string, role = "assistant"): any {
+  return {
+    type: "message_update",
+    message: { role },
+    assistantMessageEvent: { type: "text_delta", delta },
+  };
+}
+
+function makeMessageEnd(opts: {
+  role?: string;
+  content?: any[];
+  stopReason?: string;
+  errorMessage?: string;
+}): any {
+  return {
+    type: "message_end",
+    message: {
+      role: opts.role ?? "assistant",
+      content: opts.content ?? [],
+      stopReason: opts.stopReason,
+      errorMessage: opts.errorMessage,
+    },
+  };
+}
+
+// ---------------------------------------------------------------------------
+// Malformed event fields
+// ---------------------------------------------------------------------------
+
+describe("OpenClawProgressProcessor — malformed events don't throw", () => {
+  test("message_update with null assistantMessageEvent returns false without throwing", () => {
+    const p = new OpenClawProgressProcessor();
+    const event = {
+      type: "message_update",
+      message: { role: "assistant" },
+      assistantMessageEvent: null,
+    };
+    let result: boolean | undefined;
+    expect(() => {
+      result = p.processEvent(event as any);
+    }).not.toThrow();
+    expect(result).toBe(false);
+    expect(p.getDelta()).toBeNull();
+  });
+
+  test("tool_execution_start with null args does not throw", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(() =>
+      p.processEvent({
+        type: "tool_execution_start",
+        toolName: "Read",
+        args: null,
+      } as any)
+    ).not.toThrow();
+  });
+
+  test("tool_execution_start with string args does not throw", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(() =>
+      p.processEvent({
+        type: "tool_execution_start",
+        toolName: "Read",
+        args: "unexpected-string",
+      } as any)
+    ).not.toThrow();
+  });
+
+  test("auto_compaction_end with neither aborted nor result returns true", () => {
+    const p = new OpenClawProgressProcessor();
+    // No aborted, no result — the current implementation still returns true
+    const result = p.processEvent({ type: "auto_compaction_end" } as any);
+    expect(result).toBe(true);
+  });
+
+  test("auto_retry_end with success=true and no finalError returns false", () => {
+    const p = new OpenClawProgressProcessor();
+    const result = p.processEvent({
+      type: "auto_retry_end",
+      success: true,
+      finalError: undefined,
+    } as any);
+    expect(result).toBe(false);
+  });
+});
+
+// ---------------------------------------------------------------------------
+// message_end content extraction edge cases
+// ---------------------------------------------------------------------------
+
+describe("message_end content extraction", () => {
+  test("empty content array produces no output", () => {
+    const p = new OpenClawProgressProcessor();
+    const result = p.processEvent(makeMessageEnd({ content: [] }));
+    expect(result).toBe(false);
+    expect(p.getDelta()).toBeNull();
+  });
+
+  test("whitespace-only text block produces no output", () => {
+    const p = new OpenClawProgressProcessor();
+    const result = p.processEvent(
+      makeMessageEnd({ content: [{ type: "text", text: "   \n\t  " }] })
+    );
+    expect(result).toBe(false);
+    expect(p.getDelta()).toBeNull();
+  });
+
+  test("non-text blocks are skipped", () => {
+    const p = new OpenClawProgressProcessor();
+    const result = p.processEvent(
+      makeMessageEnd({ content: [{ type: "tool_use", id: "x" }] })
+    );
+    expect(result).toBe(false);
+  });
+
+  test("multiple text blocks concatenated", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(
+      makeMessageEnd({
+        content: [
+          { type: "text", text: "Hello" },
+          { type: "text", text: " World" },
+        ],
+      })
+    );
+    expect(p.getDelta()).toContain("Hello World");
+  });
+
+  test("message_end with error does not append content text", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(
+      makeMessageEnd({
+        stopReason: "error",
+        errorMessage: "Boom",
+        content: [{ type: "text", text: "Should not appear" }],
+      })
+    );
+    expect(p.getDelta()).toBeNull();
+    expect(p.consumeFatalErrorMessage()).toBe("Boom");
+  });
+
+  test("message_end after streaming skips re-extraction", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeTextDelta("streamed text"));
+    p.getDelta();
+
+    // Now simulate message_end with content
+    const result = p.processEvent(
+      makeMessageEnd({ content: [{ type: "text", text: "duplicate" }] })
+    );
+    expect(result).toBe(false);
+    expect(p.getDelta()).toBeNull();
+  });
+
+  test("after reset(), message_end re-extracts text even if streaming happened before", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeTextDelta("before reset"));
+    p.getDelta();
+
+    p.reset();
+
+    // Now message_end should be able to extract (hasStreamedText reset)
+    const result = p.processEvent(
+      makeMessageEnd({ content: [{ type: "text", text: "after reset" }] })
+    );
+    expect(result).toBe(true);
+    expect(p.getDelta()).toContain("after reset");
+  });
+});
+
+// ---------------------------------------------------------------------------
+// getDelta monotonicity and snapshot
+// ---------------------------------------------------------------------------
+
+describe("getDelta and getOutputSnapshot", () => {
+  test("multiple incremental deltas return cumulative suffix each time", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeTextDelta("A"));
+    expect(p.getDelta()).toBe("A");
+
+    p.processEvent(makeTextDelta("B"));
+    expect(p.getDelta()).toBe("B");
+
+    p.processEvent(makeTextDelta("C"));
+    expect(p.getDelta()).toBe("C");
+  });
+
+  test("getOutputSnapshot returns full accumulated output", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeTextDelta("part1"));
+    p.getDelta(); // consume
+
+    p.processEvent(makeTextDelta(" part2"));
+    p.getDelta(); // consume
+
+    // Snapshot reflects everything even after getDelta consumed it
+    expect(p.getOutputSnapshot()).toBe("part1 part2");
+  });
+
+  test("getDelta returns full content when it diverges from lastSentContent", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeTextDelta("original"));
+    p.getDelta(); // lastSentContent = "original"
+
+    // Simulate a reset + new content that doesn't start with old prefix
+    p.reset();
+    p.processEvent(makeTextDelta("completely different"));
+    // Full content is returned since it doesn't start with lastSentContent (empty after reset)
+    expect(p.getDelta()).toBe("completely different");
+  });
+});
+
+// ---------------------------------------------------------------------------
+// Thinking accumulation across verbose/non-verbose
+// ---------------------------------------------------------------------------
+
+describe("thinking accumulation", () => {
+  test("thinking deltas accumulate across multiple events", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent({
+      type: "message_update",
+      message: { role: "assistant" },
+      assistantMessageEvent: { type: "thinking_delta", delta: "step one " },
+    } as any);
+    p.processEvent({
+      type: "message_update",
+      message: { role: "assistant" },
+      assistantMessageEvent: { type: "thinking_delta", delta: "step two" },
+    } as any);
+    expect(p.getCurrentThinking()).toBe("step one step two");
+  });
+
+  test("reset() clears accumulated thinking", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent({
+      type: "message_update",
+      message: { role: "assistant" },
+      assistantMessageEvent: { type: "thinking_delta", delta: "thoughts" },
+    } as any);
+    p.reset();
+    expect(p.getCurrentThinking()).toBeNull();
+  });
+});

package/src/__tests__/processor.test.ts

@@ -0,0 +1,225 @@
+import { describe, expect, test } from "bun:test";
+import { OpenClawProgressProcessor } from "../openclaw/processor";
+
+function makeEvent(type: string, extra: Record<string, any> = {}): any {
+  return { type, ...extra };
+}
+
+function makeMessageUpdate(
+  assistantEventType: string,
+  delta = "",
+  role = "assistant"
+): any {
+  return makeEvent("message_update", {
+    message: { role },
+    assistantMessageEvent: { type: assistantEventType, delta },
+  });
+}
+
+describe("OpenClawProgressProcessor", () => {
+  test("processEvent returns false for non-assistant message_update", () => {
+    const p = new OpenClawProgressProcessor();
+    const event = makeMessageUpdate("text_delta", "hi", "user");
+    expect(p.processEvent(event)).toBe(false);
+  });
+
+  test("text_delta appends to output", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeMessageUpdate("text_delta", "Hello"));
+    p.processEvent(makeMessageUpdate("text_delta", " world"));
+    expect(p.getDelta()).toBe("Hello world");
+  });
+
+  test("thinking_delta does not append by default", () => {
+    const p = new OpenClawProgressProcessor();
+    const result = p.processEvent(
+      makeMessageUpdate("thinking_delta", "thinking...")
+    );
+    expect(result).toBe(false);
+    expect(p.getDelta()).toBeNull();
+  });
+
+  test("thinking_delta appends when verbose", () => {
+    const p = new OpenClawProgressProcessor();
+    p.setVerboseLogging(true);
+    const result = p.processEvent(
+      makeMessageUpdate("thinking_delta", "thinking...")
+    );
+    expect(result).toBe(true);
+    expect(p.getDelta()).toContain("thinking...");
+  });
+
+  test("thinking_start/end output in verbose mode only", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(p.processEvent(makeMessageUpdate("thinking_start"))).toBe(false);
+    expect(p.processEvent(makeMessageUpdate("thinking_end"))).toBe(false);
+
+    p.setVerboseLogging(true);
+    expect(p.processEvent(makeMessageUpdate("thinking_start"))).toBe(true);
+    expect(p.processEvent(makeMessageUpdate("thinking_end"))).toBe(true);
+  });
+
+  test("getCurrentThinking tracks thinking content", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(p.getCurrentThinking()).toBeNull();
+    p.processEvent(makeMessageUpdate("thinking_delta", "step 1"));
+    expect(p.getCurrentThinking()).toBe("step 1");
+  });
+
+  test("message_end with error stores fatal error", () => {
+    const p = new OpenClawProgressProcessor();
+    const event = makeEvent("message_end", {
+      message: {
+        role: "assistant",
+        stopReason: "error",
+        errorMessage: "Something broke",
+      },
+    });
+    expect(p.processEvent(event)).toBe(false);
+    expect(p.consumeFatalErrorMessage()).toBe("Something broke");
+    // Consumed, so second call returns null
+    expect(p.consumeFatalErrorMessage()).toBeNull();
+  });
+
+  test("message_end extracts text when no streaming happened", () => {
+    const p = new OpenClawProgressProcessor();
+    const event = makeEvent("message_end", {
+      message: {
+        role: "assistant",
+        content: [{ type: "text", text: "Final answer" }],
+      },
+    });
+    expect(p.processEvent(event)).toBe(true);
+    expect(p.getDelta()).toContain("Final answer");
+  });
+
+  test("message_end skips extraction if text already streamed", () => {
+    const p = new OpenClawProgressProcessor();
+    // Stream some text first
+    p.processEvent(makeMessageUpdate("text_delta", "streamed"));
+    p.getDelta(); // consume
+    // Now message_end should skip
+    const event = makeEvent("message_end", {
+      message: {
+        role: "assistant",
+        content: [{ type: "text", text: "Final" }],
+      },
+    });
+    expect(p.processEvent(event)).toBe(false);
+  });
+
+  test("auto_compaction_start appends message", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(p.processEvent(makeEvent("auto_compaction_start"))).toBe(true);
+    expect(p.getDelta()).toContain("Compacting context");
+  });
+
+  test("auto_compaction_end with aborted", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(
+      p.processEvent(makeEvent("auto_compaction_end", { aborted: true }))
+    ).toBe(true);
+    expect(p.getDelta()).toContain("Compaction aborted");
+  });
+
+  test("auto_compaction_end with result", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(
+      p.processEvent(makeEvent("auto_compaction_end", { result: {} }))
+    ).toBe(true);
+    expect(p.getDelta()).toContain("Context compacted");
+  });
+
+  test("auto_retry_start appends retry message", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(
+      p.processEvent(
+        makeEvent("auto_retry_start", { attempt: 2, maxAttempts: 3 })
+      )
+    ).toBe(true);
+    expect(p.getDelta()).toContain("Retrying (attempt 2/3)");
+  });
+
+  test("auto_retry_end with failure appends error", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(
+      p.processEvent(
+        makeEvent("auto_retry_end", {
+          success: false,
+          finalError: "timeout",
+        })
+      )
+    ).toBe(true);
+    expect(p.getDelta()).toContain("Retry failed: timeout");
+  });
+
+  test("auto_retry_end with success returns false", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(p.processEvent(makeEvent("auto_retry_end", { success: true }))).toBe(
+      false
+    );
+  });
+
+  test("unknown event type returns false", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(p.processEvent(makeEvent("unknown_event"))).toBe(false);
+  });
+});
+
+describe("getDelta", () => {
+  test("returns null when no content", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(p.getDelta()).toBeNull();
+  });
+
+  test("returns null when content unchanged", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeMessageUpdate("text_delta", "hello"));
+    p.getDelta(); // consume
+    expect(p.getDelta()).toBeNull();
+  });
+
+  test("returns only the new suffix on incremental append", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeMessageUpdate("text_delta", "Hello"));
+    p.getDelta(); // consume "Hello"
+    p.processEvent(makeMessageUpdate("text_delta", " world"));
+    expect(p.getDelta()).toBe(" world");
+  });
+
+  test("returns full content on first call", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeMessageUpdate("text_delta", "first"));
+    expect(p.getDelta()).toBe("first");
+  });
+});
+
+describe("finalResult lifecycle", () => {
+  test("set and get final result", () => {
+    const p = new OpenClawProgressProcessor();
+    expect(p.getFinalResult()).toBeNull();
+
+    p.setFinalResult({ text: "done", isFinal: true });
+    const result = p.getFinalResult();
+    expect(result).toEqual({ text: "done", isFinal: true });
+
+    // Consumed, so next call returns null
+    expect(p.getFinalResult()).toBeNull();
+  });
+});
+
+describe("reset", () => {
+  test("clears all state", () => {
+    const p = new OpenClawProgressProcessor();
+    p.processEvent(makeMessageUpdate("text_delta", "content"));
+    p.processEvent(makeMessageUpdate("thinking_delta", "thought"));
+    p.setFinalResult({ text: "done", isFinal: true });
+
+    p.reset();
+
+    expect(p.getDelta()).toBeNull();
+    expect(p.getCurrentThinking()).toBeNull();
+    expect(p.getFinalResult()).toBeNull();
+    expect(p.consumeFatalErrorMessage()).toBeNull();
+  });
+});

package/src/__tests__/replace-base-prompt-identity.test.ts

@@ -0,0 +1,41 @@
+import { describe, expect, test } from "bun:test";
+import { replaceBasePromptIdentity } from "../openclaw/worker";
+
+const PI_OPENER =
+  "You are an expert coding assistant operating inside pi, a coding agent harness. You help users by reading files, executing commands, editing code, and writing new files.";
+
+describe("replaceBasePromptIdentity", () => {
+  test("replaces the pi-coding-agent opener with agent identity, preserving the rest", () => {
+    const base = `${PI_OPENER}\n\nAvailable tools:\n- read: Read file contents\n\nGuidelines:\n- Be concise`;
+    const identity = "You are a healthcare operations assistant.";
+    const out = replaceBasePromptIdentity(base, identity);
+
+    expect(out.startsWith(identity)).toBe(true);
+    expect(out).not.toContain("expert coding assistant");
+    // Preserved harness footer
+    expect(out).toContain("Available tools:");
+    expect(out).toContain("Guidelines:");
+  });
+
+  test("falls back to prepending identity when upstream opener wording drifts", () => {
+    const base =
+      "You are some other intro that the upstream package switched to.\n\nAvailable tools:\n- read";
+    const identity = "You are a healthcare operations assistant.";
+    const out = replaceBasePromptIdentity(base, identity);
+
+    expect(out.startsWith(identity)).toBe(true);
+    // Original base prompt is still there (we didn't accidentally drop it)
+    expect(out).toContain("Available tools:");
+    expect(out).toContain("some other intro");
+  });
+
+  test("multi-line identity is inserted as a single block", () => {
+    const base = `${PI_OPENER}\n\nAvailable tools:\n- read`;
+    const identity =
+      "You are a careops bot.\n\nYou speak only in plain English.";
+    const out = replaceBasePromptIdentity(base, identity);
+
+    expect(out.startsWith(identity)).toBe(true);
+    expect(out).toContain("Available tools:");
+  });
+});