test: pin which monitors WaitFor waits for

Author: chengluyu

apps/kimi-code/test/e2e/tui-monitor.e2e.test.ts

@@ -12,6 +12,7 @@ import { promisify } from 'node:util';
 import { afterAll, describe, expect, it, vi } from 'vitest';
 
 import {
+  bash,
   foreground,
   hasTmux,
   monitor,
@@ -71,8 +72,8 @@ describe.skipIf(!ENABLED)('TUI e2e — Monitor', () => {
     MONITOR_ON,
   ));
 
-  it('WaitFor returns as soon as a monitor prints', scenario(
-    [monitor('sleep 2; echo "tests passed"; sleep 60'), waitFor(60), say('Tests passed, moving on.')],
+  it('WaitFor on a monitor returns as soon as it prints', scenario(
+    [monitor('sleep 2; echo "tests passed"; sleep 60'), waitForMonitor(60, 'ticker'), say('Tests passed, moving on.')],
     async ({ tui, model }) => {
       await tui.submit('Watch the tests and wait for the result');
       await tui.see('Wait ended by monitor output', 'wait-ended-by-event');
@@ -84,6 +85,28 @@ describe.skipIf(!ENABLED)('TUI e2e — Monitor', () => {
     MONITOR_ON,
   ));
 
+  it('WaitFor without a task id waits for the build, not for monitor output', scenario(
+    [
+      bash('sleep 6; echo build ok'),
+      monitor(`${gate('crash')}; echo "dev: 500 on /api"; sleep 60`, 'dev server'),
+      waitFor(30),
+      say('Build done; now the 500.'),
+    ],
+    async ({ tui, model, workDir }) => {
+      await tui.submit('Build, and keep an eye on the dev server');
+      await tui.see(/Waiting \d+s/);
+      writeFileSync(join(workDir, 'crash'), '');
+      await pause(2_500);
+      expect(await tui.screen()).toMatch(/Waiting \d+s/);
+      await tui.see('Build done; now the 500.', 'build-then-event');
+
+      expect(model.toolResult(3)).toContain('wait_status: completed');
+      expect(model.toolResult(3)).toContain('build ok');
+      expect(model.userTexts(3).join('\n')).toContain('dev: 500 on /api');
+    },
+    MONITOR_ON,
+  ));
+
   it('A monitor that floods output is stopped', scenario(
     [monitor('yes flood', 'flood'), say('Started.'), say('It was stopped.'), say('Noted.')],
     async ({ tui, model }) => {