Make ClientLoader work with multiple engines on js/wasmJs and so test…

… CIO engine on nodejs(js+wasmJs)

ktor-client/ktor-client-plugins/ktor-client-auth/common/test/io/ktor/client/plugins/auth/AuthTest.kt

@@ -23,7 +23,7 @@ import kotlin.test.assertFailsWith
 class AuthTest : ClientLoader() {
 
     @Test
-    fun testDigestAuthLegacy() = clientTests(listOf("Js", "native")) {
+    fun testDigestAuthLegacy() = clientTests(listOf("Js", "native:*")) {
         config {
             install(Auth) {
                 digest {
@@ -43,7 +43,7 @@ class AuthTest : ClientLoader() {
     }
 
     @Test
-    fun testDigestAuth() = clientTests(listOf("Js", "native")) {
+    fun testDigestAuth() = clientTests(listOf("Js", "native:*")) {
         config {
             install(Auth) {
                 digest {
@@ -60,7 +60,7 @@ class AuthTest : ClientLoader() {
     }
 
     @Test
-    fun testDigestAuthPerRealm() = clientTests(listOf("Js", "native")) {
+    fun testDigestAuthPerRealm() = clientTests(listOf("Js", "native:*")) {
         config {
             install(Auth) {
                 digest {
@@ -84,7 +84,7 @@ class AuthTest : ClientLoader() {
     }
 
     @Test
-    fun testDigestAuthSHA256() = clientTests(listOf("Js", "native")) {
+    fun testDigestAuthSHA256() = clientTests(listOf("Js", "native:*")) {
         config {
             install(Auth) {
                 digest {

ktor-client/ktor-client-tests/common/src/io/ktor/client/tests/utils/ClientLoader.kt

@@ -5,12 +5,20 @@
 package io.ktor.client.tests.utils
 
 import io.ktor.client.engine.*
-import kotlinx.coroutines.test.*
+import kotlinx.coroutines.test.TestResult
+import kotlinx.coroutines.test.runTest
+import kotlin.time.Duration
+import kotlin.time.Duration.Companion.minutes
+
+internal expect val enginesToTest: Iterable<HttpClientEngineFactory<HttpClientEngineConfig>>
+internal expect val platformName: String
+internal expect fun platformDumpCoroutines()
+internal expect fun platformWaitForAllCoroutines()
 
 /**
  * Helper interface to test client.
  */
-expect abstract class ClientLoader(timeoutSeconds: Int = 60) {
+abstract class ClientLoader(private val timeout: Duration = 1.minutes) {
     /**
      * Perform test against all clients from dependencies.
      */
@@ -19,10 +27,110 @@ expect abstract class ClientLoader(timeoutSeconds: Int = 60) {
         onlyWithEngine: String? = null,
         retries: Int = 1,
         block: suspend TestClientBuilder<HttpClientEngineConfig>.() -> Unit
-    ): TestResult
+    ): TestResult = runTest(timeout = timeout) {
+        val skipPatterns = skipEngines.map(SkipEnginePattern::parse)
+
+        val failures: List<TestFailure> = enginesToTest.mapNotNull { engineFactory ->
+            val engineName = engineFactory.engineName
+
+            if (shouldRun(engineName, skipPatterns, onlyWithEngine)) {
+                try {
+                    println("Run test with engine $engineName")
+                    // run test here
+                    performTestWithEngine(engineFactory, this@ClientLoader, retries, block)
+                    null // engine test passed
+                } catch (cause: Throwable) {
+                    // engine test failed, save failure to report after run for every engine.
+                    TestFailure(engineName, cause)
+                }
+            } else {
+                println("Skipping test with engine $engineName")
+                null // engine skipped
+            }
+        }
+
+        if (failures.isNotEmpty()) {
+            val message = buildString {
+                appendLine("Test failed for engines: ${failures.map { it.engineName }}")
+                failures.forEach {
+                    appendLine("Test failed for engine '$platformName:${it.engineName}' with:")
+                    appendLine(it.cause.stackTraceToString().prependIndent("  "))
+                }
+            }
+            throw AssertionError(message)
+        }
+    }
+
+    private fun shouldRun(
+        engineName: String,
+        skipEnginePatterns: List<SkipEnginePattern>,
+        onlyWithEngine: String?
+    ): Boolean {
+        val lowercaseEngineName = engineName.lowercase()
+        if (onlyWithEngine != null && onlyWithEngine.lowercase() != lowercaseEngineName) return false
+
+        skipEnginePatterns.forEach {
+            if (it.matches(lowercaseEngineName)) return false
+        }
+
+        return true
+    }
 
     /**
      * Print coroutines in debug mode.
      */
-    fun dumpCoroutines()
+    fun dumpCoroutines(): Unit = platformDumpCoroutines()
+
+    // Issues to fix before unlocking:
+    // 1. Pinger & Ponger in ws
+    // 2. Nonce generator
+    // @After
+    fun waitForAllCoroutines(): Unit = platformWaitForAllCoroutines()
 }
+
+internal val HttpClientEngineFactory<*>.engineName: String
+    get() = this::class.simpleName!!
+
+private data class SkipEnginePattern(
+    val skippedPlatform: String?, // null means * or empty
+    val skippedEngine: String?, // null means * or empty
+) {
+    fun matches(engineName: String): Boolean {
+        var result = true
+        if (skippedEngine != null) {
+            result = result && engineName == skippedEngine
+        }
+        if (result && skippedPlatform != null) {
+            result = result && platformName.startsWith(skippedPlatform)
+        }
+        return result
+    }
+
+    companion object {
+        fun parse(pattern: String): SkipEnginePattern {
+            val parts = pattern.lowercase().split(":").map { it.takeIf { it != "*" } }
+            val platform: String?
+            val engine: String?
+            when (parts.size) {
+                1 -> {
+                    platform = null
+                    engine = parts[0]
+                }
+
+                2 -> {
+                    platform = parts[0]
+                    engine = parts[1]
+                }
+
+                else -> error("Skip engine pattern should consist of two parts: PLATFORM:ENGINE or ENGINE")
+            }
+
+            if (platform == null && engine == null) {
+                error("Skip engine pattern should consist of two parts: PLATFORM:ENGINE or ENGINE")
+            }
+            return SkipEnginePattern(platform, engine)
+        }
+    }
+}
+
+private class TestFailure(val engineName: String, val cause: Throwable)

ktor-client/ktor-client-tests/common/src/io/ktor/client/tests/utils/CommonClientTestUtils.kt

@@ -2,15 +2,13 @@
  * Copyright 2014-2024 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
  */
 
-@file:Suppress("NO_EXPLICIT_RETURN_TYPE_IN_API_MODE_WARNING", "KDocMissingDocumentation")
-
 package io.ktor.client.tests.utils
 
 import io.ktor.client.*
 import io.ktor.client.engine.*
 import io.ktor.utils.io.core.*
 import kotlinx.coroutines.*
-import kotlinx.coroutines.test.*
+import kotlinx.coroutines.test.runTest
 import kotlin.time.Duration.Companion.milliseconds
 
 /**
@@ -65,14 +63,23 @@ private fun testWithClient(
 /**
  * Perform test with selected client engine [factory].
  */
-@OptIn(DelicateCoroutinesApi::class)
 fun <T : HttpClientEngineConfig> testWithEngine(
     factory: HttpClientEngineFactory<T>,
     loader: ClientLoader? = null,
     timeoutMillis: Long = 60L * 1000L,
     retries: Int = 1,
     block: suspend TestClientBuilder<T>.() -> Unit
 ) = runTest(timeout = timeoutMillis.milliseconds) {
+    performTestWithEngine(factory, loader, retries, block)
+}
+
+@OptIn(DelicateCoroutinesApi::class)
+suspend fun <T : HttpClientEngineConfig> performTestWithEngine(
+    factory: HttpClientEngineFactory<T>,
+    loader: ClientLoader? = null,
+    retries: Int = 1,
+    block: suspend TestClientBuilder<T>.() -> Unit
+) {
     val builder = TestClientBuilder<T>().apply { block() }
 
     if (builder.dumpAfterDelay > 0 && loader != null) {

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/ContentTest.kt

@@ -18,8 +18,9 @@ import io.ktor.http.content.*
 import io.ktor.serialization.kotlinx.json.*
 import io.ktor.utils.io.*
 import io.ktor.utils.io.core.*
-import kotlinx.coroutines.*
-import kotlinx.io.*
+import kotlinx.coroutines.cancel
+import kotlinx.coroutines.withTimeoutOrNull
+import kotlinx.io.readByteArray
 import kotlin.test.*
 import kotlin.time.Duration.Companion.minutes
 
@@ -42,7 +43,7 @@ val testArrays = testSize.map {
     makeArray(it)
 }
 
-class ContentTest : ClientLoader(5 * 60) {
+class ContentTest : ClientLoader(timeout = 5.minutes) {
 
     @Test
     fun testGetFormData() = clientTests {

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/EventsTest.kt

@@ -74,7 +74,7 @@ class EventsTest : ClientLoader() {
     }
 
     @Test
-    fun testRedirectEvent() = clientTests(listOf("js")) {
+    fun testRedirectEvent() = clientTests(listOf("Js")) {
         test { client ->
             counter.value = 0
             client.monitor.subscribe(HttpResponseRedirectEvent) {

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/HttpRequestRetryTest.kt

@@ -9,10 +9,8 @@ import io.ktor.client.plugins.*
 import io.ktor.client.request.*
 import io.ktor.client.tests.utils.*
 import io.ktor.http.*
-import io.ktor.utils.io.errors.*
 import kotlinx.coroutines.*
 import kotlinx.io.IOException
-import kotlin.math.*
 import kotlin.test.*
 
 class HttpRequestRetryTest {

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/HttpTimeoutTest.kt

@@ -327,7 +327,7 @@ class HttpTimeoutTest : ClientLoader() {
     // Fix https://youtrack.jetbrains.com/issue/KTOR-7885
     @Ignore
     @Test
-    fun testRedirect() = clientTests(listOf("js"), retries = 5) {
+    fun testRedirect() = clientTests(listOf("Js"), retries = 5) {
         config {
             install(HttpTimeout) { requestTimeoutMillis = 10000 }
         }
@@ -344,7 +344,7 @@ class HttpTimeoutTest : ClientLoader() {
 
     // Js can't configure test timeout in browser
     @Test
-    fun testRedirectPerRequestAttributes() = clientTests(listOf("js")) {
+    fun testRedirectPerRequestAttributes() = clientTests(listOf("Js")) {
         config {
             install(HttpTimeout)
         }
@@ -429,7 +429,7 @@ class HttpTimeoutTest : ClientLoader() {
     }
 
     @Test
-    fun testConnectionRefusedException() = clientTests(listOf("Js", "native:*", "win:*")) {
+    fun testConnectionRefusedException() = clientTests(listOf("Js", "native:*", "jvm/win:*")) {
         config {
             install(HttpTimeout) { connectTimeoutMillis = 1000 }
         }

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/JsonTest.kt

@@ -21,7 +21,7 @@ class JsonTest : ClientLoader() {
     data class Result<T>(val message: String, val data: T)
 
     @Test
-    fun testUserGenerics() = clientTests(listOf("js")) {
+    fun testUserGenerics() = clientTests(listOf("Js")) {
         config {
             install(ContentNegotiation) { json() }
         }

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/MultiPartFormDataTest.kt

@@ -25,7 +25,7 @@ class MultiPartFormDataTest : ClientLoader() {
     }
 
     @Test
-    fun testMultiPartFormData() = clientTests(listOf("native")) {
+    fun testMultiPartFormData() = clientTests(listOf("native:*")) {
         test { client ->
             val result = client.preparePost("$TEST_SERVER/multipart") {
                 setBody(

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/WebSocketTest.kt

@@ -329,7 +329,7 @@ class WebSocketTest : ClientLoader() {
     @Ignore // TODO KTOR-7088
     @Test
     fun testImmediateReceiveAfterConnect() = clientTests(
-        ENGINES_WITHOUT_WS + "Darwin" + "js" // TODO KTOR-7088
+        ENGINES_WITHOUT_WS + "Darwin" + "Js" // TODO KTOR-7088
     ) {
         config {
             install(WebSockets)

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/plugins/CacheLegacyStorageTest.kt

@@ -569,7 +569,7 @@ class CacheLegacyStorageTest : ClientLoader() {
     }
 
     @Test
-    fun testPublicAndPrivateCache() = clientTests(listOf("native")) {
+    fun testPublicAndPrivateCache() = clientTests(listOf("native:*")) {
         val publicStorage = HttpCacheStorage.Unlimited()
         val privateStorage = HttpCacheStorage.Unlimited()
         config {

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/plugins/CookiesIntegrationTests.kt

@@ -106,7 +106,7 @@ class CookiesIntegrationTests : ClientLoader() {
     }
 
     @Test
-    fun testPath() = clientTests(listOf("js")) {
+    fun testPath() = clientTests(listOf("Js")) {
         config {
             install(HttpCookies)
         }

ktor-client/ktor-client-tests/common/test/io/ktor/client/tests/plugins/ServerSentEventsTest.kt

@@ -19,12 +19,15 @@ import io.ktor.test.dispatcher.*
 import io.ktor.utils.io.*
 import io.ktor.utils.io.charsets.*
 import kotlinx.coroutines.*
-import kotlinx.coroutines.flow.*
-import kotlin.coroutines.*
+import kotlinx.coroutines.flow.collectIndexed
+import kotlinx.coroutines.flow.single
+import kotlin.coroutines.CoroutineContext
+import kotlin.coroutines.resume
+import kotlin.coroutines.suspendCoroutine
 import kotlin.test.*
-import kotlin.test.assertFailsWith
+import kotlin.time.Duration.Companion.minutes
 
-class ServerSentEventsTest : ClientLoader(timeoutSeconds = 120) {
+class ServerSentEventsTest : ClientLoader(timeout = 2.minutes) {
 
     @Test
     fun testExceptionIfSseIsNotInstalled() = testSuspend {

ktor-client/ktor-client-tests/jsAndWasmShared/src/io/ktor/client/tests/utils/ClientLoader.jsAndWasmShared.kt

@@ -0,0 +1,12 @@
+// ktlint-disable filename
+/*
+ * Copyright 2014-2019 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package io.ktor.client.tests.utils
+
+import io.ktor.client.engine.*
+import io.ktor.client.engine.js.*
+
+internal actual val enginesToTest: Iterable<HttpClientEngineFactory<HttpClientEngineConfig>> get() = listOf(Js)
+internal actual val platformName: String get() = "web"