1

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

@@ -100,6 +100,7 @@ test('puzzle workspace submits the work form instead of agent chat', () => {
     workDescription: '一套雨夜猫街主题拼图。',
     pictureDescription: '一只猫在雨夜灯牌下回头。',
     referenceImageSrc: null,
+    imageModel: 'original',
   });
   expect(screen.queryByRole('button', { name: '补充剩余设定' })).toBeNull();
   expect(screen.queryByText('旧会话消息不再渲染为聊天入口。')).toBeNull();
@@ -129,10 +130,44 @@ test('puzzle workspace falls back to compile action for restored sessions', () =
     workDescription: '雾港遗迹拼图',
     pictureDescription: '潮雾中的灯塔与断桥',
     referenceImageSrc: null,
+    imageModel: 'original',
     candidateCount: 1,
   });
 });
 
+test('puzzle workspace switches the image model from the description box', () => {
+  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.change(screen.getByLabelText('画面描述'), {
+    target: { value: '一只猫在雨夜灯牌下回头。' },
+  });
+  fireEvent.click(screen.getByRole('button', { name: '图片模型' }));
+  fireEvent.click(screen.getByRole('menuitemradio', { name: 'nanobanana2' }));
+  fireEvent.click(screen.getByRole('button', { name: /生成草稿/u }));
+
+  expect(onCreateFromForm).toHaveBeenCalledWith(
+    expect.objectContaining({
+      imageModel: 'gemini-3.1-flash-image-preview',
+    }),
+  );
+});
+
 test('puzzle workspace restores form draft fields and autosaves edits', () => {
   vi.useFakeTimers();
   const onAutoSaveForm = vi.fn();
@@ -208,5 +243,6 @@ test('puzzle workspace restores form draft fields and autosaves edits', () => {
     workDescription: '旧街雨夜的拼图草稿。',
     pictureDescription: '旧街灯牌下的猫和发光雨伞。',
     referenceImageSrc: null,
+    imageModel: 'original',
   });
 });