Merge branch '1.5.x'

This commit is contained in:
Stephane Nicoll
2016-12-22 10:33:39 +01:00
3 changed files with 46 additions and 21 deletions

View File

@@ -64,12 +64,21 @@ public class CassandraAutoConfigurationTests {
@Test
public void createCustomizeCluster() {
load(ClusterConfig.class);
load(MockCustomizerConfig.class);
assertThat(this.context.getBeanNamesForType(Cluster.class).length).isEqualTo(1);
assertThat(this.context.getBeanNamesForType(ClusterCustomizer.class).length)
assertThat(this.context.getBeanNamesForType(ClusterBuilderCustomizer.class).length)
.isEqualTo(1);
}
@Test
public void customizerOverridesAutoConfig() {
load(SimpleCustomizerConfig.class,
"spring.data.cassandra.cluster-name=testcluster");
assertThat(this.context.getBeanNamesForType(Cluster.class).length).isEqualTo(1);
Cluster cluster = this.context.getBean(Cluster.class);
assertThat(cluster.getClusterName()).isEqualTo("overridden-name");
}
private void load(String... environment) {
load(null, environment);
}
@@ -87,11 +96,26 @@ public class CassandraAutoConfigurationTests {
}
@Configuration
static class ClusterConfig {
static class MockCustomizerConfig {
@Bean
public ClusterCustomizer customizer() {
return mock(ClusterCustomizer.class);
public ClusterBuilderCustomizer customizer() {
return mock(ClusterBuilderCustomizer.class);
}
}
@Configuration
static class SimpleCustomizerConfig {
@Bean
public ClusterBuilderCustomizer customizer() {
return new ClusterBuilderCustomizer() {
@Override
public void customize(Cluster.Builder clusterBuilder) {
clusterBuilder.withClusterName("overridden-name");
}
};
}
}