Fix flaky ClientStreaming_ResponseCompletesWithoutResponseBody… (#20338)

Author: JamesNK

src/Hosting/TestHost/src/HttpContextBuilder.cs

@@ -115,10 +115,25 @@ async Task RunRequestAsync()
                     // This could throw an error if there was a pending server read. Needs to
                     // happen before completing the response so the response returns the error.
                     var requestBodyInProgress = RequestBodyReadInProgress();
+                    if (requestBodyInProgress)
+                    {
+                        // If request is still in progress then abort it.
+                        CancelRequestBody();
+                    }
 
                     // Matches Kestrel server: response is completed before request is drained
                     await CompleteResponseAsync();
-                    await CompleteRequestAsync(requestBodyInProgress);
+
+                    if (!requestBodyInProgress)
+                    {
+                        // Writer was already completed in send request callback.
+                        await _requestPipe.Reader.CompleteAsync();
+
+                        // Don't wait for request to drain. It could block indefinitely. In a real server
+                        // we would wait for a timeout and then kill the socket.
+                        // Potential future improvement: add logging that the request timed out
+                    }
+
                     _application.DisposeContext(_testContext, exception: null);
                 }
                 catch (Exception ex)
@@ -165,24 +180,6 @@ internal void ClientInitiatedAbort()
             CancelRequestBody();
         }
 
-        private async Task CompleteRequestAsync(bool requestBodyInProgress)
-        {
-            if (requestBodyInProgress)
-            {
-                // If request is still in progress then abort it.
-                CancelRequestBody();
-            }
-            else
-            {
-                // Writer was already completed in send request callback.
-                await _requestPipe.Reader.CompleteAsync();
-            }
-
-            // Don't wait for request to drain. It could block indefinitely. In a real server
-            // we would wait for a timeout and then kill the socket.
-            // Potential future improvement: add logging that the request timed out
-        }
-
         private bool RequestBodyReadInProgress()
         {
             try