Fix IOCP count config var test

The TcpListener was being disposed too early.
dotnet · Aug 19, 2024 · ef1eea2 · ef1eea2
1 parent 623b402
commit ef1eea2
Showing 1 changed file with 15 additions and 8 deletions.
diff --git a/src/libraries/System.Threading.ThreadPool/tests/ThreadPoolTests.cs b/src/libraries/System.Threading.ThreadPool/tests/ThreadPoolTests.cs
@@ -1394,22 +1394,26 @@ static async Task RunAsyncIOTest()
                         var done = new AutoResetEvent(false);
 
                         // Receiver
+                        bool stop = false;
                         var receiveBuffer = new byte[1];
-                        using var listener = new TcpListener(IPAddress.Loopback, 0);
+                        var listener = new TcpListener(IPAddress.Loopback, 0);
                         listener.Start();
                         var t = ThreadTestHelpers.CreateGuardedThread(
                             out Action checkForThreadErrors,
                             out Action waitForThread,
                             async () =>
                             {
-                                while (true)
+                                using (listener)
                                 {
-                                    // Accept a connection, receive a byte
-                                    using var socket = await listener.AcceptSocketAsync();
-                                    int bytesRead =
-                                        await socket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), SocketFlags.None);
-                                    Assert.Equal(1, bytesRead);
-                                    done.Set(); // indicate byte received
+                                    while (!stop)
+                                    {
+                                        // Accept a connection, receive a byte
+                                        using var socket = await listener.AcceptSocketAsync();
+                                        int bytesRead =
+                                            await socket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), SocketFlags.None);
+                                        Assert.Equal(1, bytesRead);
+                                        done.Set(); // indicate byte received
+                                    }
                                 }
                             });
                         t.IsBackground = true;
@@ -1427,6 +1431,9 @@ static async Task RunAsyncIOTest()
                             Assert.Equal(1, bytesSent);
                             done.CheckedWait(); // wait for byte to the received
                         }
+
+                        stop = true;
+                        waitForThread();
                     }
                 }).Dispose();
             }, ioCompletionPortCount.ToString()).Dispose();