1

src/components/puzzle-agent/PuzzleAgentWorkspace.interaction.test.tsx

@@ -1,7 +1,6 @@
-/* @vitest-environment jsdom */
+/* @vitest-environment jsdom */
 
-import { render, screen } from '@testing-library/react';
-import userEvent from '@testing-library/user-event';
+import { fireEvent, render, screen } from '@testing-library/react';
 import { beforeEach, expect, test, vi } from 'vitest';
 
 import type { PuzzleAgentSessionSnapshot } from '../../../packages/shared/src/contracts/puzzleAgentSession';
@@ -50,11 +49,11 @@ const baseSession: PuzzleAgentSessionSnapshot = {
       id: 'message-1',
       role: 'assistant',
       kind: 'chat',
-      text: '画面主体已经清楚，继续收束剩余关键词。',
+      text: '旧会话消息不再渲染为聊天入口。',
       createdAt: '2026-04-24T10:00:00.000Z',
     },
   ],
-  lastAssistantReply: '画面主体已经清楚，继续收束剩余关键词。',
+  lastAssistantReply: '旧会话消息不再渲染为聊天入口。',
   publishedProfileId: null,
   suggestedActions: [],
   resultPreview: null,
@@ -67,64 +66,54 @@ beforeEach(() => {
   }
 });
 
-test('puzzle workspace submits quick keyword fill request after two turns', async () => {
-  const user = userEvent.setup();
-  const onSubmitMessage = vi.fn();
+test('puzzle workspace submits the two-field form instead of agent chat', () => {
+  const onCreateFromForm = vi.fn();
+
+  render(
+    <PuzzleAgentWorkspace
+      session={null}
+      onBack={() => {}}
+      onSubmitMessage={() => {}}
+      onExecuteAction={() => {}}
+      onCreateFromForm={onCreateFromForm}
+    />,
+  );
+
+  fireEvent.change(screen.getByLabelText('拼图标题'), {
+    target: { value: '暖灯猫街' },
+  });
+  fireEvent.change(screen.getByLabelText('画面描述'), {
+    target: { value: '一只猫在雨夜灯牌下回头。' },
+  });
+  fireEvent.click(screen.getByRole('button', { name: /生成草稿/u }));
+
+  expect(onCreateFromForm).toHaveBeenCalledWith({
+    seedText: '暖灯猫街',
+    pictureDescription: '一只猫在雨夜灯牌下回头。',
+    referenceImageSrc: null,
+  });
+  expect(screen.queryByRole('button', { name: '补充剩余设定' })).toBeNull();
+  expect(screen.queryByText('旧会话消息不再渲染为聊天入口。')).toBeNull();
+});
+
+test('puzzle workspace falls back to compile action for restored sessions', () => {
+  const onExecuteAction = vi.fn();
 
   render(
     <PuzzleAgentWorkspace
       session={baseSession}
       onBack={() => {}}
-      onSubmitMessage={onSubmitMessage}
-      onExecuteAction={() => {}}
-    />,
-  );
-
-  await user.click(screen.getByRole('button', { name: '补充剩余设定' }));
-
-  expect(onSubmitMessage).toHaveBeenCalledWith(
-    expect.objectContaining({
-      text: '请补充剩余设定。',
-      quickFillRequested: true,
-    }),
-  );
-});
-
-test('puzzle workspace hides keyword fill before two turns', () => {
-  render(
-    <PuzzleAgentWorkspace
-      session={{ ...baseSession, currentTurn: 1 }}
-      onBack={() => {}}
       onSubmitMessage={() => {}}
-      onExecuteAction={() => {}}
+      onExecuteAction={onExecuteAction}
     />,
   );
 
-  expect(screen.queryByRole('button', { name: '补充剩余设定' })).toBeNull();
-});
+  fireEvent.click(screen.getByRole('button', { name: /生成草稿/u }));
 
-test('puzzle workspace does not render progress action messages as chat bubbles', () => {
-  render(
-    <PuzzleAgentWorkspace
-      session={{
-        ...baseSession,
-        messages: [
-          ...baseSession.messages,
-          {
-            id: 'message-action-result-1',
-            role: 'assistant',
-            kind: 'action_result',
-            text: '拼图结果页草稿已生成。',
-            createdAt: '2026-04-24T10:01:00.000Z',
-          },
-        ],
-      }}
-      onBack={() => {}}
-      onSubmitMessage={() => {}}
-      onExecuteAction={() => {}}
-    />,
-  );
-
-  expect(screen.getByText('画面主体已经清楚，继续收束剩余关键词。')).toBeTruthy();
-  expect(screen.queryByText('拼图结果页草稿已生成。')).toBeNull();
+  expect(onExecuteAction).toHaveBeenCalledWith({
+    action: 'compile_puzzle_draft',
+    promptText: '潮雾中的灯塔与断桥',
+    referenceImageSrc: null,
+    candidateCount: 1,
+  });
 });