GetParam instead of sleep

microsoft · Nov 30, 2024 · 83e1616 · 83e1616
1 parent 3d09ffe
commit 83e1616
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/test/lib/DataTest.cpp b/src/test/lib/DataTest.cpp
@@ -3815,7 +3815,18 @@ void ConnectionPriorityCommon(ConnectionPriorityTestType* ConnectionPriorityTest
             TEST_TRUE(Connections[i]->HandshakeCompleteEvent.WaitTimeout(TestWaitTimeout));
             TEST_TRUE(Connections[i]->HandshakeComplete);
         }
-        CxPlatSleep(300); // Flush operations
+
+        // GetParam to flush operations in Connection.
+        // Sometimes processing state Connection messes up test results.
+        for (uint8_t i = 0; i < NumConnections; ++i) {
+            QUIC_ADDR Addr;
+            uint32_t AddrSize = sizeof(QUIC_ADDR);
+            TEST_QUIC_SUCCEEDED(
+                Connections[i]->GetParam(
+                QUIC_PARAM_CONN_LOCAL_ADDRESS,
+                &StatSize,
+                &BaseStat));
+        }
 
         ConnectionPriorityTest(Connections, NumConnections, Streams, Buffer);