fixup! KTOR-2036 Fix CIO connection limit

ktorio · Aug 31, 2022 · 01c2830 · 01c2830
1 parent 82ac773
commit 01c2830
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/ktor-client/ktor-client-cio/jvm/test/io/ktor/client/engine/cio/CIORequestTest.kt b/ktor-client/ktor-client-cio/jvm/test/io/ktor/client/engine/cio/CIORequestTest.kt
@@ -5,6 +5,7 @@
 package io.ktor.client.engine.cio
 
 import io.ktor.client.call.*
+import io.ktor.client.network.sockets.*
 import io.ktor.client.plugins.*
 import io.ktor.client.request.*
 import io.ktor.client.statement.*
@@ -147,13 +148,19 @@ class CIORequestTest : TestWithKtor() {
         }
 
         test { client ->
+            var fail: Throwable? = null
             for (i in 0..1000) {
                 try {
                     client.get("http://something.wrong").body<String>()
-                } catch (cause: UnresolvedAddressException) {
-                    // ignore
+                } catch (cause: Throwable) {
+                    fail = cause
                 }
             }
+
+            assertNotNull(fail)
+            if (fail !is ConnectTimeoutException && fail !is UnresolvedAddressException) {
+                fail("Expected ConnectTimeoutException or UnresolvedAddressException, got $fail", fail)
+            }
         }
     }
 }