1

src/components/custom-world-agent/CustomWorldAgentWorkspace.interaction.test.tsx

@@ -160,6 +160,25 @@ test('workspace exposes draft action when progress reaches 100', async () => {
   });
 });
 
+test('workspace hides draft action before progress reaches 100', () => {
+  render(
+    <CustomWorldAgentWorkspace
+      session={{
+        ...baseSession,
+        progressPercent: 99,
+      }}
+      activeOperation={null}
+      onBack={() => {}}
+      onSubmitMessage={() => {}}
+      onExecuteAction={() => {}}
+    />,
+  );
+
+  expect(
+    screen.queryByRole('button', { name: '生成游戏设定草稿' }),
+  ).toBeNull();
+});
+
 test('workspace submits recommended reply from thread', async () => {
   const user = userEvent.setup();
   const onSubmitMessage = vi.fn();