Use vi.runAllTimersAsync()

Author: chrmarti

src/extension/prompt/node/test/chatMLFetcherRetry.spec.ts

@@ -97,8 +97,8 @@ describe('ChatMLFetcherImpl retry logic', () => {
 			mockFetcherService.queueResponse(createSuccessResponse('Hello!')); // retry
 
 			const resultPromise = fetcher.fetchMany(createBaseOpts(), cancellationTokenSource.token);
-			// Advance timers to skip the 1000ms delay before connectivity check
-			await vi.advanceTimersByTimeAsync(1000);
+			// Run all pending timers to complete the connectivity check delays
+			await vi.runAllTimersAsync();
 			const result = await resultPromise;
 
 			expect(result.type).toBe(ChatFetchResponseType.Success);
@@ -112,7 +112,7 @@ describe('ChatMLFetcherImpl retry logic', () => {
 			mockFetcherService.queueResponse(createSuccessResponse('Success!')); // retry
 
 			const resultPromise = fetcher.fetchMany(createBaseOpts(), cancellationTokenSource.token);
-			await vi.advanceTimersByTimeAsync(1000);
+			await vi.runAllTimersAsync();
 			const result = await resultPromise;
 
 			expect(result.type).toBe(ChatFetchResponseType.Success);
@@ -163,7 +163,7 @@ describe('ChatMLFetcherImpl retry logic', () => {
 			mockFetcherService.queueResponse(createSuccessResponse('Success!')); // retry
 
 			const resultPromise = fetcher.fetchMany(createBaseOpts(), cancellationTokenSource.token);
-			await vi.advanceTimersByTimeAsync(1000);
+			await vi.runAllTimersAsync();
 			const result = await resultPromise;
 
 			expect(result.type).toBe(ChatFetchResponseType.Success);
@@ -191,7 +191,7 @@ describe('ChatMLFetcherImpl retry logic', () => {
 			mockFetcherService.queueResponse(createSuccessResponse('Success!')); // retry
 
 			const resultPromise = fetcher.fetchMany(createBaseOpts(), cancellationTokenSource.token);
-			await vi.advanceTimersByTimeAsync(1000);
+			await vi.runAllTimersAsync();
 			const result = await resultPromise;
 
 			expect(result.type).toBe(ChatFetchResponseType.Success);
@@ -206,7 +206,7 @@ describe('ChatMLFetcherImpl retry logic', () => {
 
 			// Should still retry on 500 even with invalid entry in config
 			const resultPromise = fetcher.fetchMany(createBaseOpts(), cancellationTokenSource.token);
-			await vi.advanceTimersByTimeAsync(1000);
+			await vi.runAllTimersAsync();
 			const result = await resultPromise;
 
 			expect(result.type).toBe(ChatFetchResponseType.Success);
@@ -249,8 +249,8 @@ describe('ChatMLFetcherImpl retry logic', () => {
 			mockFetcherService.queueError(createNetworkError('ENOTFOUND')); // 3rd connectivity check
 
 			const resultPromise = fetcher.fetchMany(createBaseOpts(), cancellationTokenSource.token);
-			// Advance through all connectivity check delays: 1000 + 10000 + 10000 = 21000ms
-			await vi.advanceTimersByTimeAsync(21000);
+			// Run all pending timers to complete all connectivity check retries
+			await vi.runAllTimersAsync();
 			const result = await resultPromise;
 
 			// Should fail because connectivity check never succeeded