Fix AbortedStream_ResetsAndDrainsRequest_RefusesFramesAfterCooldownExpires (#9487)

Author: jkotalik

src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2TestBase.cs

@@ -1163,6 +1163,13 @@ internal void VerifyGoAway(Http2Frame frame, int expectedLastStreamId, Http2Erro
 
         internal async Task WaitForConnectionErrorAsync<TException>(bool ignoreNonGoAwayFrames, int expectedLastStreamId, Http2ErrorCode expectedErrorCode, params string[] expectedErrorMessage)
             where TException : Exception
+        {
+            await WaitForConnectionErrorAsyncDoNotCloseTransport<TException>(ignoreNonGoAwayFrames, expectedLastStreamId, expectedErrorCode, expectedErrorMessage);
+            _pair.Application.Output.Complete();
+        }
+
+        internal async Task WaitForConnectionErrorAsyncDoNotCloseTransport<TException>(bool ignoreNonGoAwayFrames, int expectedLastStreamId, Http2ErrorCode expectedErrorCode, params string[] expectedErrorMessage)
+            where TException : Exception
         {
             var frame = await ReceiveFrameAsync();
 
@@ -1184,7 +1191,7 @@ internal async Task WaitForConnectionErrorAsync<TException>(bool ignoreNonGoAway
             }
 
             await _connectionTask.DefaultTimeout();
-            _pair.Application.Output.Complete();
+            TestApplicationErrorLogger.LogInformation("Stopping Connection From ConnectionErrorAsync");
         }
 
         internal async Task WaitForStreamErrorAsync(int expectedStreamId, Http2ErrorCode expectedErrorCode, string expectedErrorMessage)

src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2TimeoutTests.cs

@@ -11,6 +11,7 @@
 using Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Infrastructure;
 using Microsoft.AspNetCore.Testing;
 using Microsoft.AspNetCore.Testing.xunit;
+using Microsoft.Extensions.Logging.Testing;
 using Microsoft.Net.Http.Headers;
 using Moq;
 using Xunit;
@@ -193,11 +194,13 @@ public async Task ResponseDrain_SlowerThanMinimumDataRate_AbortsConnection()
         }
 
         [Theory]
-        [Flaky("https://github.com/aspnet/AspNetCore-Internal/issues/1879", FlakyOn.All)]
+        [Repeat(20)]
         [InlineData((int)Http2FrameType.DATA)]
         [InlineData((int)Http2FrameType.CONTINUATION)]
         public async Task AbortedStream_ResetsAndDrainsRequest_RefusesFramesAfterCooldownExpires(int intFinalFrameType)
         {
+            var closeLock = new object();
+            var closed = false;
             var finalFrameType = (Http2FrameType)intFinalFrameType;
             // Remove callback that completes _pair.Application.Output on abort.
             _mockConnectionContext.Reset();
@@ -216,8 +219,6 @@ public async Task AbortedStream_ResetsAndDrainsRequest_RefusesFramesAfterCooldow
 
             await WaitForStreamErrorAsync(1, Http2ErrorCode.INTERNAL_ERROR, "The connection was aborted by the application.");
 
-            var cts = new CancellationTokenSource();
-
             async Task AdvanceClockAndSendFrames()
             {
                 if (finalFrameType == Http2FrameType.CONTINUATION)
@@ -227,7 +228,7 @@ async Task AdvanceClockAndSendFrames()
                 }
 
                 // There's a race when the appfunc is exiting about how soon it unregisters the stream, so retry until success.
-                while (!cts.Token.IsCancellationRequested)
+                while (!closed)
                 {
                     // Just past the timeout
                     mockSystemClock.UtcNow += Constants.RequestBodyDrainTimeout + TimeSpan.FromTicks(1);
@@ -247,24 +248,24 @@ async Task AdvanceClockAndSendFrames()
                             throw new NotImplementedException(finalFrameType.ToString());
                     }
 
-                    if (!cts.Token.IsCancellationRequested)
-                    {
-                        await Task.Delay(10);
-                    }
+                    // TODO how do I force a function to go async?
+                    await Task.Delay(1);
                 }
             }
 
             var sendTask = AdvanceClockAndSendFrames();
 
-            await WaitForConnectionErrorAsync<Http2ConnectionErrorException>(
+            await WaitForConnectionErrorAsyncDoNotCloseTransport<Http2ConnectionErrorException>(
                 ignoreNonGoAwayFrames: false,
                 expectedLastStreamId: 1,
                 Http2ErrorCode.STREAM_CLOSED,
                 CoreStrings.FormatHttp2ErrorStreamClosed(finalFrameType, 1));
 
-            cts.Cancel();
+            closed = true;
 
             await sendTask.DefaultTimeout();
+
+            _pair.Application.Output.Complete();
         }
 
         [Fact]