101 lines
3.5 KiB
TypeScript
101 lines
3.5 KiB
TypeScript
import { closeElectronApp, expect, getStableWindow, installIpcMocks, test } from './fixtures/electron';
|
|
|
|
const MAIN_SESSION_KEY = 'agent:main:main';
|
|
|
|
function stableStringify(value: unknown): string {
|
|
if (value == null || typeof value !== 'object') return JSON.stringify(value);
|
|
if (Array.isArray(value)) return `[${value.map((item) => stableStringify(item)).join(',')}]`;
|
|
const entries = Object.entries(value as Record<string, unknown>)
|
|
.sort(([left], [right]) => left.localeCompare(right))
|
|
.map(([key, entryValue]) => `${JSON.stringify(key)}:${stableStringify(entryValue)}`);
|
|
return `{${entries.join(',')}}`;
|
|
}
|
|
|
|
test.describe('ClawX chat run state events', () => {
|
|
test('keeps stop control active across non-terminal gateway phase end', async ({ launchElectronApp }) => {
|
|
const app = await launchElectronApp({ skipSetup: true });
|
|
|
|
try {
|
|
await installIpcMocks(app, {
|
|
gatewayStatus: { state: 'running', port: 18789, pid: 12345, gatewayReady: true },
|
|
gatewayRpc: {
|
|
[stableStringify(['sessions.list', { includeDerivedTitles: true, includeLastMessage: true }])]: {
|
|
success: true,
|
|
result: {
|
|
sessions: [{ key: MAIN_SESSION_KEY, displayName: 'main' }],
|
|
},
|
|
},
|
|
[stableStringify(['chat.history', { sessionKey: MAIN_SESSION_KEY, limit: 200 }])]: {
|
|
success: true,
|
|
result: { messages: [] },
|
|
},
|
|
[stableStringify(['chat.send', null])]: {
|
|
success: true,
|
|
result: { runId: 'run-e2e' },
|
|
},
|
|
},
|
|
hostApi: {
|
|
[stableStringify(['/api/gateway/status', 'GET'])]: {
|
|
ok: true,
|
|
data: {
|
|
status: 200,
|
|
ok: true,
|
|
json: { state: 'running', port: 18789, pid: 12345, gatewayReady: true },
|
|
},
|
|
},
|
|
[stableStringify(['/api/agents', 'GET'])]: {
|
|
ok: true,
|
|
data: {
|
|
status: 200,
|
|
ok: true,
|
|
json: { success: true, agents: [{ id: 'main', name: 'Main' }] },
|
|
},
|
|
},
|
|
},
|
|
});
|
|
|
|
const page = await getStableWindow(app);
|
|
try {
|
|
await page.reload();
|
|
} catch (error) {
|
|
if (!String(error).includes('ERR_FILE_NOT_FOUND')) {
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
await expect(page.getByTestId('chat-composer-input')).toBeEnabled({ timeout: 30_000 });
|
|
await page.getByTestId('chat-composer-input').fill('run long task');
|
|
await page.getByTestId('chat-composer-send').click();
|
|
await expect(page.getByTestId('chat-composer-send')).toHaveAttribute('title', 'Stop');
|
|
|
|
await app.evaluate(({ BrowserWindow }) => {
|
|
BrowserWindow.getAllWindows()[0]?.webContents.send('gateway:notification', {
|
|
method: 'agent',
|
|
params: {
|
|
runId: 'run-e2e',
|
|
sessionKey: 'agent:main:main',
|
|
data: { phase: 'end' },
|
|
},
|
|
});
|
|
});
|
|
|
|
await expect(page.getByTestId('chat-composer-send')).toHaveAttribute('title', 'Stop');
|
|
|
|
await app.evaluate(({ BrowserWindow }) => {
|
|
BrowserWindow.getAllWindows()[0]?.webContents.send('gateway:notification', {
|
|
method: 'agent',
|
|
params: {
|
|
runId: 'run-e2e',
|
|
sessionKey: 'agent:main:main',
|
|
data: { phase: 'completed' },
|
|
},
|
|
});
|
|
});
|
|
|
|
await expect(page.getByTestId('chat-composer-send')).toHaveAttribute('title', 'Send');
|
|
} finally {
|
|
await closeElectronApp(app);
|
|
}
|
|
});
|
|
});
|