From 9040213b46c98a7ff57a2cbde1d5e6bc40871000 Mon Sep 17 00:00:00 2001
From: Andrew Casey <amcasey@users.noreply.github.com>
Date: Thu, 22 Aug 2024 15:20:16 -0700
Subject: [PATCH] Fix
 StreamPool_MultipleStreamsConcurrent_StreamsReturnedToPool

Wait for streams to complete before triggering a pool update since they won't be eligible for reuse if they haven't completed.

For #39477
---
 .../test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs b/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs
index d858d0dcfd28..011875ccec75 100644
--- a/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs
+++ b/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs
@@ -474,6 +474,8 @@ await ExpectAsync(Http2FrameType.DATA,
             withFlags: (byte)Http2DataFrameFlags.END_STREAM,
             withStreamId: 3);
 
+        await WaitForAllStreamsAsync().DefaultTimeout();
+
         // TriggerTick will trigger the stream to be returned to the pool so we can assert it
         TriggerTick();