mirror of
https://github.com/spring-projects/spring-boot.git
synced 2024-08-29 03:06:45 +08:00
Broaden testing of ReactiveElasticsearchRestClientAutoConfiguration
Closes gh-27919
This commit is contained in:
parent
08e79ee472
commit
206176425a
@ -16,6 +16,11 @@
|
|||||||
|
|
||||||
package org.springframework.boot.autoconfigure.data.elasticsearch;
|
package org.springframework.boot.autoconfigure.data.elasticsearch;
|
||||||
|
|
||||||
|
import java.net.InetSocketAddress;
|
||||||
|
import java.time.Duration;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import org.assertj.core.api.InstanceOfAssertFactories;
|
||||||
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||||
@ -24,6 +29,9 @@ import org.springframework.context.annotation.Bean;
|
|||||||
import org.springframework.context.annotation.Configuration;
|
import org.springframework.context.annotation.Configuration;
|
||||||
import org.springframework.data.elasticsearch.client.ClientConfiguration;
|
import org.springframework.data.elasticsearch.client.ClientConfiguration;
|
||||||
import org.springframework.data.elasticsearch.client.reactive.ReactiveElasticsearchClient;
|
import org.springframework.data.elasticsearch.client.reactive.ReactiveElasticsearchClient;
|
||||||
|
import org.springframework.http.HttpHeaders;
|
||||||
|
import org.springframework.http.codec.CodecConfigurer.DefaultCodecConfig;
|
||||||
|
import org.springframework.web.reactive.function.client.WebClient;
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.mock;
|
||||||
@ -57,6 +65,86 @@ class ReactiveElasticsearchRestClientAutoConfigurationTests {
|
|||||||
.hasSingleBean(ClientConfiguration.class).hasBean("customClientConfiguration"));
|
.hasSingleBean(ClientConfiguration.class).hasBean("customClientConfiguration"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenEndpointIsCustomizedThenClientConfigurationHasCustomEndpoint() {
|
||||||
|
this.contextRunner.withPropertyValues("spring.data.elasticsearch.client.reactive.endpoints=localhost:9876")
|
||||||
|
.run((context) -> {
|
||||||
|
List<InetSocketAddress> endpoints = context.getBean(ClientConfiguration.class).getEndpoints();
|
||||||
|
assertThat(endpoints).hasSize(1);
|
||||||
|
assertThat(endpoints.get(0).getHostString()).isEqualTo("localhost");
|
||||||
|
assertThat(endpoints.get(0).getPort()).isEqualTo(9876);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenMultipleEndpointsAreConfiguredThenClientConfigurationHasMultipleEndpoints() {
|
||||||
|
this.contextRunner
|
||||||
|
.withPropertyValues("spring.data.elasticsearch.client.reactive.endpoints=localhost:9876,localhost:8765")
|
||||||
|
.run((context) -> {
|
||||||
|
List<InetSocketAddress> endpoints = context.getBean(ClientConfiguration.class).getEndpoints();
|
||||||
|
assertThat(endpoints).hasSize(2);
|
||||||
|
assertThat(endpoints.get(0).getHostString()).isEqualTo("localhost");
|
||||||
|
assertThat(endpoints.get(0).getPort()).isEqualTo(9876);
|
||||||
|
assertThat(endpoints.get(1).getHostString()).isEqualTo("localhost");
|
||||||
|
assertThat(endpoints.get(1).getPort()).isEqualTo(8765);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenConfiguredToUseSslThenClientConfigurationUsesSsl() {
|
||||||
|
this.contextRunner.withPropertyValues("spring.data.elasticsearch.client.reactive.use-ssl=true")
|
||||||
|
.run((context) -> assertThat(context.getBean(ClientConfiguration.class).useSsl()).isTrue());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenSocketTimeoutIsNotConfiguredThenClientConfigurationUsesDefault() {
|
||||||
|
this.contextRunner.run((context) -> assertThat(context.getBean(ClientConfiguration.class).getSocketTimeout())
|
||||||
|
.isEqualTo(Duration.ofSeconds(5)));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenConnectionTimeoutIsNotConfiguredThenClientConfigurationUsesDefault() {
|
||||||
|
this.contextRunner.run((context) -> assertThat(context.getBean(ClientConfiguration.class).getConnectTimeout())
|
||||||
|
.isEqualTo(Duration.ofSeconds(10)));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenSocketTimeoutIsConfiguredThenClientConfigurationHasCustomSocketTimeout() {
|
||||||
|
this.contextRunner.withPropertyValues("spring.data.elasticsearch.client.reactive.socket-timeout=2s")
|
||||||
|
.run((context) -> assertThat(context.getBean(ClientConfiguration.class).getSocketTimeout())
|
||||||
|
.isEqualTo(Duration.ofSeconds(2)));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenConnectionTimeoutIsConfiguredThenClientConfigurationHasCustomConnectTimeout() {
|
||||||
|
this.contextRunner.withPropertyValues("spring.data.elasticsearch.client.reactive.connection-timeout=2s")
|
||||||
|
.run((context) -> assertThat(context.getBean(ClientConfiguration.class).getConnectTimeout())
|
||||||
|
.isEqualTo(Duration.ofSeconds(2)));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenCredentialsAreConfiguredThenClientConfigurationHasDefaultAuthorizationHeader() {
|
||||||
|
this.contextRunner
|
||||||
|
.withPropertyValues("spring.data.elasticsearch.client.reactive.username=alice",
|
||||||
|
"spring.data.elasticsearch.client.reactive.password=secret")
|
||||||
|
.run((context) -> assertThat(
|
||||||
|
context.getBean(ClientConfiguration.class).getDefaultHeaders().get(HttpHeaders.AUTHORIZATION))
|
||||||
|
.containsExactly("Basic YWxpY2U6c2VjcmV0"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void whenMaxInMemorySizeIsConfiguredThenUnderlyingWebClientHasCustomMaxInMemorySize() {
|
||||||
|
this.contextRunner.withPropertyValues("spring.data.elasticsearch.client.reactive.max-in-memory-size=1MB")
|
||||||
|
.run((context) -> {
|
||||||
|
WebClient client = context.getBean(ClientConfiguration.class).getWebClientConfigurer()
|
||||||
|
.apply(WebClient.create());
|
||||||
|
assertThat(client).extracting("exchangeFunction").extracting("strategies")
|
||||||
|
.extracting("codecConfigurer").extracting("defaultCodecs")
|
||||||
|
.asInstanceOf(InstanceOfAssertFactories.type(DefaultCodecConfig.class))
|
||||||
|
.extracting(DefaultCodecConfig::maxInMemorySize).isEqualTo(1024 * 1024);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
@Configuration(proxyBeanMethods = false)
|
@Configuration(proxyBeanMethods = false)
|
||||||
static class CustomClientConfiguration {
|
static class CustomClientConfiguration {
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user