Add callback

src/handlers/testing/exec/components/progress/index.tsx

@@ -136,7 +136,7 @@ function TestRow(props: {
   );
 }
 
-const App = () => {
+const App = (props: { onListenersCreated: () => void }) => {
   const [consoleLogs, setConsoleLogs] = useState<ConsoleLog[]>([]);
   const [uncaughtErrors, setUncaughtErrors] = useState<UncaughtError[]>([]);
   const [evals, setEvals] = useState<Evaluation[]>([]);
@@ -174,6 +174,8 @@ const App = () => {
     emitter.on(EventName.EVALUATION, evalListener);
     emitter.on(EventName.RUN_ENDED, runEndListener);
 
+    props.onListenersCreated();
+
     return () => {
       emitter.off(EventName.CONSOLE_LOG, consoleLogListener);
       emitter.off(EventName.UNCAUGHT_ERROR, uncaughtErrorListener);
@@ -237,4 +239,5 @@ const App = () => {
   );
 };
 
-export const renderTestProgress = () => render(<App />);
+export const renderTestProgress = (args: { onListenersCreated: () => void }) =>
+  render(<App onListenersCreated={args.onListenersCreated} />);

src/handlers/testing/exec/index.ts

@@ -476,12 +476,13 @@ export async function exec(args: {
   port: number;
   exit1OnEvaluationFailure: boolean;
 }) {
-  renderTestProgress();
+  let listenersCreated = false;
+  renderTestProgress({ onListenersCreated: () => (listenersCreated = true) });
 
-  // Wait for listeners to be set up in the progress component
-  await new Promise((resolve) => {
-    setTimeout(resolve, 10);
-  });
+  // Wait for listeners to be created before starting the server
+  while (!listenersCreated) {
+    await new Promise((resolve) => setTimeout(resolve, 10));
+  }
 
   const runManager = new RunManager({
     apiKey: args.apiKey,