Add test isolation for process.exit in handleRunError function

workers/main/src/index.test.ts

@@ -6,21 +6,28 @@ describe('handleRunError', () => {
   let processExitSpy: ReturnType<typeof vi.spyOn>;
 
   beforeEach(() => {
+    // Use fake timers to control setTimeout in handleRunError
+    vi.useFakeTimers();
     // Mock process.exit to prevent actual process termination during tests
-    processExitSpy = vi.spyOn(process, 'exit').mockImplementation(() => {
-      throw new Error('process.exit called');
-    });
+    processExitSpy = vi
+      .spyOn(process, 'exit')
+      .mockImplementation((() => {}) as never);
   });
 
   afterEach(() => {
+    // Run all pending timers before restoring mocks to prevent race condition
+    vi.runAllTimers();
     processExitSpy.mockRestore();
+    // Restore real timers
+    vi.useRealTimers();
   });
 
   it('should log the error', () => {
     const error = new Error('test error');
     const logSpy = vi.spyOn(logger, 'error').mockImplementation(() => {});
 
     handleRunError(error);
+
     expect(logSpy).toHaveBeenCalledWith(
       `Error in main worker: ${error.message}`,
     );
@@ -46,4 +53,22 @@ describe('handleRunError', () => {
     );
     logSpy.mockRestore();
   });
+
+  it('should call process.exit with code 1 after timeout', () => {
+    const error = new Error('test error');
+    const logSpy = vi.spyOn(logger, 'error').mockImplementation(() => {});
+
+    handleRunError(error);
+
+    // Process.exit should not be called immediately
+    expect(processExitSpy).not.toHaveBeenCalled();
+
+    // Fast-forward time by 100ms
+    vi.advanceTimersByTime(100);
+
+    // Now process.exit should have been called
+    expect(processExitSpy).toHaveBeenCalledWith(1);
+
+    logSpy.mockRestore();
+  });
 });