use original

Author: mizdra

packages/codegen/src/runner.test.ts

@@ -13,7 +13,7 @@ async function sleep(ms: number): Promise<void> {
 async function waitFor(fn: () => void) {
   return new Promise<void>((resolve, reject) => {
     let error: unknown = null;
-    const intervalTimer = setInterval(() => {
+    function runFn() {
       try {
         fn();
         clearInterval(intervalTimer);
@@ -22,6 +22,10 @@ async function waitFor(fn: () => void) {
       } catch (e) {
         error = e;
       }
+    }
+    runFn(); // First, execute immediately
+    const intervalTimer = setInterval(() => {
+      runFn();
     }, 50);
     const timeoutTimer = setTimeout(() => {
       clearInterval(intervalTimer);
@@ -68,7 +72,7 @@ describe('runCMKInWatchMode', () => {
     console.log('update file');
     await writeFile(textFilePath, '1');
     const startTime = Date.now();
-    await vi.waitFor(() => {
+    await waitFor(() => {
       console.log('elapsed', Date.now() - startTime);
       assert(globalThis.changeCount === 1, `Expected changeCount to be 1, but got ${globalThis.changeCount}`);
     });