feat(tests): Add unit test for client connect function

Author: SeanChinJunKai

src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/client/Client.kt

@@ -102,7 +102,7 @@ public open class Client(
         } catch (error: Throwable) {
             close()
             if (error !is CancellationException) {
-                throw IllegalStateException("Error connecting to transport", error)
+                throw IllegalStateException("Error connecting to transport: ${error.message}")
             } else {
                 throw error
             }

src/jvmTest/kotlin/client/ClientTest.kt

@@ -6,6 +6,8 @@ import io.modelcontextprotocol.kotlin.sdk.CreateMessageResult
 import io.modelcontextprotocol.kotlin.sdk.EmptyJsonObject
 import io.modelcontextprotocol.kotlin.sdk.Implementation
 import InMemoryTransport
+import io.mockk.coEvery
+import io.mockk.spyk
 import io.modelcontextprotocol.kotlin.sdk.InitializeRequest
 import io.modelcontextprotocol.kotlin.sdk.InitializeResult
 import io.modelcontextprotocol.kotlin.sdk.JSONRPCMessage
@@ -189,6 +191,61 @@ class ClientTest {
         assertTrue(closed)
     }
 
+    @Test
+    fun `should reject due to non cancellation exception`() = runTest {
+        var closed = false
+        val clientTransport = object : AbstractTransport() {
+            override suspend fun start() {}
+
+            override suspend fun send(message: JSONRPCMessage) {
+                if (message !is JSONRPCRequest) return
+                check(message.method == Method.Defined.Initialize.value)
+
+                val result = InitializeResult(
+                    protocolVersion = LATEST_PROTOCOL_VERSION,
+                    capabilities = ServerCapabilities(),
+                    serverInfo = Implementation(
+                        name = "test",
+                        version = "1.0"
+                    )
+                )
+
+                val response = JSONRPCResponse(
+                    id = message.id,
+                    result = result
+                )
+
+                _onMessage.invoke(response)
+            }
+
+            override suspend fun close() {
+                closed = true
+            }
+        }
+
+        val mockClient = spyk(
+            Client(
+                clientInfo = Implementation(
+                    name = "test client",
+                    version = "1.0"
+                ),
+                options = ClientOptions()
+            )
+        )
+
+        coEvery{
+            mockClient.request<InitializeResult>(any())
+        } throws IllegalStateException("Test error")
+
+        val exception = assertFailsWith<IllegalStateException> {
+            mockClient.connect(clientTransport)
+        }
+
+        assertEquals("Error connecting to transport: Test error", exception.message)
+
+        assertTrue(closed)
+    }
+
     @Test
     fun `should respect server capabilities`() = runTest {
         val serverOptions = ServerOptions(