From a05d31993aa0014e638fbc8da25988d439119838 Mon Sep 17 00:00:00 2001
From: Ahmet Ibrahim Aksoy <aaksoy@microsoft.com>
Date: Thu, 21 Mar 2024 18:49:04 +0100
Subject: [PATCH] Make sure that we get the RECEIVE before connection close

---
 .../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 e02851fac73ee9..bdbf155d2d1ead 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;