1

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

@@ -239,9 +239,30 @@ describe('PuzzleResultView', () => {
       promptText: '一只猫在雨夜灯牌下回头。',
       referenceImageSrc: undefined,
       candidateCount: 1,
+      levelsJson: expect.any(String),
     });
+    const generatePayload = onExecuteAction.mock.calls[0]![0];
+    expect(JSON.parse(generatePayload.levelsJson ?? '[]')).toEqual([
+      expect.objectContaining({
+        levelId: 'puzzle-level-1',
+        levelName: '暖灯猫街',
+        pictureDescription: '一只猫在雨夜灯牌下回头。',
+      }),
+    ]);
 
-    fireEvent.click(within(dialog).getByRole('button', { name: /体验该关/u }));
+    const levelNameInput = within(dialog).getByLabelText('关卡名称');
+    const formalImageTitle = within(dialog).getByText('画面图');
+    const pictureDescriptionInput = within(dialog).getByLabelText('画面描述');
+    expect(
+      levelNameInput.compareDocumentPosition(formalImageTitle) &
+        Node.DOCUMENT_POSITION_FOLLOWING,
+    ).toBeTruthy();
+    expect(
+      formalImageTitle.compareDocumentPosition(pictureDescriptionInput) &
+        Node.DOCUMENT_POSITION_FOLLOWING,
+    ).toBeTruthy();
+
+    fireEvent.click(within(dialog).getByRole('button', { name: /关卡测试/u }));
     expect(onStartTestRun).toHaveBeenCalledWith(
       expect.objectContaining({
         levelName: '暖灯猫街',
@@ -272,7 +293,10 @@ describe('PuzzleResultView', () => {
     );
 
     fireEvent.click(screen.getByRole('button', { name: /新增关卡/u }));
-    expect(screen.getByRole('dialog', { name: '关卡详情' })).toBeTruthy();
+    const dialog = screen.getByRole('dialog', { name: '关卡详情' });
+    expect(within(dialog).getByRole('button', { name: /生成画面/u })).toBeTruthy();
+    expect(within(dialog).queryByText('画面图')).toBeNull();
+    expect(within(dialog).queryByRole('button', { name: /关卡测试/u })).toBeNull();
     fireEvent.click(screen.getByLabelText('关闭'));
     expect(screen.getAllByText('第2关').length).toBeGreaterThan(0);
 
@@ -285,7 +309,7 @@ describe('PuzzleResultView', () => {
       expect.objectContaining({
         levels: expect.arrayContaining([
           expect.objectContaining({ levelId: 'puzzle-level-1' }),
-          expect.objectContaining({ levelName: '第2关' }),
+          expect.objectContaining({ levelName: '' }),
         ]),
       }),
     );
@@ -309,6 +333,45 @@ describe('PuzzleResultView', () => {
     );
   });
 
+  test('generates image for a newly added level with the current levels snapshot', () => {
+    vi.spyOn(Date, 'now').mockReturnValue(1_775_000_000_000);
+    const onExecuteAction = vi.fn();
+
+    render(
+      <PuzzleResultView
+        session={createSession()}
+        onBack={() => {}}
+        onExecuteAction={onExecuteAction}
+      />,
+    );
+
+    fireEvent.click(screen.getByRole('button', { name: /新增关卡/u }));
+    const dialog = screen.getByRole('dialog', { name: '关卡详情' });
+    fireEvent.change(within(dialog).getByLabelText('画面描述'), {
+      target: { value: '新关卡里有一座发光钟楼。' },
+    });
+    fireEvent.click(within(dialog).getByRole('button', { name: /生成画面/u }));
+
+    expect(onExecuteAction).toHaveBeenCalledWith({
+      action: 'generate_puzzle_images',
+      levelId: 'puzzle-level-1775000000000-2',
+      promptText: '新关卡里有一座发光钟楼。',
+      referenceImageSrc: undefined,
+      candidateCount: 1,
+      levelsJson: expect.any(String),
+    });
+
+    const payload = onExecuteAction.mock.calls[0]![0];
+    expect(JSON.parse(payload.levelsJson ?? '[]')).toEqual([
+      expect.objectContaining({ levelId: 'puzzle-level-1' }),
+      expect.objectContaining({
+        levelId: 'puzzle-level-1775000000000-2',
+        levelName: '',
+        pictureDescription: '新关卡里有一座发光钟楼。',
+      }),
+    ]);
+  });
+
   test('publishes with work info and serialized levels', () => {
     const onExecuteAction = vi.fn();
 
@@ -394,6 +457,7 @@ describe('PuzzleResultView', () => {
       promptText: '屋檐下的猫与暖灯街角。',
       referenceImageSrc: '/generated-puzzle-assets/history/image.png',
       candidateCount: 1,
+      levelsJson: expect.any(String),
     });
   });
 });