Fix race in FrameAfterTrailers_UnexpectedFrameError #30754 (#31061)

dotnet · Mar 19, 2021 · cc33b22 · cc33b22
1 parent de013d4
commit cc33b22
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http3/Http3StreamTests.cs b/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http3/Http3StreamTests.cs
@@ -1685,6 +1685,10 @@ public async Task FrameAfterTrailers_UnexpectedFrameError()
             var requestStream = await InitializeConnectionAndStreamsAsync(_noopApplication);
 
             await requestStream.SendHeadersAsync(headers, endStream: false);
+
+            // The app no-ops quickly. Wait for it here so it's not a race with the error response.
+            await requestStream.ExpectHeadersAsync();
+
             await requestStream.SendDataAsync(Encoding.UTF8.GetBytes("Hello world"));
             await requestStream.SendHeadersAsync(trailers, endStream: false);
             await requestStream.SendDataAsync(Encoding.UTF8.GetBytes("This is invalid."));