Organize composer into its own module

2026-05-26 14:32:45 +07:00
parent 7e792e899a
commit 4231dbbba4
52 changed files with 935 additions and 706 deletions
@@ -0,0 +1,90 @@
+/**
+ * @vitest-environment jsdom
+ */
+import { act, renderHook } from "@testing-library/react";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { useCreateFlowStore } from "@/stores/create-flow-store";
+import type { UserMessageImageAttachment } from "@/types/stream";
+import type { AgentAttachment } from "@server/shared/messages";
+import { useDraftAgentCreateFlow, type DraftCreateAttempt } from "./create-flow";
+
+describe("useDraftAgentCreateFlow", () => {
+  beforeEach(() => {
+    useCreateFlowStore.setState({ pendingByDraftId: {} });
+  });
+
+  it("renders a prepared new-workspace create attempt as optimistic chat before continuing it", async () => {
+    const image: UserMessageImageAttachment = {
+      id: "image-1",
+      mimeType: "image/png",
+      storageType: "web-indexeddb",
+      storageKey: "image-key",
+      createdAt: 123,
+    };
+    const attachment = {
+      type: "review",
+      cwd: "/repo",
+      summary: "Review",
+    } as unknown as AgentAttachment;
+    const attempt: DraftCreateAttempt = {
+      clientMessageId: "msg-prepared",
+      text: "build this",
+      timestamp: new Date("2026-05-25T00:00:00.000Z"),
+      images: [image],
+      attachments: [attachment],
+    };
+    const createRequest = vi.fn(
+      async (ctx: {
+        attempt: DraftCreateAttempt;
+        text: string;
+        images?: UserMessageImageAttachment[];
+        attachments?: AgentAttachment[];
+        cwd: string;
+      }) => ({
+        agentId: "agent-1",
+        result: { id: "agent-1", ctx },
+      }),
+    );
+    const onCreateSuccess = vi.fn();
+
+    const { result } = renderHook(() =>
+      useDraftAgentCreateFlow({
+        draftId: "draft-1",
+        getPendingServerId: () => "server-1",
+        initialAttempt: attempt,
+        buildDraftAgent: (currentAttempt) => ({ currentAttempt }),
+        createRequest,
+        onCreateSuccess,
+      }),
+    );
+
+    expect(result.current.isSubmitting).toBe(true);
+    expect(result.current.draftAgent).toEqual({ currentAttempt: attempt });
+    expect(result.current.optimisticStreamItems).toEqual([
+      {
+        kind: "user_message",
+        id: "msg-prepared",
+        text: "build this",
+        timestamp: attempt.timestamp,
+        optimistic: true,
+        images: [image],
+        attachments: [attachment],
+      },
+    ]);
+    expect(createRequest).not.toHaveBeenCalled();
+
+    await act(async () => {
+      await result.current.continueCreateFromAttempt({ attempt, cwd: "/repo" });
+    });
+
+    expect(createRequest).toHaveBeenCalledTimes(1);
+    expect(createRequest).toHaveBeenCalledWith({
+      attempt,
+      text: "build this",
+      images: [image],
+      attachments: [attachment],
+      cwd: "/repo",
+    });
+    expect(onCreateSuccess).toHaveBeenCalledTimes(1);
+  });
+});