Genarrative/apps/ai-game-creator-shell/tests/sessionPreview.test.ts

import { describe, expect, it, vi } from 'vitest';

import type { GameCreationAppPreviewState } from '../../../packages/shared/src/contracts/gameCreationApp';
import { resolveSessionPreviewOnProjectOpen } from '../src/features/app-shell/sessionPreview';

const projectPath = '/tmp/session-preview-project';

function createInvoke(
  responses: Record<string, unknown>,
  calls: Array<[string, Record<string, unknown> | undefined]> = [],
) {
  return vi.fn(async (command: string, args?: Record<string, unknown>) => {
    calls.push([command, args]);
    if (!(command in responses)) {
      throw new Error(`unexpected invoke ${command}`);
    }
    return responses[command];
  });
}

describe('进入项目时的预览活体核验', () => {
  it('registry 里没有活体时不给会话预览，并把陈旧的 running 记录对齐成 stopped', async () => {
    const calls: Array<[string, Record<string, unknown> | undefined]> = [];
    const invoke = createInvoke(
      {
        get_local_game_preview_status: {
          status: 'stopped',
          url: null,
          port: null,
          root: null,
        },
        stop_local_game_preview: {
          status: 'stopped',
          url: null,
          port: null,
          root: null,
        },
      },
      calls,
    );

    const resolution = await resolveSessionPreviewOnProjectOpen({
      invoke,
      projectPath,
      recordedPreview: {
        status: 'running',
        url: 'http://127.0.0.1:4173/',
        port: 4173,
      },
    });

    // 没有活体预览：本次会话不带预览（工作台因此不会自动进运行视图）。
    expect(resolution.sessionPreview).toBeNull();
    // 落盘记录被判陈旧，会话 manifest 立刻按真相投影。
    expect(resolution.manifestPreviewPatch).toEqual({ status: 'stopped' });
    expect(calls).toEqual([
      ['get_local_game_preview_status', { projectPath }],
      ['stop_local_game_preview', { projectPath }],
    ]);
  });

  it('registry 确实在跑时只认活体，且不写落盘记录', async () => {
    const calls: Array<[string, Record<string, unknown> | undefined]> = [];
    const invoke = createInvoke(
      {
        get_local_game_preview_status: {
          status: 'running',
          url: 'http://127.0.0.1:5210/',
          port: 5210,
          root: projectPath,
        },
      },
      calls,
    );

    const resolution = await resolveSessionPreviewOnProjectOpen({
      invoke,
      projectPath,
      // 落盘记录落后于真相（例如同项目重启后记录没跟上）也不影响：活体说了算。
      recordedPreview: { status: 'stopped' },
    });

    expect(resolution.sessionPreview).toEqual({
      status: 'running',
      url: 'http://127.0.0.1:5210/',
      port: 5210,
    });
    expect(resolution.manifestPreviewPatch).toBeNull();
    expect(calls).toEqual([['get_local_game_preview_status', { projectPath }]]);
  });

  it('registry 不可读时按没有在跑处理，不把落盘记录当活体', async () => {
    const invoke = vi.fn(async (command: string) => {
      if (command === 'get_local_game_preview_status') {
        throw new Error('需要在陶泥儿客户端内运行');
      }
      return { status: 'stopped', url: null, port: null, root: null };
    });

    const resolution = await resolveSessionPreviewOnProjectOpen({
      invoke,
      projectPath,
      recordedPreview: {
        status: 'running',
        url: 'http://127.0.0.1:4173/',
        port: 4173,
      },
    });

    expect(resolution.sessionPreview).toBeNull();
    expect(resolution.manifestPreviewPatch).toEqual({ status: 'stopped' });
  });

  it('记录本来就不是 running 时只读活体，不发多余的停止命令', async () => {
    const calls: Array<[string, Record<string, unknown> | undefined]> = [];
    const invoke = createInvoke(
      {
        get_local_game_preview_status: {
          status: 'stopped',
          url: null,
          port: null,
          root: null,
        },
      },
      calls,
    );

    const stoppedRecord: GameCreationAppPreviewState = { status: 'stopped' };
    const resolution = await resolveSessionPreviewOnProjectOpen({
      invoke,
      projectPath,
      recordedPreview: stoppedRecord,
    });

    expect(resolution).toEqual({
      sessionPreview: null,
      manifestPreviewPatch: null,
    });
    expect(calls).toEqual([['get_local_game_preview_status', { projectPath }]]);
  });

  it('没有 invoke（浏览器里跑）时不做任何请求', async () => {
    const resolution = await resolveSessionPreviewOnProjectOpen({
      invoke: null,
      projectPath,
      recordedPreview: {
        status: 'running',
        url: 'http://127.0.0.1:4173/',
        port: 4173,
      },
    });

    expect(resolution).toEqual({
      sessionPreview: null,
      manifestPreviewPatch: null,
    });
  });
});