1

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

@@ -589,6 +589,45 @@ describe('PuzzleResultView', () => {
     });
   });
 
+  test('uses the saved level picture reference when regenerating a level image', () => {
+    const onExecuteAction = vi.fn();
+    const session = createSession({
+      draft: {
+        ...createSession().draft!,
+        levels: [
+          {
+            ...createSession().draft!.levels![0]!,
+            pictureReference: '/generated-puzzle-assets/history/saved-reference.png',
+          },
+        ],
+      },
+    });
+
+    render(
+      <PuzzleResultView
+        session={session}
+        onBack={() => {}}
+        onExecuteAction={onExecuteAction}
+      />,
+    );
+
+    fireEvent.click(screen.getByText('雨夜猫街'));
+    fireEvent.click(screen.getByRole('button', { name: /重新生成画面/u }));
+    fireEvent.click(
+      within(screen.getByRole('dialog', { name: '确认消耗光点' })).getByRole(
+        'button',
+        { name: '确定' },
+      ),
+    );
+
+    expect(onExecuteAction).toHaveBeenCalledWith(
+      expect.objectContaining({
+        action: 'generate_puzzle_images',
+        referenceImageSrc: '/generated-puzzle-assets/history/saved-reference.png',
+      }),
+    );
+  });
+
   test('passes the selected image model when regenerating a level image', () => {
     const onExecuteAction = vi.fn();