1

src/components/custom-world-home/CustomWorldCreationHub.test.tsx

@@ -1,8 +1,8 @@
 import { renderToStaticMarkup } from 'react-dom/server';
 import { expect, test } from 'vitest';
 
-import { derivePlatformCreationTypes } from '../platform-entry/platformEntryCreationTypes';
 import type { CreationEntryConfig } from '../../services/creationEntryConfigService';
+import { derivePlatformCreationTypes } from '../platform-entry/platformEntryCreationTypes';
 import { CustomWorldCreationHub } from './CustomWorldCreationHub';
 
 const noopCreateType = () => {};
@@ -77,8 +77,9 @@ const testEntryConfig = {
   ],
 } satisfies CreationEntryConfig;
 
-const testCreationTypes = derivePlatformCreationTypes(testEntryConfig.creationTypes);
-
+const testCreationTypes = derivePlatformCreationTypes(
+  testEntryConfig.creationTypes,
+);
 
 test('creation hub draft card renders compiled work summary fields', () => {
   const html = renderToStaticMarkup(
@@ -174,6 +175,50 @@ test('creation hub renders puzzle works in the same unified list with puzzle tag
   expect(html).not.toContain('我的拼图作品');
 });
 
+test('creation hub marks generating and newly completed drafts', () => {
+  const html = renderToStaticMarkup(
+    <CustomWorldCreationHub
+      items={[]}
+      puzzleItems={[
+        {
+          workId: 'puzzle-work-session-1',
+          profileId: 'puzzle-profile-session-1',
+          ownerUserId: 'user-1',
+          authorDisplayName: '测试作者',
+          workTitle: '潮雾拼图草稿',
+          workDescription: '正在生成首张拼图主视觉。',
+          levelName: '潮雾拼图',
+          summary: '正在生成首张拼图主视觉。',
+          themeTags: ['潮雾'],
+          coverImageSrc: null,
+          publicationStatus: 'draft',
+          updatedAt: new Date('2026-04-22T10:00:00.000Z').toISOString(),
+          publishedAt: null,
+          publishReady: false,
+          sourceSessionId: 'puzzle-session-1',
+        },
+      ]}
+      loading={false}
+      error={null}
+      onRetry={() => {}}
+      onCreateType={noopCreateType}
+      onOpenDraft={() => {}}
+      onEnterPublished={() => {}}
+      entryConfig={testEntryConfig}
+      creationTypes={testCreationTypes}
+      onOpenPuzzleDetail={() => {}}
+      getWorkState={(item) =>
+        item.kind === 'puzzle'
+          ? { isGenerating: true, hasUnreadUpdate: true }
+          : null
+      }
+    />,
+  );
+
+  expect(html).toContain('生成中');
+  expect(html).toContain('aria-label="新生成完成"');
+});
+
 test('creation hub published work spans full mobile row', () => {
   const html = renderToStaticMarkup(
     <CustomWorldCreationHub