feat(web): add harness event runtime bridge

2026-09-07 16:58:23 +08:00
parent acc5e26b1c
commit 63ea92b770
7 changed files with 812 additions and 0 deletions
@@ -0,0 +1,70 @@
+import { act, renderHook } from "@testing-library/react";
+import type { AppendMessage } from "@assistant-ui/react";
+import { describe, expect, it, vi } from "vitest";
+import { extractMessageInput, useHarnessRuntime } from "../src/assistant/use-harness-runtime.js";
+import { createInitialHarnessState, type HarnessClientState } from "../src/store/reducer.js";
+
+function appendMessage(text: string): AppendMessage {
+  return {
+    role: "user",
+    content: [{ type: "text", text }],
+    attachments: [],
+    createdAt: new Date("2026-09-07T08:00:00.000Z"),
+    metadata: { custom: {} },
+    parentId: null,
+    sourceId: null,
+    runConfig: undefined,
+  };
+}
+
+describe("useHarnessRuntime", () => {
+  it("turns a composer message into one harness API call", async () => {
+    const sendMessage = vi.fn().mockResolvedValue({ runId: "run_1" });
+    const cancel = vi.fn().mockResolvedValue({ cancelled: true });
+    let snapshot: HarnessClientState = createInitialHarnessState();
+    const subscribers = new Set<() => void>();
+    const store = {
+      getSnapshot: () => snapshot,
+      subscribe: (subscriber: () => void) => {
+        subscribers.add(subscriber);
+        return () => subscribers.delete(subscriber);
+      },
+      sendMessage,
+      cancel,
+    };
+
+    const { result } = renderHook(() => useHarnessRuntime(store));
+    await act(async () => {
+      result.current.thread.append(appendMessage("  hello agent  "));
+      await Promise.resolve();
+    });
+
+    expect(sendMessage).toHaveBeenCalledTimes(1);
+    expect(sendMessage).toHaveBeenCalledWith({ text: "hello agent", attachmentIds: [] });
+
+    snapshot = { ...snapshot, runs: { run_1: { runId: "run_1", modelId: "test", status: "running" } } };
+    act(() => subscribers.forEach((subscriber) => subscriber()));
+    await act(async () => {
+      result.current.thread.cancelRun();
+      await Promise.resolve();
+    });
+    expect(cancel).toHaveBeenCalledTimes(1);
+  });
+
+  it("extracts uploaded attachment ids without retaining file data", () => {
+    const message = appendMessage("caption");
+    const input = extractMessageInput({
+      ...message,
+      attachments: [{
+        id: "att_one",
+        type: "file",
+        name: "one.png",
+        contentType: "image/png",
+        status: { type: "complete" },
+        content: [],
+      }],
+    });
+
+    expect(input).toEqual({ text: "caption", attachmentIds: ["att_one"] });
+  });
+});