Fix tests

Author: tiginamaria

kotlin-sdk-server/src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/server/ServerSession.kt

@@ -210,12 +210,14 @@ public open class ServerSession(private val serverInfo: Implementation, options:
 
             Defined.RootsList -> {
                 if (clientCapabilities?.roots == null) {
+                    logger.error { "Client capability assertion failed: listing roots not supported" }
                     throw IllegalStateException("Client does not support listing roots (required for ${method.value})")
                 }
             }
 
             Defined.ElicitationCreate -> {
                 if (clientCapabilities?.elicitation == null) {
+                    logger.error { "Client capability assertion failed: elicitation not supported" }
                     throw IllegalStateException("Client does not support elicitation (required for ${method.value})")
                 }
             }
@@ -225,7 +227,7 @@ public open class ServerSession(private val serverInfo: Implementation, options:
             }
 
             else -> {
-                throw IllegalStateException("Server does not support $method")
+                // For notifications not specifically listed, no assertion by default
             }
         }
     }
@@ -257,7 +259,7 @@ public open class ServerSession(private val serverInfo: Implementation, options:
                 }
             }
 
-            Defined.NotificationsResourcesListChanged -> {
+            Defined.NotificationsToolsListChanged -> {
                 if (serverCapabilities.tools == null) {
                     throw IllegalStateException(
                         "Server does not support notifying of tool list changes (required for ${method.value})",
@@ -280,7 +282,7 @@ public open class ServerSession(private val serverInfo: Implementation, options:
             }
 
             else -> {
-                throw IllegalStateException("Server does not support $method or it's not a notification method")
+                // For notifications not specifically listed, no assertion by default
             }
         }
     }
@@ -340,7 +342,7 @@ public open class ServerSession(private val serverInfo: Implementation, options:
             }
 
             else -> {
-                throw IllegalStateException("Server does not support $method or it's not a request handler method")
+                // For notifications not specifically listed, no assertion by default
             }
         }
     }

kotlin-sdk-test/src/commonTest/kotlin/io/modelcontextprotocol/kotlin/sdk/integration/SseIntegrationTest.kt

@@ -31,9 +31,6 @@ import io.ktor.client.engine.cio.CIO as ClientCIO
 import io.ktor.server.cio.CIO as ServerCIO
 import io.ktor.server.sse.SSE as ServerSSE
 
-private const val URL = "127.0.0.1"
-private const val PORT = 0
-
 class SseIntegrationTest {
     @Test
     fun `client should be able to connect to sse server`() = runTest {
@@ -44,7 +41,8 @@ class SseIntegrationTest {
             withContext(Dispatchers.Default) {
                 withTimeout(1000) {
                     server = initServer()
-                    client = initClient()
+                    val port = server.engine.resolvedConnectors().first().port
+                    client = initClient(serverPort=port)
                 }
             }
         } finally {
@@ -64,12 +62,12 @@ class SseIntegrationTest {
     fun `single sse connection`() = runTest {
         var server: EmbeddedServer<CIOApplicationEngine, CIOApplicationEngine.Configuration>? = null
         var client: Client? = null
-
         try {
             withContext(Dispatchers.Default) {
                 withTimeout(1000) {
                     server = initServer()
-                    client = initClient("Client A")
+                    val port = server.engine.resolvedConnectors().first().port
+                    client = initClient("Client A", port)
 
                     val promptA = getPrompt(client, "Client A")
                     assertTrue { "Client A" in promptA }
@@ -99,8 +97,10 @@ class SseIntegrationTest {
             withContext(Dispatchers.Default) {
                 withTimeout(1000) {
                     server = initServer()
-                    clientA = initClient("Client A")
-                    clientB = initClient("Client B")
+                    val port = server.engine.resolvedConnectors().first().port
+
+                    clientA = initClient("Client A", port)
+                    clientB = initClient("Client B", port)
 
                     // Step 3: Send a prompt request from Client A
                     val promptA = getPrompt(clientA, "Client A")
@@ -118,7 +118,7 @@ class SseIntegrationTest {
         }
     }
 
-    private suspend fun initClient(name: String = ""): Client {
+    private suspend fun initClient(name: String = "", serverPort: Int): Client {
         val client = Client(
             Implementation(name = name, version = "1.0.0"),
         )
@@ -131,7 +131,7 @@ class SseIntegrationTest {
         val transport = httpClient.mcpSseTransport {
             url {
                 host = URL
-                port = PORT
+                port = serverPort
             }
         }
 
@@ -195,7 +195,12 @@ class SseIntegrationTest {
             ),
         )
 
-        return (response?.messages?.first()?.content as? TextContent)?.text
+        return (response.messages.first().content as? TextContent)?.text
             ?: error("Failed to receive prompt for Client $clientName")
     }
+
+    companion object {
+        private const val URL = "127.0.0.1"
+        private const val PORT = 0
+    }
 }