Fix unit test consistency (#333)

libp2p · Sep 28, 2023 · 48a66ab · 48a66ab
1 parent a90e169
commit 48a66ab
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/libp2p/src/test/kotlin/io/libp2p/mux/yamux/YamuxHandlerTest.kt b/libp2p/src/test/kotlin/io/libp2p/mux/yamux/YamuxHandlerTest.kt
@@ -261,8 +261,11 @@ class YamuxHandlerTest : MuxHandlerAbstractTest() {
 
         val range = 1..messagesToSend
 
-        // 100 window updates should be sent to ensure buffer is flushed and all messages are sent so will send them at random times
-        val windowUpdatesIndices = (range).shuffled().take(100).toSet()
+        // 100 window updates should be sent to ensure buffer is flushed and all messages are sent
+        // so will send them at random times ensuring maxBufferedConnectionWrites can never be reached
+        val windowUpdatesIndices = (range).chunked(100).flatMap {
+            it.shuffled().take(20)
+        }
 
         for (i in range) {
             if (i in windowUpdatesIndices) {