HTTP/3: Test abort at end of request (#43183)

Author: JamesNK

src/Servers/Kestrel/test/Interop.FunctionalTests/Http3/Http3RequestTests.cs

@@ -537,6 +537,58 @@ public async Task POST_ServerAbort_ClientReceivesAbort(HttpProtocols protocol)
         }
     }
 
+    [ConditionalFact]
+    [MsQuicSupported]
+    public async Task POST_ServerAbortAfterWrite_ClientReceivesAbort()
+    {
+        // Arrange
+        var builder = CreateHostBuilder(async context =>
+        {
+            Logger.LogInformation("Server writing content.");
+            await context.Response.Body.WriteAsync(new byte[16]);
+
+            // Note that there is a race here on what is sent before the abort is processed.
+            // Abort may happen before or after response headers have been sent.
+            Logger.LogInformation("Server aborting.");
+            context.Abort();
+        }, protocol: HttpProtocols.Http3);
+
+        using (var host = builder.Build())
+        using (var client = HttpHelpers.CreateClient())
+        {
+            await host.StartAsync().DefaultTimeout();
+
+            for (var i = 0; i < 100; i++)
+            {
+                Logger.LogInformation($"Client sending request {i}");
+
+                var request = new HttpRequestMessage(HttpMethod.Get, $"https://127.0.0.1:{host.GetPort()}/");
+                request.Version = GetProtocol(HttpProtocols.Http3);
+                request.VersionPolicy = HttpVersionPolicy.RequestVersionExact;
+
+                // Act
+                var sendTask = client.SendAsync(request, CancellationToken.None);
+
+                // Assert
+                var ex = await Assert.ThrowsAsync<HttpRequestException>(async () =>
+                {
+                    // Note that there is a race here on what is sent before the abort is processed.
+                    // Abort may happen before or after response headers have been sent.
+                    Logger.LogInformation($"Client awaiting response {i}");
+                    var response = await sendTask;
+
+                    Logger.LogInformation($"Client awaiting content {i}");
+                    await response.Content.ReadAsByteArrayAsync();
+                }).DefaultTimeout();
+
+                var protocolException = ex.GetProtocolException();
+                Assert.Equal((long)Http3ErrorCode.InternalError, protocolException.ErrorCode);
+            }
+
+            await host.StopAsync().DefaultTimeout();
+        }
+    }
+
     // Verify HTTP/2 and HTTP/3 match behavior
     [ConditionalTheory]
     [MsQuicSupported]

src/Servers/Kestrel/test/Interop.FunctionalTests/HttpHelpers.cs

@@ -19,6 +19,22 @@ namespace Interop.FunctionalTests;
 
 internal static class HttpHelpers
 {
+    public static HttpProtocolException GetProtocolException(this Exception ex)
+    {
+        var current = ex;
+        while (current != null)
+        {
+            if (current is HttpProtocolException httpProtocolException)
+            {
+                return httpProtocolException;
+            }
+
+            current = current.InnerException;
+        }
+
+        throw new Exception($"Couldn't find {nameof(HttpProtocolException)}. Original error: {ex}");
+    }
+
     public static HttpMessageInvoker CreateClient(TimeSpan? idleTimeout = null, TimeSpan? expect100ContinueTimeout = null, bool includeClientCert = false)
     {
         var handler = new SocketsHttpHandler();