fixup! Introduce Kotlin integration tests

Signed-off-by: Sergey Karpov <sergey.karpov@jetbrains.com>

Author: skarpovdev

kotlin-sdk-test/src/jvmTest/kotlin/io/modelcontextprotocol/kotlin/sdk/integration/kotlin/PromptIntegrationTest.kt

@@ -100,7 +100,7 @@ class PromptIntegrationTest : KotlinTestBase() {
             if (includeImage) {
                 assistantContents.add(
                     ImageContent(
-                        data = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg==",
+                        data = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BmMIQAAAABJRU5ErkJggg==",
                         mimeType = "image/png",
                     ),
                 )
@@ -153,7 +153,8 @@ class PromptIntegrationTest : KotlinTestBase() {
                     PromptMessage(
                         role = Role.assistant,
                         content = TextContent(
-                            text = "As an AI, I don't have personal opinions, but I can provide information about $topic.",
+                            text = "As an AI, I don't have personal opinions," +
+                                " but I can provide information about $topic.",
                         ),
                     ),
                     PromptMessage(
@@ -193,8 +194,12 @@ class PromptIntegrationTest : KotlinTestBase() {
             ),
         ) { request ->
             val args = request.arguments ?: emptyMap()
-            val arg1 = args["requiredArg1"] ?: throw IllegalArgumentException("Missing required argument: requiredArg1")
-            val arg2 = args["requiredArg2"] ?: throw IllegalArgumentException("Missing required argument: requiredArg2")
+            val arg1 = args["requiredArg1"] ?: throw IllegalArgumentException(
+                "Missing required argument: requiredArg1",
+            )
+            val arg2 = args["requiredArg2"] ?: throw IllegalArgumentException(
+                "Missing required argument: requiredArg2",
+            )
             val optArg = args["optionalArg"] ?: "default"
 
             GetPromptResult(
@@ -222,14 +227,22 @@ class PromptIntegrationTest : KotlinTestBase() {
 
         val testPrompt = result.prompts.find { it.name == testPromptName }
         assertNotNull(testPrompt, "Test prompt should be in the list")
-        assertEquals(testPromptDescription, testPrompt.description, "Prompt description should match")
+        assertEquals(
+            testPromptDescription,
+            testPrompt.description,
+            "Prompt description should match",
+        )
 
         val arguments = testPrompt.arguments ?: error("Prompt arguments should not be null")
         assertTrue(arguments.isNotEmpty(), "Prompt arguments should not be empty")
 
         val nameArg = arguments.find { it.name == "name" }
         assertNotNull(nameArg, "Name argument should be in the list")
-        assertEquals("The name to greet", nameArg.description, "Argument description should match")
+        assertEquals(
+            "The name to greet",
+            nameArg.description,
+            "Argument description should match",
+        )
         assertEquals(true, nameArg.required, "Argument required flag should match")
     }
 
@@ -244,7 +257,11 @@ class PromptIntegrationTest : KotlinTestBase() {
         )
 
         assertNotNull(result, "Get prompt result should not be null")
-        assertEquals(testPromptDescription, result.description, "Prompt description should match")
+        assertEquals(
+            testPromptDescription,
+            result.description,
+            "Prompt description should match",
+        )
 
         assertTrue(result.messages.isNotEmpty(), "Prompt messages should not be empty")
         assertEquals(2, result.messages.size, "Prompt should have 2 messages")
@@ -254,7 +271,11 @@ class PromptIntegrationTest : KotlinTestBase() {
         val userContent = userMessage.content as? TextContent
         assertNotNull(userContent, "User message content should be TextContent")
         assertNotNull(userContent.text, "User message text should not be null")
-        assertEquals("Hello, $testName!", userContent.text, "User message content should match")
+        assertEquals(
+            "Hello, $testName!",
+            userContent.text,
+            "User message content should match",
+        )
 
         val assistantMessage = result.messages.find { it.role == Role.assistant }
         assertNotNull(assistantMessage, "Assistant message should be in the list")
@@ -277,7 +298,11 @@ class PromptIntegrationTest : KotlinTestBase() {
 
         val argsDef = strictPrompt.arguments ?: error("Prompt arguments should not be null")
         val requiredArgs = argsDef.filter { it.required == true }
-        assertEquals(2, requiredArgs.size, "Strict prompt should have 2 required arguments")
+        assertEquals(
+            2,
+            requiredArgs.size,
+            "Strict prompt should have 2 required arguments",
+        )
 
         // test missing required arg
         val exception = assertThrows<IllegalStateException> {
@@ -291,7 +316,11 @@ class PromptIntegrationTest : KotlinTestBase() {
             }
         }
 
-        assertEquals(exception.message?.contains("requiredArg2"), true, "Exception should mention the missing argument")
+        assertEquals(
+            true,
+            exception.message?.contains("requiredArg2"),
+            "Exception should mention the missing argument",
+        )
 
         // test with no args
         val exception2 = assertThrows<IllegalStateException> {
@@ -348,7 +377,11 @@ class PromptIntegrationTest : KotlinTestBase() {
         )
 
         assertNotNull(result, "Get prompt result should not be null")
-        assertEquals(complexPromptDescription, result.description, "Prompt description should match")
+        assertEquals(
+            complexPromptDescription,
+            result.description,
+            "Prompt description should match",
+        )
 
         assertTrue(result.messages.isNotEmpty(), "Prompt messages should not be empty")
         assertEquals(2, result.messages.size, "Prompt should have 2 messages")
@@ -365,7 +398,10 @@ class PromptIntegrationTest : KotlinTestBase() {
         val assistantContent = assistantMessage.content as? TextContent
         assertNotNull(assistantContent, "Assistant message content should be TextContent")
         val assistantText = requireNotNull(assistantContent.text)
-        assertTrue(assistantText.contains(topic), "Assistant message should contain the topic")
+        assertTrue(
+            assistantText.contains(topic),
+            "Assistant message should contain the topic",
+        )
 
         val resultNoImage = client.getPrompt(
             GetPromptRequest(
@@ -392,7 +428,11 @@ class PromptIntegrationTest : KotlinTestBase() {
         )
 
         assertNotNull(result, "Get prompt result should not be null")
-        assertEquals(conversationPromptDescription, result.description, "Prompt description should match")
+        assertEquals(
+            conversationPromptDescription,
+            result.description,
+            "Prompt description should match",
+        )
 
         assertTrue(result.messages.isNotEmpty(), "Prompt messages should not be empty")
         assertEquals(6, result.messages.size, "Prompt should have 6 messages")
@@ -405,7 +445,11 @@ class PromptIntegrationTest : KotlinTestBase() {
 
         for (i in 0 until result.messages.size) {
             val expectedRole = if (i % 2 == 0) Role.user else Role.assistant
-            assertEquals(expectedRole, result.messages[i].role, "Message $i should have role $expectedRole")
+            assertEquals(
+                expectedRole,
+                result.messages[i].role,
+                "Message $i should have role $expectedRole",
+            )
         }
 
         for (message in result.messages) {

kotlin-sdk-test/src/jvmTest/kotlin/io/modelcontextprotocol/kotlin/sdk/integration/kotlin/ToolIntegrationTest.kt

@@ -166,7 +166,9 @@ class ToolIntegrationTest : KotlinTestBase() {
             val formattedResult = "%.${precision}f".format(result)
 
             val textContent = if (showSteps) {
-                "Operation: $operation\nA: $a\nB: $b\nResult: $formattedResult\nTags: ${tags.joinToString(", ")}"
+                "Operation: $operation\nA: $a\nB: $b\nResult: $formattedResult\nTags: ${
+                    tags.joinToString(", ")
+                }"
             } else {
                 "Result: $formattedResult"
             }
@@ -279,7 +281,7 @@ class ToolIntegrationTest : KotlinTestBase() {
             if (includeImage) {
                 content.add(
                     ImageContent(
-                        data = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg==",
+                        data = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQDwAEhQGAhKmMIQAAAABJRU5ErkJggg==",
                         mimeType = "image/png",
                     ),
                 )
@@ -304,7 +306,11 @@ class ToolIntegrationTest : KotlinTestBase() {
 
         val testTool = result.tools.find { it.name == testToolName }
         assertNotNull(testTool, "Test tool should be in the list")
-        assertEquals(testToolDescription, testTool.description, "Tool description should match")
+        assertEquals(
+            testToolDescription,
+            testTool.description,
+            "Tool description should match",
+        )
     }
 
     @Test
@@ -347,7 +353,10 @@ class ToolIntegrationTest : KotlinTestBase() {
             val contentText = requireNotNull(content.text)
 
             assertTrue(contentText.contains("Operation"), "Result should contain operation")
-            assertTrue(contentText.contains("multiply"), "Result should contain multiply operation")
+            assertTrue(
+                contentText.contains("multiply"),
+                "Result should contain multiply operation",
+            )
             assertTrue(contentText.contains("5.5"), "Result should contain first operand")
             assertTrue(contentText.contains("2.0"), "Result should contain second operand")
             assertTrue(contentText.contains("11"), "Result should contain result value")
@@ -421,12 +430,20 @@ class ToolIntegrationTest : KotlinTestBase() {
         val result = client.callTool(multiContentToolName, arguments)
 
         val toolResult = assertCallToolResult(result)
-        assertEquals(2, toolResult.content.size, "Tool result should have 2 content items")
+        assertEquals(
+            2,
+            toolResult.content.size,
+            "Tool result should have 2 content items",
+        )
 
         val textContent = toolResult.content.firstOrNull { it is TextContent } as? TextContent
         assertNotNull(textContent, "Result should contain TextContent")
         assertNotNull(textContent.text, "Text content should not be null")
-        assertEquals("Text content: $testText", textContent.text, "Text content should match")
+        assertEquals(
+            "Text content: $testText",
+            textContent.text,
+            "Text content should match",
+        )
 
         val imageContent = toolResult.content.firstOrNull { it is ImageContent } as? ImageContent
         assertNotNull(imageContent, "Result should contain ImageContent")
@@ -445,7 +462,11 @@ class ToolIntegrationTest : KotlinTestBase() {
         val textOnlyResult = client.callTool(multiContentToolName, textOnlyArgs)
 
         val textOnlyToolResult = assertCallToolResult(textOnlyResult, "Text-only: ")
-        assertEquals(1, textOnlyToolResult.content.size, "Text-only result should have 1 content item")
+        assertEquals(
+            1,
+            textOnlyToolResult.content.size,
+            "Text-only result should have 1 content item",
+        )
 
         assertTextContent(textOnlyToolResult.content.firstOrNull(), "Text content: $testText")
     }