Update RetryHttpHandlerTest.cs

Author: jadewang-db

csharp/test/Drivers/Apache/Spark/RetryHttpHandlerTest.cs

@@ -48,25 +48,25 @@ public async Task RetryAfterHandlerProcesses503Response()
 
             // Create the RetryHttpHandler with retry enabled and a 5-second timeout
             var retryHandler = new RetryHttpHandler(mockHandler, true, 5);
-            
+
             // Create an HttpClient with our handler
             var httpClient = new HttpClient(retryHandler);
-            
+
             // Set the mock handler to return a success response after the first retry
             mockHandler.SetResponseAfterRetryCount(1, new HttpResponseMessage(HttpStatusCode.OK)
             {
                 Content = new StringContent("Success")
             });
-            
+
             // Send a request
             var response = await httpClient.GetAsync("http://test.com");
-            
+
             // Verify the response is OK
             Assert.Equal(HttpStatusCode.OK, response.StatusCode);
             Assert.Equal("Success", await response.Content.ReadAsStringAsync());
             Assert.Equal(2, mockHandler.RequestCount); // Initial request + 1 retry
         }
-        
+
         /// <summary>
         /// Tests that the RetryHttpHandler throws an exception when the retry timeout is exceeded.
         /// </summary>
@@ -83,22 +83,22 @@ public async Task RetryAfterHandlerThrowsWhenTimeoutExceeded()
 
             // Create the RetryHttpHandler with retry enabled and a 1-second timeout
             var retryHandler = new RetryHttpHandler(mockHandler, true, 1);
-            
+
             // Create an HttpClient with our handler
             var httpClient = new HttpClient(retryHandler);
-            
+
             // Send a request and expect an AdbcException
-            var exception = await Assert.ThrowsAsync<AdbcException>(async () => 
+            var exception = await Assert.ThrowsAsync<AdbcException>(async () =>
                 await httpClient.GetAsync("http://test.com"));
-            
+
             // Verify the exception has the correct SQL state in the message
             Assert.Contains("[SQLState: 08001]", exception.Message);
             Assert.Equal(AdbcStatusCode.IOError, exception.Status);
-            
+
             // Verify we only tried once (since the Retry-After value of 2 exceeds our timeout of 1)
             Assert.Equal(1, mockHandler.RequestCount);
         }
-        
+
         /// <summary>
         /// Tests that the RetryHttpHandler doesn't retry when retry is disabled.
         /// </summary>
@@ -115,19 +115,19 @@ public async Task RetryAfterHandlerDoesNotRetryWhenDisabled()
 
             // Create the RetryHttpHandler with retry disabled
             var retryHandler = new RetryHttpHandler(mockHandler, false, 5);
-            
+
             // Create an HttpClient with our handler
             var httpClient = new HttpClient(retryHandler);
-            
+
             // Send a request
             var response = await httpClient.GetAsync("http://test.com");
-            
+
             // Verify the response is 503
             Assert.Equal(HttpStatusCode.ServiceUnavailable, response.StatusCode);
             Assert.Equal("Service Unavailable", await response.Content.ReadAsStringAsync());
             Assert.Equal(1, mockHandler.RequestCount); // Only the initial request, no retries
         }
-        
+
         /// <summary>
         /// Tests that the RetryHttpHandler handles non-503 responses correctly.
         /// </summary>
@@ -143,19 +143,19 @@ public async Task RetryAfterHandlerHandlesNon503Response()
 
             // Create the RetryHttpHandler with retry enabled
             var retryHandler = new RetryHttpHandler(mockHandler, true, 5);
-            
+
             // Create an HttpClient with our handler
             var httpClient = new HttpClient(retryHandler);
-            
+
             // Send a request
             var response = await httpClient.GetAsync("http://test.com");
-            
+
             // Verify the response is 404
             Assert.Equal(HttpStatusCode.NotFound, response.StatusCode);
             Assert.Equal("Not Found", await response.Content.ReadAsStringAsync());
             Assert.Equal(1, mockHandler.RequestCount); // Only the initial request, no retries
         }
-        
+
         /// <summary>
         /// Tests that the RetryHttpHandler handles 503 responses without Retry-After headers correctly.
         /// </summary>
@@ -171,19 +171,19 @@ public async Task RetryAfterHandlerHandles503WithoutRetryAfterHeader()
 
             // Create the RetryHttpHandler with retry enabled
             var retryHandler = new RetryHttpHandler(mockHandler, true, 5);
-            
+
             // Create an HttpClient with our handler
             var httpClient = new HttpClient(retryHandler);
-            
+
             // Send a request
             var response = await httpClient.GetAsync("http://test.com");
-            
+
             // Verify the response is 503
             Assert.Equal(HttpStatusCode.ServiceUnavailable, response.StatusCode);
             Assert.Equal("Service Unavailable", await response.Content.ReadAsStringAsync());
             Assert.Equal(1, mockHandler.RequestCount); // Only the initial request, no retries
         }
-        
+
         /// <summary>
         /// Tests that the RetryHttpHandler handles invalid Retry-After headers correctly.
         /// </summary>
@@ -207,19 +207,19 @@ public async Task RetryAfterHandlerHandlesInvalidRetryAfterHeader()
 
             // Create the RetryHttpHandler with retry enabled
             var retryHandler = new RetryHttpHandler(mockHandler, true, 5);
-            
+
             // Create an HttpClient with our handler
             var httpClient = new HttpClient(retryHandler);
-            
+
             // Send a request
             response = await httpClient.GetAsync("http://test.com");
-            
+
             // Verify the response is 503
             Assert.Equal(HttpStatusCode.ServiceUnavailable, response.StatusCode);
             Assert.Equal("Service Unavailable", await response.Content.ReadAsStringAsync());
             Assert.Equal(1, mockHandler.RequestCount); // Only the initial request, no retries
         }
-        
+
         /// <summary>
         /// Mock HttpMessageHandler for testing the RetryHttpHandler.
         /// </summary>
@@ -228,38 +228,38 @@ private class MockHttpMessageHandler : HttpMessageHandler
             private readonly HttpResponseMessage _defaultResponse;
             private HttpResponseMessage? _responseAfterRetryCount;
             private int _retryCountForResponse;
-            
+
             public int RequestCount { get; private set; }
-            
+
             public MockHttpMessageHandler(HttpResponseMessage defaultResponse)
             {
                 _defaultResponse = defaultResponse;
             }
-            
+
             public void SetResponseAfterRetryCount(int retryCount, HttpResponseMessage response)
             {
                 _retryCountForResponse = retryCount;
                 _responseAfterRetryCount = response;
             }
-            
+
             protected override Task<HttpResponseMessage> SendAsync(
                 HttpRequestMessage request,
                 CancellationToken cancellationToken)
             {
                 RequestCount++;
-                
+
                 if (_responseAfterRetryCount != null && RequestCount > _retryCountForResponse)
                 {
                     return Task.FromResult(_responseAfterRetryCount);
                 }
-                
+
                 // Create a new response instance to avoid modifying the original
                 var response = new HttpResponseMessage
                 {
                     StatusCode = _defaultResponse.StatusCode,
                     Content = _defaultResponse.Content
                 };
-                
+
                 // Copy headers only if they exist
                 if (_defaultResponse.Headers.Contains("Retry-After"))
                 {
@@ -268,9 +268,9 @@ protected override Task<HttpResponseMessage> SendAsync(
                         response.Headers.Add("Retry-After", value);
                     }
                 }
-                
+
                 return Task.FromResult(response);
             }
         }
     }
-}
+}