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

refactor(client): enhance HttpClientSseClientTransport with flexible customization API (#117)

Browse source 
- Add builder customizeClient() and customizeRequest() methods
- Enable HTTP client and request configuration through consumer-based customization
- Deprecate direct constructors in favor of the more flexible builder approach
- Add test coverage for customization capabilities

Co-authored-by: Christian Tzolov <christian.tzolov@broadcom.com>
Signed-off-by: Christian Tzolov <christian.tzolov@broadcom.com>
This commit is contained in:
a.darafeyeu 2025-04-07 14:30:04 +02:00 • committed by Christian Tzolov
commit fab434c088
1 changed files with 185 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

12
mcp-spring/mcp-spring-webmvc/src/test/java/io/modelcontextprotocol/server/WebMvcSseIntegrationTests.java
View file

@ -44,7 +44,7 @@ import static org.assertj.core.api.Assertions.assertThat;
import static org.awaitility.Awaitility.await;
import static org.mockito.Mockito.mock;
public class WebMvcSseIntegrationTests {
class WebMvcSseIntegrationTests {
private static final int PORT = 8183;
@ -79,13 +79,13 @@ public class WebMvcSseIntegrationTests {
try {
tomcatServer.tomcat().start();
assertThat(tomcatServer.tomcat().getServer().getState() == LifecycleState.STARTED);
assertThat(tomcatServer.tomcat().getServer().getState()).isEqualTo(LifecycleState.STARTED);
}
catch (Exception e) {
throw new RuntimeException("Failed to start Tomcat", e);
}
clientBuilder = McpClient.sync(new HttpClientSseClientTransport("http://localhost:" + PORT));
clientBuilder = McpClient.sync(HttpClientSseClientTransport.builder("http://localhost:" + PORT).build());
// Get the transport from Spring context
mcpServerTransportProvider = tomcatServer.appContext().getBean(WebMvcSseServerTransportProvider.class);
@ -200,8 +200,7 @@ public class WebMvcSseIntegrationTests {
CallToolResult response = mcpClient.callTool(new McpSchema.CallToolRequest("tool1", Map.of()));
assertThat(response).isNotNull();
assertThat(response).isEqualTo(callResponse);
assertThat(response).isNotNull().isEqualTo(callResponse);
mcpClient.close();
mcpServer.close();
@ -410,8 +409,7 @@ public class WebMvcSseIntegrationTests {
CallToolResult response = mcpClient.callTool(new McpSchema.CallToolRequest("tool1", Map.of()));
assertThat(response).isNotNull();
assertThat(response).isEqualTo(callResponse);
assertThat(response).isNotNull().isEqualTo(callResponse);
mcpClient.close();
mcpServer.close();