Enhance TestSuiteDNS

Author: q-litzler

src/commonMain/kotlin/com/algolia/search/client/ClientSearch.kt

@@ -35,7 +35,7 @@ import kotlinx.coroutines.*
  * Client to perform operations on indices.
  */
 public class ClientSearch private constructor(
-    private val transport: Transport
+    internal val transport: Transport
 ) :
     EndpointMultipleIndex by EndpointMultipleIndexImpl(transport),
     EndpointAPIKey by EndpointAPIKeyImpl(transport),

src/commonTest/kotlin/suite/TestSuiteDNS.kt

@@ -2,43 +2,62 @@ package suite
 
 import clientAdmin1
 import com.algolia.search.client.ClientSearch
-import com.algolia.search.configuration.CallType
 import com.algolia.search.configuration.ConfigurationSearch
-import com.algolia.search.configuration.RetryableHost
-import com.algolia.search.model.Time
+import com.algolia.search.model.IndexName
+import com.algolia.search.model.response.ResponseSearch
+import com.algolia.search.serialize.Json
+import io.ktor.client.engine.mock.MockEngine
+import io.ktor.client.engine.mock.respond
 import io.ktor.client.features.logging.LogLevel
+import io.ktor.http.ContentType
+import io.ktor.http.headersOf
+import kotlinx.coroutines.delay
+import kotlinx.coroutines.io.ByteReadChannel
 import runBlocking
+import shouldBeFalse
+import shouldBeTrue
+import shouldEqual
 import kotlin.test.Test
 
 
 internal class TestSuiteDNS {
 
+    private val readTimeout = 500L
+    private val shouldTimeout = 600L
+    private val shouldNotTimeout = 400L
+    private var requestCount = 0
+    private val mockEngine = MockEngine {
+        val delay = if (requestCount == 0) shouldTimeout else shouldNotTimeout
+
+        requestCount++
+        delay(delay)
+        content
+    }
+
+    private val content = respond(
+        headers = headersOf(
+            "Content-Type",
+            listOf(ContentType.Application.Json.toString())
+        ),
+        content = ByteReadChannel(Json.stringify(ResponseSearch.serializer(), ResponseSearch()))
+    )
+    private val client = ClientSearch(
+        ConfigurationSearch(
+            clientAdmin1.applicationID,
+            clientAdmin1.apiKey,
+            readTimeout = readTimeout,
+            engine = mockEngine,
+            logLevel = LogLevel.INFO
+        )
+    )
+
     @Test
     fun test() {
         runBlocking {
-            val applicationID = clientAdmin1.applicationID
-
-            val client = ClientSearch(
-                ConfigurationSearch(
-                    applicationID,
-                    clientAdmin1.apiKey,
-                    hosts = listOf(
-                        RetryableHost("algolia.biz", CallType.Read),
-                        RetryableHost("$applicationID-1.algolianet.com"),
-                        RetryableHost("$applicationID-2.algolianet.com"),
-                        RetryableHost("$applicationID-3.algolianet.com")
-                    ),
-                    logLevel = LogLevel.INFO
-                )
-            )
-            val timer = Time.getCurrentTimeMillis()
-
-            client.apply {
-                repeat(10) {
-                    listIndices()
-                }
-            }
-            println("Time elapsed in milliseconds: ${(Time.getCurrentTimeMillis() - timer)}")
+            val index = client.initIndex(IndexName("test"))
+
+            index.search()
+            client.transport.hosts.first().retryCount shouldEqual 1
         }
     }
 }