perf(tests): replace fixed highlight waits with ink polls and trim remaining sleeps

Author: Gustav-Simonsson

scripts/ci-timings.json

@@ -104,7 +104,7 @@ describe("runWithWatchdog", () => {
       command: process.execPath,
       args: [
         "-e",
-        'process.on("SIGTERM", () => {}); setTimeout(() => process.exit(3), 600);',
+        'process.on("SIGTERM", () => {}); setTimeout(() => process.exit(3), 400);',
       ],
       stallMs: SILENT_CHILD_STALL_MS,
       onStall: () => {

scripts/test-parallel.test.ts

@@ -57,7 +57,7 @@ describe("runGuardedShell", () => {
     const feed = createShellOutputFeed();
     let finished = false;
     const running = runGuardedShell(
-      { command: "echo first; sleep 0.02; echo second; sleep 0.4" },
+      { command: "echo first; sleep 0.02; echo second; sleep 0.2" },
       neverAbort(),
       undefined,
       undefined,
@@ -85,7 +85,7 @@ describe("runGuardedShell", () => {
   test("omitted timeout does not arm a timer", async () => {
     const start = Date.now();
     const { exitCode, timedOut, output } = await runGuardedShell(
-      { command: "sleep 0.25; echo done" },
+      { command: "sleep 0.1; echo done" },
       neverAbort(),
     );
     expect(timedOut).toBe(false);
@@ -118,7 +118,7 @@ describe("runGuardedShell", () => {
   test("returns partial output and a timed-out flag instead of throwing", async () => {
     const start = Date.now();
     const { exitCode, timedOut, output } = await runGuardedShell(
-      { command: "echo early; sleep 60", timeout: 200 },
+      { command: "echo early; sleep 60", timeout: 100 },
       neverAbort(),
     );
     expect(timedOut).toBe(true);
@@ -350,7 +350,7 @@ describe("background run_shell (shellGuardPlugin)", () => {
     const result = await runWith(registry, {
       id: "bg1",
       name: "run_shell",
-      arguments: { command: "sleep 0.2; echo finished", background: true },
+      arguments: { command: "sleep 0.05; echo finished", background: true },
     });
     expect(result.isError).toBeUndefined();
     const parsed = JSON.parse(String(result.content)) as {
@@ -656,11 +656,11 @@ describe("shellGuardPlugin", () => {
     const result = await run({
       id: "c2",
       name: "run_shell",
-      arguments: { command: "echo before; sleep 60", timeout: 120 },
+      arguments: { command: "echo before; sleep 60", timeout: 80 },
     });
     expect(result.isError).toBeUndefined();
     expect(result.content).toContain("before");
-    expect(result.content).toMatch(/timed out after 120ms and was terminated/);
+    expect(result.content).toMatch(/timed out after 80ms and was terminated/);
     expect(result.content).toContain("background:true");
   });
 
@@ -684,13 +684,13 @@ describe("shellGuardPlugin", () => {
 
   test("applies a configured default timeout when none is passed", async () => {
     const handler = defined(
-      shellGuardPlugin(process.cwd(), { defaultMs: 90 }).middleware,
+      shellGuardPlugin(process.cwd(), { defaultMs: 60 }).middleware,
     )(fallback);
     const result = await handler(
       { id: "c2c", name: "run_shell", arguments: { command: "sleep 60" } },
       neverAbort(),
     );
-    expect(result.content).toMatch(/timed out after 90ms/);
+    expect(result.content).toMatch(/timed out after 60ms/);
   });
 
   test("omitted timeout with no settings default still completes a short command", async () => {
@@ -701,7 +701,7 @@ describe("shellGuardPlugin", () => {
       {
         id: "c2d",
         name: "run_shell",
-        arguments: { command: "sleep 0.2; echo ok" },
+        arguments: { command: "sleep 0.05; echo ok" },
       },
       neverAbort(),
     );
@@ -1044,7 +1044,7 @@ describe("shellGuardPlugin", () => {
 
   test("treats timeout 0 as the configured default", async () => {
     const handler = defined(
-      shellGuardPlugin(process.cwd(), { defaultMs: 90, maxMs: 100 }).middleware,
+      shellGuardPlugin(process.cwd(), { defaultMs: 60, maxMs: 70 }).middleware,
     )(fallback);
     const result = await handler(
       {
@@ -1054,7 +1054,7 @@ describe("shellGuardPlugin", () => {
       },
       neverAbort(),
     );
-    expect(result.content).toMatch(/timed out after 90ms/);
+    expect(result.content).toMatch(/timed out after 60ms/);
   });
 
   test("returns promptly when the search tool ignores the budget", async () => {

src/plugins/shell-guard-plugin.test.ts

@@ -103,7 +103,7 @@ describe("lifecycle hook payload delivery", () => {
     process.on("unhandledRejection", onUnhandled);
     try {
       await manager.dispatchPostRun(summary);
-      await new Promise((resolve) => setTimeout(resolve, 100));
+      await new Promise((resolve) => setTimeout(resolve, 40));
     } finally {
       process.off("unhandledRejection", onUnhandled);
     }