// @vitest-environment jsdom
import {
act,
cleanup,
fireEvent,
render,
screen,
waitFor,
within,
} from '@testing-library/react';
import React from 'react';
import { afterEach, expect, it, vi } from 'vitest';
import { DesignWorkspacePanel } from '../src/features/project-workspace/DesignWorkspacePanel';
vi.mock('../src/components/ChatMarkdownMessage', () => ({
ChatMarkdownMessage: () => null,
}));
afterEach(() => {
cleanup();
vi.unstubAllEnvs();
delete window.__TAURI__;
});
it('shows imported references after a workspace event without reloading the design session', async () => {
let imported = false;
let onUpdate:
| ((event: { payload: { projectPath: string; kind: string } }) => void)
| undefined;
const invoke = vi.fn(async (command: string) => {
if (command === 'is_design_agent_debug_enabled') return false;
if (command === 'hydrate_design_agent_session') return null;
if (command === 'list_design_workspace')
return imported
? [
{ path: 'references', kind: 'directory' },
{ path: 'references/需求.md', kind: 'file' },
]
: [];
if (command === 'read_design_workspace_file') return '# 导入的需求';
throw new Error(`Unexpected command: ${command}`);
});
window.__TAURI__ = {
core: { invoke },
event: {
listen: vi.fn(async (_name, callback) => {
onUpdate = callback;
return () => undefined;
}),
},
} as unknown as typeof window.__TAURI__;
render();
await waitFor(() => expect(onUpdate).toBeDefined());
await waitFor(() =>
expect(invoke).toHaveBeenCalledWith('list_design_workspace', {
projectPath: 'imported-project',
}),
);
imported = true;
act(() =>
onUpdate!({
payload: { projectPath: 'imported-project', kind: 'workspace' },
}),
);
fireEvent.click(await screen.findByRole('button', { name: '需求.md' }));
await waitFor(() =>
expect(invoke).toHaveBeenCalledWith('read_design_workspace_file', {
projectPath: 'imported-project',
path: 'references/需求.md',
}),
);
expect(
invoke.mock.calls.filter(
([command]) => command === 'hydrate_design_agent_session',
),
).toHaveLength(1);
});
it('prepares debug fixtures from the header and refreshes the phase and files without a manual refresh', async () => {
let prepared = false;
const invoke = vi.fn(async (command: string) => {
if (command === 'is_design_agent_debug_enabled') {
return true;
}
if (command === 'debug_fast_forward_design_session') {
prepared = true;
return { activeRuntime: 'design' };
}
if (command === 'hydrate_design_agent_session') {
return { session: { currentPhase: prepared ? 'consultant' : 'concept' } };
}
if (command === 'list_design_workspace') {
return prepared ? [{ path: 'debug.md', kind: 'file', size: 10 }] : [];
}
throw new Error(`Unexpected command: ${command}`);
});
window.__TAURI__ = { core: { invoke } } as unknown as typeof window.__TAURI__;
render();
await waitFor(() =>
expect(
screen
.getByRole('button', { name: '刷新策划工作区' })
.hasAttribute('disabled'),
).toBe(false),
);
const header = screen
.getByRole('heading', { name: '概念设计' })
.closest('header')!;
fireEvent.click(
within(header).getByRole('button', { name: '快速准备做成游戏测试' }),
);
await screen.findByRole('heading', { name: '顾问' });
await screen.findByRole('button', { name: 'debug.md' });
expect(invoke).toHaveBeenCalledWith('debug_fast_forward_design_session', {
projectPath: 'test-project',
targetPhase: 'consultant',
});
});