From be1e91c74fb3f80f6971d14ff187dcd994047a86 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Wed, 12 Mar 2025 10:08:41 +0100 Subject: [PATCH] Add condition tests for MCP client transport auto-configurations - Add SseHttpClientTransportAutoConfigurationTests to verify conditional bean creation for HTTP client transports - Add SseWebFluxTransportAutoConfigurationTests to verify conditional bean creation for WebFlux client transports - Test both auto-configurations with various conditions including presence/absence of WebFluxSseClientTransport and MCP client enabled/disabled states Signed-off-by: Christian Tzolov --- ...ClientTransportAutoConfigurationTests.java | 63 +++++++++++++++++++ ...ebFluxTransportAutoConfigurationTests.java | 59 +++++++++++++++++ 2 files changed, 122 insertions(+) create mode 100644 auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseHttpClientTransportAutoConfigurationTests.java create mode 100644 auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseWebFluxTransportAutoConfigurationTests.java diff --git a/auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseHttpClientTransportAutoConfigurationTests.java b/auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseHttpClientTransportAutoConfigurationTests.java new file mode 100644 index 00000000000..aa50c553b63 --- /dev/null +++ b/auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseHttpClientTransportAutoConfigurationTests.java @@ -0,0 +1,63 @@ +/* + * Copyright 2025-2025 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.ai.autoconfigure.mcp.client; + +import org.junit.jupiter.api.Test; + +import org.springframework.boot.autoconfigure.AutoConfigurations; +import org.springframework.boot.test.context.FilteredClassLoader; +import org.springframework.boot.test.context.runner.ApplicationContextRunner; + +import static org.assertj.core.api.Assertions.assertThat; + +public class SseHttpClientTransportAutoConfigurationTests { + + private final ApplicationContextRunner applicationContext = new ApplicationContextRunner() + .withConfiguration(AutoConfigurations.of(SseHttpClientTransportAutoConfiguration.class)); + + @Test + void mcpHttpClientTransportsNotPresentIfMissingWebFluxSseClientTransportPresent() { + + this.applicationContext.run((context) -> { + assertThat(context.containsBean("mcpHttpClientTransports")).isFalse(); + }); + } + + @Test + void mcpHttpClientTransportsPresentIfMissingWebFluxSseClientTransportNotPresent() { + + this.applicationContext + .withClassLoader( + new FilteredClassLoader("io.modelcontextprotocol.client.transport.WebFluxSseClientTransport")) + .run((context) -> { + assertThat(context.containsBean("mcpHttpClientTransports")).isTrue(); + }); + } + + @Test + void mcpHttpClientTransportsNotPresentIfMcpClientDisabled() { + + this.applicationContext + .withClassLoader( + new FilteredClassLoader("io.modelcontextprotocol.client.transport.WebFluxSseClientTransport")) + .withPropertyValues("spring.ai.mcp.client.enabled", "false") + .run((context) -> { + assertThat(context.containsBean("mcpHttpClientTransports")).isFalse(); + }); + } + +} diff --git a/auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseWebFluxTransportAutoConfigurationTests.java b/auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseWebFluxTransportAutoConfigurationTests.java new file mode 100644 index 00000000000..78394b194e0 --- /dev/null +++ b/auto-configurations/spring-ai-mcp-client/src/test/java/org/springframework/ai/autoconfigure/mcp/client/SseWebFluxTransportAutoConfigurationTests.java @@ -0,0 +1,59 @@ +/* + * Copyright 2025-2025 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.ai.autoconfigure.mcp.client; + +import org.junit.jupiter.api.Test; + +import org.springframework.boot.autoconfigure.AutoConfigurations; +import org.springframework.boot.test.context.FilteredClassLoader; +import org.springframework.boot.test.context.runner.ApplicationContextRunner; + +import static org.assertj.core.api.Assertions.assertThat; + +public class SseWebFluxTransportAutoConfigurationTests { + + private final ApplicationContextRunner applicationContext = new ApplicationContextRunner() + .withConfiguration(AutoConfigurations.of(SseWebFluxTransportAutoConfiguration.class)); + + @Test + void webFluxClientTransportsPresentIfWebFluxSseClientTransportPresent() { + + this.applicationContext.run((context) -> { + assertThat(context.containsBean("webFluxClientTransports")).isTrue(); + }); + } + + @Test + void webFluxClientTransportsNotPresentIfMissingWebFluxSseClientTransportNotPresent() { + + this.applicationContext + .withClassLoader( + new FilteredClassLoader("io.modelcontextprotocol.client.transport.WebFluxSseClientTransport")) + .run((context) -> { + assertThat(context.containsBean("webFluxClientTransports")).isFalse(); + }); + } + + @Test + void webFluxClientTransportsNotPresentIfMcpClientDisabled() { + + this.applicationContext.withPropertyValues("spring.ai.mcp.client.enabled", "false").run((context) -> { + assertThat(context.containsBean("webFluxClientTransports")).isFalse(); + }); + } + +}