1

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

@@ -249,6 +249,7 @@ describe('PuzzleResultView', () => {
       promptText: '一只猫在雨夜灯牌下回头。',
       referenceImageSrc: undefined,
       imageModel: 'gpt-image-2',
+      aiRedraw: true,
       candidateCount: 1,
       workTitle: '暖灯猫街作品',
       workDescription: '一套雨夜猫街主题拼图。',
@@ -263,8 +264,15 @@ describe('PuzzleResultView', () => {
         levelId: 'puzzle-level-1',
         levelName: '暖灯猫街',
         pictureDescription: '一只猫在雨夜灯牌下回头。',
+        generationStatus: 'generating',
       }),
     ]);
+    expect(within(dialog).getByText('预计剩余 90 秒')).toBeTruthy();
+    expect(
+      within(dialog).queryByPlaceholderText('参考图链接或资产ID'),
+    ).toBeNull();
+    const levelList = screen.getByLabelText('拼图关卡列表');
+    expect(within(levelList).getAllByText('生成中').length).toBeGreaterThan(0);
 
     const levelNameInput = within(dialog).getByLabelText('关卡名称');
     const formalImageTitle = within(dialog).getByText('画面图');
@@ -314,7 +322,10 @@ describe('PuzzleResultView', () => {
       within(dialog).getByRole('button', { name: /生成画面/u }),
     ).toBeTruthy();
     expect(within(dialog).getByText('消耗2光点')).toBeTruthy();
-    expect(within(dialog).queryByText('画面图')).toBeNull();
+    expect(
+      within(dialog).getByText('等待时间可以制作更多关卡哦~'),
+    ).toBeTruthy();
+    expect(within(dialog).getByText('画面图')).toBeTruthy();
     expect(
       within(dialog).queryByRole('button', { name: /关卡测试/u }),
     ).toBeNull();
@@ -385,6 +396,7 @@ describe('PuzzleResultView', () => {
       promptText: '新关卡里有一座发光钟楼。',
       referenceImageSrc: undefined,
       imageModel: 'gpt-image-2',
+      aiRedraw: true,
       candidateCount: 1,
       workTitle: '暖灯猫街作品',
       workDescription: '一套雨夜猫街主题拼图。',
@@ -400,10 +412,90 @@ describe('PuzzleResultView', () => {
         levelId: 'puzzle-level-1775000000000-2',
         levelName: '',
         pictureDescription: '新关卡里有一座发光钟楼。',
+        generationStatus: 'generating',
       }),
     ]);
   });
 
+  test('keeps generation progress visible after closing and reopening level dialog', () => {
+    const onExecuteAction = vi.fn();
+
+    render(
+      <PuzzleResultView
+        session={createSession()}
+        onBack={() => {}}
+        onExecuteAction={onExecuteAction}
+      />,
+    );
+
+    fireEvent.click(screen.getByText('雨夜猫街'));
+    fireEvent.click(screen.getByRole('button', { name: /重新生成画面/u }));
+    fireEvent.click(
+      within(screen.getByRole('dialog', { name: '确认消耗光点' })).getByRole(
+        'button',
+        { name: '确定' },
+      ),
+    );
+    fireEvent.click(screen.getByLabelText('关闭'));
+
+    expect(
+      within(screen.getByLabelText('拼图关卡列表')).getAllByText('生成中')
+        .length,
+    ).toBeGreaterThan(0);
+
+    fireEvent.click(screen.getByText('雨夜猫街'));
+    const reopenedDialog = screen.getByRole('dialog', { name: '关卡详情' });
+    expect(
+      within(reopenedDialog).getByRole('progressbar', { name: '画面生成进度' }),
+    ).toBeTruthy();
+    expect(within(reopenedDialog).getByText('预计剩余 90 秒')).toBeTruthy();
+  });
+
+  test('allows parallel draft editing while a level image is generating but blocks publish', () => {
+    const onExecuteAction = vi.fn();
+    const onStartTestRun = vi.fn();
+
+    render(
+      <PuzzleResultView
+        session={createSession()}
+        onBack={() => {}}
+        onExecuteAction={onExecuteAction}
+        onStartTestRun={onStartTestRun}
+      />,
+    );
+
+    fireEvent.click(screen.getByText('雨夜猫街'));
+    fireEvent.click(screen.getByRole('button', { name: /重新生成画面/u }));
+    fireEvent.click(
+      within(screen.getByRole('dialog', { name: '确认消耗光点' })).getByRole(
+        'button',
+        { name: '确定' },
+      ),
+    );
+    fireEvent.change(screen.getByLabelText('关卡名称'), {
+      target: { value: '继续编辑的猫街' },
+    });
+    fireEvent.click(screen.getByRole('button', { name: /关卡测试/u }));
+
+    expect(onStartTestRun).toHaveBeenCalledWith(
+      expect.objectContaining({
+        levelName: '继续编辑的猫街',
+      }),
+    );
+
+    fireEvent.click(screen.getByLabelText('关闭'));
+    fireEvent.click(screen.getByRole('button', { name: /新增关卡/u }));
+    expect(screen.getByRole('dialog', { name: '关卡详情' })).toBeTruthy();
+    fireEvent.click(screen.getByLabelText('关闭'));
+
+    fireEvent.click(screen.getByRole('button', { name: /发布/u }));
+    const publishDialog = screen.getByRole('dialog', { name: '发布拼图作品' });
+    expect(within(publishDialog).getByText('还有关卡画面正在生成。')).toBeTruthy();
+    expect(
+      within(publishDialog).getByRole('button', { name: '发布到广场' }),
+    ).toHaveProperty('disabled', true);
+  });
+
   test('publishes with work info and serialized levels', () => {
     const onExecuteAction = vi.fn();
 
@@ -552,19 +644,26 @@ describe('PuzzleResultView', () => {
     );
 
     fireEvent.click(screen.getByText('雨夜猫街'));
-    fireEvent.click(screen.getByLabelText('从历史拼图素材库选择'));
+    const dialog = screen.getByRole('dialog', { name: '关卡详情' });
+    const uploadInput = within(dialog).getByLabelText('上传参考图', {
+      selector: 'input',
+    });
+    expect(uploadInput.closest('.platform-subpanel')).toBeTruthy();
+    const historyButton = within(dialog).getByRole('button', {
+      name: '选择历史图片',
+    });
+    expect(within(historyButton).getByText('历史')).toBeTruthy();
+    fireEvent.click(historyButton);
 
     const picker = await screen.findByRole('dialog', {
-      name: '选择历史拼图素材',
+      name: '选择历史图片',
     });
     fireEvent.click(
       await within(picker).findByRole('button', { name: /账号 user-1/u }),
     );
 
     await waitFor(() => {
-      expect(
-        screen.queryByRole('dialog', { name: '选择历史拼图素材' }),
-      ).toBeNull();
+      expect(screen.queryByRole('dialog', { name: '选择历史图片' })).toBeNull();
     });
 
     fireEvent.click(screen.getByRole('button', { name: /重新生成画面/u }));
@@ -580,6 +679,7 @@ describe('PuzzleResultView', () => {
       promptText: '屋檐下的猫与暖灯街角。',
       referenceImageSrc: '/generated-puzzle-assets/history/image.png',
       imageModel: 'gpt-image-2',
+      aiRedraw: true,
       candidateCount: 1,
       workTitle: '暖灯猫街作品',
       workDescription: '一套雨夜猫街主题拼图。',
@@ -624,8 +724,10 @@ describe('PuzzleResultView', () => {
       expect.objectContaining({
         action: 'generate_puzzle_images',
         referenceImageSrc: '/generated-puzzle-assets/history/saved-reference.png',
+        aiRedraw: true,
       }),
     );
+    expect(screen.queryByPlaceholderText('参考图链接或资产ID')).toBeNull();
   });
 
   test('passes the selected image model when regenerating a level image', () => {