1

src/components/puzzle-result/PuzzleResultView.test.tsx

@@ -232,13 +232,16 @@ describe('PuzzleResultView', () => {
     fireEvent.change(within(dialog).getByLabelText('画面描述'), {
       target: { value: '一只猫在雨夜灯牌下回头。' },
     });
-    fireEvent.click(within(dialog).getByRole('button', { name: /重新生成画面/u }));
+    fireEvent.click(
+      within(dialog).getByRole('button', { name: /重新生成画面/u }),
+    );
 
     expect(onExecuteAction).toHaveBeenCalledWith({
       action: 'generate_puzzle_images',
       levelId: 'puzzle-level-1',
       promptText: '一只猫在雨夜灯牌下回头。',
       referenceImageSrc: undefined,
+      imageModel: 'original',
       candidateCount: 1,
       levelsJson: expect.any(String),
     });
@@ -295,9 +298,13 @@ describe('PuzzleResultView', () => {
 
     fireEvent.click(screen.getByRole('button', { name: /新增关卡/u }));
     const dialog = screen.getByRole('dialog', { name: '关卡详情' });
-    expect(within(dialog).getByRole('button', { name: /生成画面/u })).toBeTruthy();
+    expect(
+      within(dialog).getByRole('button', { name: /生成画面/u }),
+    ).toBeTruthy();
     expect(within(dialog).queryByText('画面图')).toBeNull();
-    expect(within(dialog).queryByRole('button', { name: /关卡测试/u })).toBeNull();
+    expect(
+      within(dialog).queryByRole('button', { name: /关卡测试/u }),
+    ).toBeNull();
     fireEvent.click(screen.getByLabelText('关闭'));
     expect(screen.getAllByText('第2关').length).toBeGreaterThan(0);
 
@@ -358,6 +365,7 @@ describe('PuzzleResultView', () => {
       levelId: 'puzzle-level-1775000000000-2',
       promptText: '新关卡里有一座发光钟楼。',
       referenceImageSrc: undefined,
+      imageModel: 'original',
       candidateCount: 1,
       levelsJson: expect.any(String),
     });
@@ -457,8 +465,38 @@ describe('PuzzleResultView', () => {
       levelId: 'puzzle-level-1',
       promptText: '屋檐下的猫与暖灯街角。',
       referenceImageSrc: '/generated-puzzle-assets/history/image.png',
+      imageModel: 'original',
       candidateCount: 1,
       levelsJson: expect.any(String),
     });
   });
+
+  test('passes the selected image model when regenerating a level image', () => {
+    const onExecuteAction = vi.fn();
+
+    render(
+      <PuzzleResultView
+        session={createSession()}
+        onBack={() => {}}
+        onExecuteAction={onExecuteAction}
+      />,
+    );
+
+    fireEvent.click(screen.getByText('雨夜猫街'));
+    const dialog = screen.getByRole('dialog', { name: '关卡详情' });
+    fireEvent.click(within(dialog).getByRole('button', { name: '图片模型' }));
+    fireEvent.click(
+      within(dialog).getByRole('menuitemradio', { name: 'gpt-image-2' }),
+    );
+    fireEvent.click(
+      within(dialog).getByRole('button', { name: /重新生成画面/u }),
+    );
+
+    expect(onExecuteAction).toHaveBeenCalledWith(
+      expect.objectContaining({
+        action: 'generate_puzzle_images',
+        imageModel: 'gpt-image-2',
+      }),
+    );
+  });
 });