use vi

Author: mizdra

packages/codegen/src/runner.test.ts

@@ -3,7 +3,7 @@ import { writeFile } from 'node:fs/promises';
 import { join } from 'node:path';
 import { platform } from 'node:process';
 import chokidar from 'chokidar';
-import { describe, test } from 'vitest';
+import { describe, test, vi } from 'vitest';
 
 async function sleep(ms: number): Promise<void> {
   // eslint-disable-next-line no-promise-executor-return
@@ -68,7 +68,7 @@ describe('runCMKInWatchMode', () => {
     console.log('update file');
     await writeFile(textFilePath, '1');
     const startTime = Date.now();
-    await waitFor(() => {
+    await vi.waitFor(() => {
       console.log('elapsed', Date.now() - startTime);
       assert(globalThis.changeCount === 1, `Expected changeCount to be 1, but got ${globalThis.changeCount}`);
     });