1

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

@@ -100,8 +100,9 @@ test('puzzle workspace submits the work form instead of agent chat', () => {
     workDescription: '一套雨夜猫街主题拼图。',
     pictureDescription: '一只猫在雨夜灯牌下回头。',
     referenceImageSrc: null,
-    imageModel: 'original',
+    imageModel: 'gpt-image-2',
   });
+  expect(screen.getByText('消耗2光点')).toBeTruthy();
   expect(screen.queryByRole('button', { name: '补充剩余设定' })).toBeNull();
   expect(screen.queryByText('旧会话消息不再渲染为聊天入口。')).toBeNull();
 });
@@ -130,7 +131,7 @@ test('puzzle workspace falls back to compile action for restored sessions', () =
     workDescription: '雾港遗迹拼图',
     pictureDescription: '潮雾中的灯塔与断桥',
     referenceImageSrc: null,
-    imageModel: 'original',
+    imageModel: 'gpt-image-2',
     candidateCount: 1,
   });
 });
@@ -158,6 +159,7 @@ test('puzzle workspace switches the image model from the description box', () =>
     target: { value: '一只猫在雨夜灯牌下回头。' },
   });
   fireEvent.click(screen.getByRole('button', { name: '图片模型' }));
+  expect(screen.queryByRole('menuitemradio', { name: '原模型' })).toBeNull();
   fireEvent.click(screen.getByRole('menuitemradio', { name: 'nanobanana2' }));
   fireEvent.click(screen.getByRole('button', { name: /生成草稿/u }));
 
@@ -243,6 +245,6 @@ test('puzzle workspace restores form draft fields and autosaves edits', () => {
     workDescription: '旧街雨夜的拼图草稿。',
     pictureDescription: '旧街灯牌下的猫和发光雨伞。',
     referenceImageSrc: null,
-    imageModel: 'original',
+    imageModel: 'gpt-image-2',
   });
 });