fix: flaky test

Author: schummar

src/index.ts

@@ -112,6 +112,7 @@ export async function runp<const TCommands extends Commands = Commands>(
   const results = await Promise.all(tasks.map((task) => task.result));
   await new Promise<void>((resolve) => setTimeout(resolve));
   stop?.();
+  await new Promise<void>((resolve) => setTimeout(resolve));
   return results as { [K in keyof TCommands]: RunpResult };
 }
 

test/subCommands.test.ts

@@ -1,9 +1,8 @@
-import { expect, test } from 'vitest';
+import { test } from 'vitest';
 import { runp } from '../src';
-import { poll, TestTerminal } from './_helpers';
-import { setTimeout } from 'timers/promises';
+import { TestTerminal } from './_helpers';
 
-test('subCommands', async () => {
+test('subCommands', async ({ expect }) => {
   const term = new TestTerminal({ cols: 30, rows: 20 });
 
   await runp({
@@ -28,9 +27,7 @@ test('subCommands', async () => {
     linearOutput: true,
   });
 
-  await poll(
-    () =>
-      expect(term.getBuffer()).toMatchInlineSnapshot(`
+  expect(term.getBuffer()).toMatchInlineSnapshot(`
     [
       "                              ",
       "-- [subCommand 1] ->          ",
@@ -53,7 +50,5 @@ test('subCommands', async () => {
       "                              ",
       "                              ",
     ]
-  `),
-    1000,
-  );
+  `);
 });