From ac07ea6583cfa7e6896c2fe97b595f2a395ed3ea Mon Sep 17 00:00:00 2001
From: Ahmet Ibrahim Aksoy <aaksoy@microsoft.com>
Date: Thu, 21 Mar 2024 20:30:41 +0100
Subject: [PATCH] [QUIC] Make sure that we get the RECEIVE before connection
 close (#100093)

---
 .../System.Net.Quic/tests/FunctionalTests/QuicStreamTests.cs  | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/libraries/System.Net.Quic/tests/FunctionalTests/QuicStreamTests.cs b/src/libraries/System.Net.Quic/tests/FunctionalTests/QuicStreamTests.cs
index e02851fac73ee..bdbf155d2d1ea 100644
--- a/src/libraries/System.Net.Quic/tests/FunctionalTests/QuicStreamTests.cs
+++ b/src/libraries/System.Net.Quic/tests/FunctionalTests/QuicStreamTests.cs
@@ -1252,6 +1252,8 @@ await RunClientServer(
                     serverSem.Release();
                     await clientSem.WaitAsync();
 
+                    var _ = await stream.ReadAsync(new byte[0]);
+
                     if (closeServer)
                     {
                         expectedError = QuicError.OperationAborted;
@@ -1282,6 +1284,8 @@ await RunClientServer(
                     clientSem.Release();
                     await serverSem.WaitAsync();
 
+                    var _ = await stream.ReadAsync(new byte[0]);
+
                     if (!closeServer)
                     {
                         expectedError = QuicError.OperationAborted;