1

src/components/rpg-entry/useRpgCreationEnterWorld.test.tsx

@@ -4,7 +4,7 @@ import { act, render } from '@testing-library/react';
 import { describe, expect, it, vi } from 'vitest';
 
 import type { CustomWorldAgentSessionSnapshot } from '../../../packages/shared/src/contracts/customWorldAgent';
-import { WorldType, type CustomWorldProfile } from '../../types';
+import { type CustomWorldProfile, WorldType } from '../../types';
 import { useRpgCreationEnterWorld } from './useRpgCreationEnterWorld';
 
 function buildProfile(params: {
@@ -88,7 +88,11 @@ function buildSession(): CustomWorldAgentSessionSnapshot {
     stage: 'ready_to_publish',
     focusCardId: null,
     creatorIntent: null,
-    creatorIntentReadiness: { isReady: true, completedKeys: [], missingKeys: [] },
+    creatorIntentReadiness: {
+      isReady: true,
+      completedKeys: [],
+      missingKeys: [],
+    },
     anchorPack: null,
     lockState: null,
     draftProfile: null,
@@ -110,15 +114,15 @@ function buildSession(): CustomWorldAgentSessionSnapshot {
 }
 
 describe('useRpgCreationEnterWorld', () => {
-  it('Agent 草稿进入游戏时使用 session draft profile 的角色形象', async () => {
+  it('Agent 草稿测试进入游戏时使用结果页当前 profile 的角色形象', async () => {
     const staleResultProfile = buildProfile({
       id: 'stale-result',
       name: '旧结果页快照',
       imageSrc: '/template/old-role.png',
     });
-    const draftProfile = buildProfile({
+    const resultProfile = buildProfile({
       id: 'draft-profile',
-      name: '草稿真相源',
+      name: '结果页真相源',
       imageSrc: '/generated-characters/draft-role/portrait.png',
     });
     const handleCustomWorldSelect = vi.fn();
@@ -130,7 +134,7 @@ describe('useRpgCreationEnterWorld', () => {
         isAgentDraftResultView: true,
         activeAgentSessionId: 'session-1',
         generatedCustomWorldProfile: staleResultProfile,
-        agentSessionProfile: draftProfile,
+        agentSessionProfile: resultProfile,
         agentSession: buildSession(),
         handleCustomWorldSelect,
         executePublishWorld,
@@ -138,7 +142,10 @@ describe('useRpgCreationEnterWorld', () => {
       });
 
       return (
-        <button type="button" onClick={() => void enterWorldForTestFromCurrentResult()}>
+        <button
+          type="button"
+          onClick={() => void enterWorldForTestFromCurrentResult()}
+        >
           进入
         </button>
       );
@@ -150,9 +157,12 @@ describe('useRpgCreationEnterWorld', () => {
     });
 
     expect(executePublishWorld).not.toHaveBeenCalled();
-    expect(handleCustomWorldSelect).toHaveBeenCalledWith(draftProfile);
-    expect(handleCustomWorldSelect.mock.calls[0]?.[0].playableNpcs[0]?.imageSrc).toBe(
-      '/generated-characters/draft-role/portrait.png',
-    );
+    expect(handleCustomWorldSelect).toHaveBeenCalledWith(resultProfile, {
+      mode: 'test',
+      returnStage: 'custom-world-result',
+    });
+    expect(
+      handleCustomWorldSelect.mock.calls[0]?.[0].playableNpcs[0]?.imageSrc,
+    ).toBe('/generated-characters/draft-role/portrait.png');
   });
 });