a49b9f5914
- Add copyright headers to files missing them - Add blank lines after copyright headers for consistency - Remove unused imports in test files - Remove unnecessary keep-alive warning logs in WebFlux and WebMvc transport providers Signed-off-by: Christian Tzolov <christian.tzolov@broadcom.com>
45 lines
1.4 KiB
Java
45 lines
1.4 KiB
Java
/*
|
|
* Copyright 2025-2025 the original author or authors.
|
|
*/
|
|
|
|
package io.modelcontextprotocol.client;
|
|
|
|
import org.junit.jupiter.api.Timeout;
|
|
import org.springframework.web.reactive.function.client.WebClient;
|
|
import org.testcontainers.containers.GenericContainer;
|
|
import org.testcontainers.containers.wait.strategy.Wait;
|
|
|
|
import io.modelcontextprotocol.client.transport.WebClientStreamableHttpTransport;
|
|
import io.modelcontextprotocol.spec.McpClientTransport;
|
|
|
|
@Timeout(15)
|
|
public class WebClientStreamableHttpSyncClientTests extends AbstractMcpSyncClientTests {
|
|
|
|
static String host = "http://localhost:3001";
|
|
|
|
// Uses the https://github.com/tzolov/mcp-everything-server-docker-image
|
|
@SuppressWarnings("resource")
|
|
GenericContainer<?> container = new GenericContainer<>("docker.io/tzolov/mcp-everything-server:v2")
|
|
.withCommand("node dist/index.js streamableHttp")
|
|
.withLogConsumer(outputFrame -> System.out.println(outputFrame.getUtf8String()))
|
|
.withExposedPorts(3001)
|
|
.waitingFor(Wait.forHttp("/").forStatusCode(404));
|
|
|
|
@Override
|
|
protected McpClientTransport createMcpTransport() {
|
|
return WebClientStreamableHttpTransport.builder(WebClient.builder().baseUrl(host)).build();
|
|
}
|
|
|
|
@Override
|
|
protected void onStart() {
|
|
container.start();
|
|
int port = container.getMappedPort(3001);
|
|
host = "http://" + container.getHost() + ":" + port;
|
|
}
|
|
|
|
@Override
|
|
public void onClose() {
|
|
container.stop();
|
|
}
|
|
|
|
}
|