modelcontextprotocol/java-sdk
Watch
1
0
Fork
You've already forked java-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

code cleaning

Browse source
This commit is contained in:
Christian Tzolov 2024-12-10 11:46:58 +01:00
commit 1b6d661470
3 changed files with 191 additions and 100 deletions
Download patch file Download diff file Expand all files Collapse all files

2
mcp/src/test/java/spring/ai/experimental/mcp/client/McpAsyncClientTests.java
View file

@ -23,8 +23,6 @@ import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Timeout;
import spring.ai.experimental.mcp.client.McpAsyncClient;
import spring.ai.experimental.mcp.client.McpClient;
import spring.ai.experimental.mcp.client.stdio.ServerParameters;
import spring.ai.experimental.mcp.client.stdio.StdioServerTransport;
import spring.ai.experimental.mcp.spec.McpSchema.CallToolRequest;

11
mcp/src/test/java/spring/ai/experimental/mcp/client/McpSyncClientTests.java
View file

@ -16,29 +16,26 @@
package spring.ai.experimental.mcp.client;
import java.time.Duration;
import java.util.List;
import java.util.Map;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Timeout;
import spring.ai.experimental.mcp.client.McpClient;
import spring.ai.experimental.mcp.client.McpSyncClient;
import spring.ai.experimental.mcp.client.stdio.ServerParameters;
import spring.ai.experimental.mcp.client.stdio.StdioServerTransport;
import spring.ai.experimental.mcp.spec.McpSchema.CallToolRequest;
import spring.ai.experimental.mcp.spec.McpSchema.CallToolResult;
import spring.ai.experimental.mcp.spec.McpSchema.Content;
import spring.ai.experimental.mcp.spec.McpSchema.ListResourcesResult;
import spring.ai.experimental.mcp.spec.McpSchema.ListToolsResult;
import spring.ai.experimental.mcp.spec.McpSchema.Resource;
import spring.ai.experimental.mcp.spec.McpSchema.TextContent;
import spring.ai.experimental.mcp.spec.McpSchema.Tool;
import static org.assertj.core.api.Assertions.*;
import com.fasterxml.jackson.databind.ObjectMapper;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatCode;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
/**
* Unit tests for MCP Client Session functionality.

4
mcp/src/test/java/spring/ai/experimental/mcp/spec/DefaultMcpTransportTests.java
View file

@ -107,7 +107,7 @@ class DefaultMcpTransportTests {
@Test
void customErrorHandlerShouldReceiveErrors() {
AtomicReference<String> receivedError = new AtomicReference<>();
transport.setErrorHandler(receivedError::set);
transport.setInboundErrorHandler(receivedError::set);
transport.start();
String errorMessage = "Test error";
@ -124,7 +124,7 @@ class DefaultMcpTransportTests {
AtomicReference<String> receivedError = new AtomicReference<>();
transport.setInboudMessageHandler(receivedMessage::set);
transport.setErrorHandler(receivedError::set);
transport.setInboundErrorHandler(receivedError::set);
transport.start();
JSONRPCRequest message = new JSONRPCRequest(