chore: fix failing test

Author: ghiscoding

src/__tests__/cli.spec.ts

@@ -1,4 +1,4 @@
-import { readdir, rmSync, writeFileSync } from 'node:fs';
+import { existsSync, readdir, rmSync, writeFileSync } from 'node:fs';
 import { afterAll, afterEach, beforeEach, describe, expect, test, vi } from 'vitest';
 
 import { createDir } from '../index';
@@ -42,28 +42,43 @@ describe('copyfiles', () => {
     // Mock process.exit so it doesn't kill the test runner
     // @ts-ignore
     const exitSpy = vi.spyOn(process, 'exit').mockImplementation((code?: string | number | null | undefined) => {
-      // Do nothing
+      if (code && code !== 0) {
+        exitSpy.mockRestore();
+        done(new Error(`process.exit called with code ${code}`));
+      }
+      // Do nothing for code 0
     });
 
     import('../cli')
       .then(() => {
-        // Wait a tick to ensure file writes are complete
-        setTimeout(() => {
+        // Wait until output2/input2 exists, then check files
+        const start = Date.now();
+        const check = () => {
+          if (!existsSync('output2/input2')) {
+            if (Date.now() - start > 55) {
+              exitSpy.mockRestore();
+              return done(new Error('Timeout: output2/input2 was not created'));
+            }
+            setTimeout(check, 50);
+            return;
+          }
           readdir('output2/input2', (err, files) => {
-            expect(files).toEqual(['a.txt', 'b.txt']);
-            exitSpy.mockRestore();
-            done();
+            try {
+              expect(err).toBeNull();
+              expect(files).toEqual(['a.txt', 'b.txt']);
+              exitSpy.mockRestore();
+              done();
+            } catch (e) {
+              exitSpy.mockRestore();
+              done(e);
+            }
           });
-        }, 100); // 100ms delay to allow async file writes
+        };
+        check();
       })
       .catch(e => {
-        setTimeout(() => {
-          readdir('output2/input2', (err, files) => {
-            expect(files).toEqual(['a.txt', 'b.txt']);
-            exitSpy.mockRestore();
-            done();
-          });
-        }, 100);
+        exitSpy.mockRestore();
+        done(e);
       });
-  }))
+  }), 250);
 })