Commit a869d25d authored by Phillip Webb's avatar Phillip Webb

Polish

parent eac4c7e8
...@@ -59,7 +59,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -59,7 +59,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests {
@Rule @Rule
public final ExpectedException thrown = ExpectedException.none(); public final ExpectedException thrown = ExpectedException.none();
private final MBeanServer mBeanServer = ManagementFactory.getPlatformMBeanServer(); private final MBeanServer server = ManagementFactory.getPlatformMBeanServer();
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner() private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
.withConfiguration(AutoConfigurations.of(JmxAutoConfiguration.class, .withConfiguration(AutoConfigurations.of(JmxAutoConfiguration.class,
...@@ -70,7 +70,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -70,7 +70,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests {
throws MalformedObjectNameException, InstanceNotFoundException { throws MalformedObjectNameException, InstanceNotFoundException {
this.contextRunner.run((context) -> { this.contextRunner.run((context) -> {
this.thrown.expect(InstanceNotFoundException.class); this.thrown.expect(InstanceNotFoundException.class);
this.mBeanServer.getObjectInstance(createDefaultObjectName()); this.server.getObjectInstance(createDefaultObjectName());
}); });
} }
...@@ -78,7 +78,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -78,7 +78,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests {
public void registeredWithProperty() throws Exception { public void registeredWithProperty() throws Exception {
this.contextRunner.withPropertyValues(ENABLE_ADMIN_PROP).run((context) -> { this.contextRunner.withPropertyValues(ENABLE_ADMIN_PROP).run((context) -> {
ObjectName objectName = createDefaultObjectName(); ObjectName objectName = createDefaultObjectName();
ObjectInstance objectInstance = this.mBeanServer.getObjectInstance(objectName); ObjectInstance objectInstance = this.server.getObjectInstance(objectName);
assertThat(objectInstance).as("Lifecycle bean should have been registered") assertThat(objectInstance).as("Lifecycle bean should have been registered")
.isNotNull(); .isNotNull();
}); });
...@@ -87,18 +87,20 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -87,18 +87,20 @@ public class SpringApplicationAdminJmxAutoConfigurationTests {
@Test @Test
public void registerWithCustomJmxName() throws InstanceNotFoundException { public void registerWithCustomJmxName() throws InstanceNotFoundException {
String customJmxName = "org.acme:name=FooBar"; String customJmxName = "org.acme:name=FooBar";
this.contextRunner.withSystemProperties( this.contextRunner
"spring.application.admin.jmx-name=" + customJmxName) .withSystemProperties(
"spring.application.admin.jmx-name=" + customJmxName)
.withPropertyValues(ENABLE_ADMIN_PROP).run((context) -> { .withPropertyValues(ENABLE_ADMIN_PROP).run((context) -> {
try { try {
this.mBeanServer.getObjectInstance(createObjectName(customJmxName)); this.server.getObjectInstance(createObjectName(customJmxName));
} }
catch (InstanceNotFoundException ex) { catch (InstanceNotFoundException ex) {
fail("Admin MBean should have been exposed with custom name"); fail("Admin MBean should have been exposed with custom name");
} }
this.thrown.expect(InstanceNotFoundException.class); // Should not be exposed this.thrown.expect(InstanceNotFoundException.class); // Should not be
this.mBeanServer.getObjectInstance(createDefaultObjectName()); // exposed
}); this.server.getObjectInstance(createDefaultObjectName());
});
} }
@Test @Test
...@@ -110,7 +112,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -110,7 +112,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests {
SpringApplicationAdminJmxAutoConfiguration.class) SpringApplicationAdminJmxAutoConfiguration.class)
.run("--" + ENABLE_ADMIN_PROP, "--server.port=0")) { .run("--" + ENABLE_ADMIN_PROP, "--server.port=0")) {
assertThat(context).isInstanceOf(ServletWebServerApplicationContext.class); assertThat(context).isInstanceOf(ServletWebServerApplicationContext.class);
assertThat(this.mBeanServer.getAttribute(createDefaultObjectName(), assertThat(this.server.getAttribute(createDefaultObjectName(),
"EmbeddedWebApplication")).isEqualTo(Boolean.TRUE); "EmbeddedWebApplication")).isEqualTo(Boolean.TRUE);
int expected = ((ServletWebServerApplicationContext) context).getWebServer() int expected = ((ServletWebServerApplicationContext) context).getWebServer()
.getPort(); .getPort();
...@@ -128,11 +130,10 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -128,11 +130,10 @@ public class SpringApplicationAdminJmxAutoConfigurationTests {
.child(JmxAutoConfiguration.class, .child(JmxAutoConfiguration.class,
SpringApplicationAdminJmxAutoConfiguration.class) SpringApplicationAdminJmxAutoConfiguration.class)
.web(WebApplicationType.NONE); .web(WebApplicationType.NONE);
try (ConfigurableApplicationContext parent = parentBuilder try (ConfigurableApplicationContext parent = parentBuilder
.run("--" + ENABLE_ADMIN_PROP); .run("--" + ENABLE_ADMIN_PROP);
ConfigurableApplicationContext child = childBuilder ConfigurableApplicationContext child = childBuilder
.run("--" + ENABLE_ADMIN_PROP)) { .run("--" + ENABLE_ADMIN_PROP)) {
BeanFactoryUtils.beanOfType(parent.getBeanFactory(), BeanFactoryUtils.beanOfType(parent.getBeanFactory(),
SpringApplicationAdminMXBeanRegistrar.class); SpringApplicationAdminMXBeanRegistrar.class);
this.thrown.expect(NoSuchBeanDefinitionException.class); this.thrown.expect(NoSuchBeanDefinitionException.class);
...@@ -155,7 +156,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -155,7 +156,7 @@ public class SpringApplicationAdminJmxAutoConfigurationTests {
} }
private String getProperty(ObjectName objectName, String key) throws Exception { private String getProperty(ObjectName objectName, String key) throws Exception {
return (String) this.mBeanServer.invoke(objectName, "getProperty", return (String) this.server.invoke(objectName, "getProperty",
new Object[] { key }, new String[] { String.class.getName() }); new Object[] { key }, new String[] { String.class.getName() });
} }
......
...@@ -46,12 +46,12 @@ public class AopAutoConfigurationTests { ...@@ -46,12 +46,12 @@ public class AopAutoConfigurationTests {
public void aopDisabled() { public void aopDisabled() {
this.contextRunner.withUserConfiguration(TestConfiguration.class) this.contextRunner.withUserConfiguration(TestConfiguration.class)
.withPropertyValues("spring.aop.auto:false").run((context) -> { .withPropertyValues("spring.aop.auto:false").run((context) -> {
TestAspect aspect = context.getBean(TestAspect.class); TestAspect aspect = context.getBean(TestAspect.class);
assertThat(aspect.isCalled()).isFalse(); assertThat(aspect.isCalled()).isFalse();
TestBean bean = context.getBean(TestBean.class); TestBean bean = context.getBean(TestBean.class);
bean.foo(); bean.foo();
assertThat(aspect.isCalled()).isFalse(); assertThat(aspect.isCalled()).isFalse();
}); });
} }
@Test @Test
......
...@@ -54,43 +54,46 @@ public class BatchAutoConfigurationWithoutJpaTests { ...@@ -54,43 +54,46 @@ public class BatchAutoConfigurationWithoutJpaTests {
@Test @Test
public void jdbcWithDefaultSettings() throws Exception { public void jdbcWithDefaultSettings() throws Exception {
this.contextRunner.withUserConfiguration(DefaultConfiguration.class, this.contextRunner
EmbeddedDataSourceConfiguration.class).withPropertyValues( .withUserConfiguration(DefaultConfiguration.class,
"spring.datasource.generate-unique-name=true").run((context) -> { EmbeddedDataSourceConfiguration.class)
assertThat(context).hasSingleBean(JobLauncher.class); .withPropertyValues("spring.datasource.generate-unique-name=true")
assertThat(context).hasSingleBean(JobExplorer.class); .run((context) -> {
assertThat(context).hasSingleBean(JobRepository.class); assertThat(context).hasSingleBean(JobLauncher.class);
assertThat(context).hasSingleBean(PlatformTransactionManager.class); assertThat(context).hasSingleBean(JobExplorer.class);
assertThat(context.getBean(PlatformTransactionManager.class).toString()) assertThat(context).hasSingleBean(JobRepository.class);
.contains("DataSourceTransactionManager"); assertThat(context).hasSingleBean(PlatformTransactionManager.class);
assertThat( assertThat(
context.getBean(BatchProperties.class).getInitializer().isEnabled()) context.getBean(PlatformTransactionManager.class).toString())
.isTrue(); .contains("DataSourceTransactionManager");
assertThat(new JdbcTemplate(context.getBean(DataSource.class)) assertThat(context.getBean(BatchProperties.class).getInitializer()
.queryForList("select * from BATCH_JOB_EXECUTION")).isEmpty(); .isEnabled()).isTrue();
assertThat( assertThat(new JdbcTemplate(context.getBean(DataSource.class))
context.getBean(JobExplorer.class).findRunningJobExecutions("test")) .queryForList("select * from BATCH_JOB_EXECUTION")).isEmpty();
.isEmpty(); assertThat(context.getBean(JobExplorer.class)
assertThat(context.getBean(JobRepository.class).getLastJobExecution("test", .findRunningJobExecutions("test")).isEmpty();
new JobParameters())).isNull(); assertThat(context.getBean(JobRepository.class)
}); .getLastJobExecution("test", new JobParameters())).isNull();
});
} }
@Test @Test
public void jdbcWithCustomPrefix() throws Exception { public void jdbcWithCustomPrefix() throws Exception {
this.contextRunner.withUserConfiguration(DefaultConfiguration.class, this.contextRunner
EmbeddedDataSourceConfiguration.class).withPropertyValues( .withUserConfiguration(DefaultConfiguration.class,
"spring.datasource.generate-unique-name=true", EmbeddedDataSourceConfiguration.class)
"spring.batch.schema:classpath:batch/custom-schema-hsql.sql", .withPropertyValues("spring.datasource.generate-unique-name=true",
"spring.batch.tablePrefix:PREFIX_").run((context) -> { "spring.batch.schema:classpath:batch/custom-schema-hsql.sql",
assertThat(new JdbcTemplate(context.getBean(DataSource.class)) "spring.batch.tablePrefix:PREFIX_")
.queryForList("select * from PREFIX_JOB_EXECUTION")).isEmpty(); .run((context) -> {
assertThat( assertThat(new JdbcTemplate(context.getBean(DataSource.class))
context.getBean(JobExplorer.class).findRunningJobExecutions("test")) .queryForList("select * from PREFIX_JOB_EXECUTION"))
.isEmpty(); .isEmpty();
assertThat(context.getBean(JobRepository.class).getLastJobExecution("test", assertThat(context.getBean(JobExplorer.class)
new JobParameters())).isNull(); .findRunningJobExecutions("test")).isEmpty();
}); assertThat(context.getBean(JobRepository.class)
.getLastJobExecution("test", new JobParameters())).isNull();
});
} }
@EnableBatchProcessing @EnableBatchProcessing
......
...@@ -80,8 +80,7 @@ public class JobLauncherCommandLineRunnerTests { ...@@ -80,8 +80,7 @@ public class JobLauncherCommandLineRunnerTests {
this.step = this.steps.get("step").tasklet(tasklet).build(); this.step = this.steps.get("step").tasklet(tasklet).build();
this.job = this.jobs.get("job").start(this.step).build(); this.job = this.jobs.get("job").start(this.step).build();
this.jobExplorer = this.context.getBean(JobExplorer.class); this.jobExplorer = this.context.getBean(JobExplorer.class);
this.runner = new JobLauncherCommandLineRunner(jobLauncher, this.runner = new JobLauncherCommandLineRunner(jobLauncher, this.jobExplorer);
this.jobExplorer);
this.context.getBean(BatchConfiguration.class).clear(); this.context.getBean(BatchConfiguration.class).clear();
} }
......
...@@ -50,31 +50,33 @@ public class CassandraAutoConfigurationTests { ...@@ -50,31 +50,33 @@ public class CassandraAutoConfigurationTests {
@Test @Test
public void createClusterWithOverrides() { public void createClusterWithOverrides() {
this.contextRunner.withPropertyValues( this.contextRunner
"spring.data.cassandra.cluster-name=testcluster").run((context) -> { .withPropertyValues("spring.data.cassandra.cluster-name=testcluster")
assertThat(context).hasSingleBean(Cluster.class); .run((context) -> {
assertThat(context.getBean(Cluster.class).getClusterName()) assertThat(context).hasSingleBean(Cluster.class);
.isEqualTo("testcluster"); assertThat(context.getBean(Cluster.class).getClusterName())
}); .isEqualTo("testcluster");
});
} }
@Test @Test
public void createCustomizeCluster() { public void createCustomizeCluster() {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(MockCustomizerConfig.class)
MockCustomizerConfig.class).run((context) -> { .run((context) -> {
assertThat(context).hasSingleBean(Cluster.class); assertThat(context).hasSingleBean(Cluster.class);
assertThat(context).hasSingleBean(ClusterBuilderCustomizer.class); assertThat(context).hasSingleBean(ClusterBuilderCustomizer.class);
}); });
} }
@Test @Test
public void customizerOverridesAutoConfig() { public void customizerOverridesAutoConfig() {
this.contextRunner.withUserConfiguration(SimpleCustomizerConfig.class) this.contextRunner.withUserConfiguration(SimpleCustomizerConfig.class)
.withPropertyValues("spring.data.cassandra.cluster-name=testcluster").run((context) -> { .withPropertyValues("spring.data.cassandra.cluster-name=testcluster")
assertThat(context).hasSingleBean(Cluster.class); .run((context) -> {
assertThat(context.getBean(Cluster.class).getClusterName()) assertThat(context).hasSingleBean(Cluster.class);
.isEqualTo("overridden-name"); assertThat(context.getBean(Cluster.class).getClusterName())
}); .isEqualTo("overridden-name");
});
} }
@Test @Test
...@@ -96,22 +98,23 @@ public class CassandraAutoConfigurationTests { ...@@ -96,22 +98,23 @@ public class CassandraAutoConfigurationTests {
@Test @Test
public void customizePoolOptions() { public void customizePoolOptions() {
this.contextRunner.withPropertyValues( this.contextRunner
"spring.data.cassandra.pool.idle-timeout=42", .withPropertyValues("spring.data.cassandra.pool.idle-timeout=42",
"spring.data.cassandra.pool.pool-timeout=52", "spring.data.cassandra.pool.pool-timeout=52",
"spring.data.cassandra.pool.heartbeat-interval=62", "spring.data.cassandra.pool.heartbeat-interval=62",
"spring.data.cassandra.pool.max-queue-size=72").run((context) -> { "spring.data.cassandra.pool.max-queue-size=72")
assertThat(context).hasSingleBean(Cluster.class); .run((context) -> {
PoolingOptions poolingOptions = context.getBean(Cluster.class) assertThat(context).hasSingleBean(Cluster.class);
.getConfiguration().getPoolingOptions(); PoolingOptions poolingOptions = context.getBean(Cluster.class)
assertThat(poolingOptions.getIdleTimeoutSeconds()).isEqualTo(42); .getConfiguration().getPoolingOptions();
assertThat(poolingOptions.getPoolTimeoutMillis()).isEqualTo(52); assertThat(poolingOptions.getIdleTimeoutSeconds()).isEqualTo(42);
assertThat(poolingOptions.getHeartbeatIntervalSeconds()).isEqualTo(62); assertThat(poolingOptions.getPoolTimeoutMillis()).isEqualTo(52);
assertThat(poolingOptions.getMaxQueueSize()).isEqualTo(72); assertThat(poolingOptions.getHeartbeatIntervalSeconds())
}); .isEqualTo(62);
assertThat(poolingOptions.getMaxQueueSize()).isEqualTo(72);
});
} }
@Configuration @Configuration
static class MockCustomizerConfig { static class MockCustomizerConfig {
......
...@@ -30,7 +30,6 @@ import org.springframework.beans.factory.support.BeanDefinitionRegistry; ...@@ -30,7 +30,6 @@ import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.RootBeanDefinition; import org.springframework.beans.factory.support.RootBeanDefinition;
import org.springframework.boot.test.context.assertj.AssertableApplicationContext; import org.springframework.boot.test.context.assertj.AssertableApplicationContext;
import org.springframework.boot.test.context.runner.ApplicationContextRunner; import org.springframework.boot.test.context.runner.ApplicationContextRunner;
import org.springframework.boot.test.context.runner.ContextConsumer;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import; import org.springframework.context.annotation.Import;
...@@ -55,40 +54,42 @@ public class ConditionalOnBeanTests { ...@@ -55,40 +54,42 @@ public class ConditionalOnBeanTests {
@Test @Test
public void testNameOnBeanCondition() { public void testNameOnBeanCondition() {
this.contextRunner.withUserConfiguration(FooConfiguration.class, this.contextRunner.withUserConfiguration(FooConfiguration.class,
OnBeanNameConfiguration.class).run(hasBarBean()); OnBeanNameConfiguration.class).run(this::hasBarBean);
} }
@Test @Test
public void testNameAndTypeOnBeanCondition() { public void testNameAndTypeOnBeanCondition() {
this.contextRunner.withUserConfiguration(FooConfiguration.class, this.contextRunner
OnBeanNameAndTypeConfiguration.class).run((context) -> .withUserConfiguration(FooConfiguration.class,
assertThat(context).doesNotHaveBean("bar")); OnBeanNameAndTypeConfiguration.class)
.run((context) -> assertThat(context).doesNotHaveBean("bar"));
} }
@Test @Test
public void testNameOnBeanConditionReverseOrder() { public void testNameOnBeanConditionReverseOrder() {
// Ideally this should be true // Ideally this should be true
this.contextRunner.withUserConfiguration(OnBeanNameConfiguration.class, this.contextRunner
FooConfiguration.class).run((context) -> .withUserConfiguration(OnBeanNameConfiguration.class,
assertThat(context).doesNotHaveBean("bar")); FooConfiguration.class)
.run((context) -> assertThat(context).doesNotHaveBean("bar"));
} }
@Test @Test
public void testClassOnBeanCondition() { public void testClassOnBeanCondition() {
this.contextRunner.withUserConfiguration(FooConfiguration.class, this.contextRunner.withUserConfiguration(FooConfiguration.class,
OnBeanClassConfiguration.class).run(hasBarBean()); OnBeanClassConfiguration.class).run(this::hasBarBean);
} }
@Test @Test
public void testClassOnBeanClassNameCondition() { public void testClassOnBeanClassNameCondition() {
this.contextRunner.withUserConfiguration(FooConfiguration.class, this.contextRunner.withUserConfiguration(FooConfiguration.class,
OnBeanClassNameConfiguration.class).run(hasBarBean()); OnBeanClassNameConfiguration.class).run(this::hasBarBean);
} }
@Test @Test
public void testOnBeanConditionWithXml() { public void testOnBeanConditionWithXml() {
this.contextRunner.withUserConfiguration(XmlConfiguration.class, this.contextRunner.withUserConfiguration(XmlConfiguration.class,
OnBeanNameConfiguration.class).run(hasBarBean()); OnBeanNameConfiguration.class).run(this::hasBarBean);
} }
@Test @Test
...@@ -101,22 +102,23 @@ public class ConditionalOnBeanTests { ...@@ -101,22 +102,23 @@ public class ConditionalOnBeanTests {
@Test @Test
public void testAnnotationOnBeanCondition() { public void testAnnotationOnBeanCondition() {
this.contextRunner.withUserConfiguration(FooConfiguration.class, this.contextRunner.withUserConfiguration(FooConfiguration.class,
OnAnnotationConfiguration.class).run(hasBarBean()); OnAnnotationConfiguration.class).run(this::hasBarBean);
} }
@Test @Test
public void testOnMissingBeanType() throws Exception { public void testOnMissingBeanType() throws Exception {
this.contextRunner.withUserConfiguration(FooConfiguration.class, this.contextRunner
OnBeanMissingClassConfiguration.class).run((context) -> .withUserConfiguration(FooConfiguration.class,
assertThat(context).doesNotHaveBean("bar")); OnBeanMissingClassConfiguration.class)
.run((context) -> assertThat(context).doesNotHaveBean("bar"));
} }
@Test @Test
public void withPropertyPlaceholderClassName() throws Exception { public void withPropertyPlaceholderClassName() throws Exception {
this.contextRunner.withUserConfiguration( this.contextRunner
PropertySourcesPlaceholderConfigurer.class, .withUserConfiguration(PropertySourcesPlaceholderConfigurer.class,
WithPropertyPlaceholderClassName.class, WithPropertyPlaceholderClassName.class,
OnBeanClassConfiguration.class) OnBeanClassConfiguration.class)
.withPropertyValues("mybeanclass=java.lang.String") .withPropertyValues("mybeanclass=java.lang.String")
.run(context -> assertThat(context).hasNotFailed()); .run(context -> assertThat(context).hasNotFailed());
} }
...@@ -125,19 +127,16 @@ public class ConditionalOnBeanTests { ...@@ -125,19 +127,16 @@ public class ConditionalOnBeanTests {
public void beanProducedByFactoryBeanIsConsideredWhenMatchingOnAnnotation() { public void beanProducedByFactoryBeanIsConsideredWhenMatchingOnAnnotation() {
this.contextRunner.withUserConfiguration(FactoryBeanConfiguration.class, this.contextRunner.withUserConfiguration(FactoryBeanConfiguration.class,
OnAnnotationWithFactoryBeanConfiguration.class).run((context) -> { OnAnnotationWithFactoryBeanConfiguration.class).run((context) -> {
assertThat(context).hasBean("bar"); assertThat(context).hasBean("bar");
assertThat(context).hasSingleBean(ExampleBean.class); assertThat(context).hasSingleBean(ExampleBean.class);
}); });
} }
private ContextConsumer<AssertableApplicationContext> hasBarBean() { private void hasBarBean(AssertableApplicationContext context) {
return (context) -> { assertThat(context).hasBean("bar");
assertThat(context).hasBean("bar"); assertThat(context.getBean("bar")).isEqualTo("bar");
assertThat(context.getBean("bar")).isEqualTo("bar");
};
} }
@Configuration @Configuration
@ConditionalOnBean(name = "foo") @ConditionalOnBean(name = "foo")
protected static class OnBeanNameConfiguration { protected static class OnBeanNameConfiguration {
......
/* /*
* Copyright 2012-2016 the original author or authors. * Copyright 2012-2017 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
...@@ -20,7 +20,6 @@ import org.junit.Test; ...@@ -20,7 +20,6 @@ import org.junit.Test;
import org.springframework.boot.test.context.assertj.AssertableApplicationContext; import org.springframework.boot.test.context.assertj.AssertableApplicationContext;
import org.springframework.boot.test.context.runner.ApplicationContextRunner; import org.springframework.boot.test.context.runner.ApplicationContextRunner;
import org.springframework.boot.test.context.runner.ContextConsumer;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import; import org.springframework.context.annotation.Import;
...@@ -40,38 +39,38 @@ public class ConditionalOnClassTests { ...@@ -40,38 +39,38 @@ public class ConditionalOnClassTests {
@Test @Test
public void testVanillaOnClassCondition() { public void testVanillaOnClassCondition() {
this.contextRunner.withUserConfiguration(BasicConfiguration.class, this.contextRunner
FooConfiguration.class).run(hasBarBean()); .withUserConfiguration(BasicConfiguration.class, FooConfiguration.class)
.run(this::hasBarBean);
} }
@Test @Test
public void testMissingOnClassCondition() { public void testMissingOnClassCondition() {
this.contextRunner.withUserConfiguration(MissingConfiguration.class, this.contextRunner
FooConfiguration.class).run((context) -> { .withUserConfiguration(MissingConfiguration.class, FooConfiguration.class)
assertThat(context).doesNotHaveBean("bar"); .run((context) -> {
assertThat(context).hasBean("foo"); assertThat(context).doesNotHaveBean("bar");
assertThat(context.getBean("foo")).isEqualTo("foo"); assertThat(context).hasBean("foo");
}); assertThat(context.getBean("foo")).isEqualTo("foo");
});
} }
@Test @Test
public void testOnClassConditionWithXml() { public void testOnClassConditionWithXml() {
this.contextRunner.withUserConfiguration(BasicConfiguration.class, this.contextRunner
XmlConfiguration.class).run(hasBarBean()); .withUserConfiguration(BasicConfiguration.class, XmlConfiguration.class)
.run(this::hasBarBean);
} }
@Test @Test
public void testOnClassConditionWithCombinedXml() { public void testOnClassConditionWithCombinedXml() {
this.contextRunner.withUserConfiguration(CombinedXmlConfiguration.class) this.contextRunner.withUserConfiguration(CombinedXmlConfiguration.class)
.run(hasBarBean()); .run(this::hasBarBean);
} }
private ContextConsumer<AssertableApplicationContext> hasBarBean() { private void hasBarBean(AssertableApplicationContext context) {
return (context) -> { assertThat(context).hasBean("bar");
assertThat(context).hasBean("bar"); assertThat(context.getBean("bar")).isEqualTo("bar");
assertThat(context.getBean("bar")).isEqualTo("bar");
};
} }
@Configuration @Configuration
......
...@@ -19,9 +19,7 @@ package org.springframework.boot.autoconfigure.condition; ...@@ -19,9 +19,7 @@ package org.springframework.boot.autoconfigure.condition;
import org.junit.Test; import org.junit.Test;
import org.springframework.boot.cloud.CloudPlatform; import org.springframework.boot.cloud.CloudPlatform;
import org.springframework.boot.test.context.assertj.AssertableApplicationContext;
import org.springframework.boot.test.context.runner.ApplicationContextRunner; import org.springframework.boot.test.context.runner.ApplicationContextRunner;
import org.springframework.boot.test.context.runner.ContextConsumer;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
...@@ -37,32 +35,21 @@ public class ConditionalOnCloudPlatformTests { ...@@ -37,32 +35,21 @@ public class ConditionalOnCloudPlatformTests {
@Test @Test
public void outcomeWhenCloudfoundryPlatformNotPresentShouldNotMatch() { public void outcomeWhenCloudfoundryPlatformNotPresentShouldNotMatch() {
this.contextRunner.withUserConfiguration(CloudFoundryPlatformConfig.class) this.contextRunner.withUserConfiguration(CloudFoundryPlatformConfig.class)
.run(match(false)); .run((context) -> assertThat(context).doesNotHaveBean("foo"));
} }
@Test @Test
public void outcomeWhenCloudfoundryPlatformPresentShouldMatch() { public void outcomeWhenCloudfoundryPlatformPresentShouldMatch() {
this.contextRunner.withUserConfiguration(CloudFoundryPlatformConfig.class) this.contextRunner.withUserConfiguration(CloudFoundryPlatformConfig.class)
.withPropertyValues("VCAP_APPLICATION:---") .withPropertyValues("VCAP_APPLICATION:---")
.run(match(true)); .run((context) -> assertThat(context).hasBean("foo"));
} }
@Test @Test
public void outcomeWhenCloudfoundryPlatformPresentAndMethodTargetShouldMatch() { public void outcomeWhenCloudfoundryPlatformPresentAndMethodTargetShouldMatch() {
this.contextRunner.withUserConfiguration(CloudFoundryPlatformOnMethodConfig.class) this.contextRunner.withUserConfiguration(CloudFoundryPlatformOnMethodConfig.class)
.withPropertyValues("VCAP_APPLICATION:---") .withPropertyValues("VCAP_APPLICATION:---")
.run(match(true)); .run((context) -> assertThat(context).hasBean("foo"));
}
private ContextConsumer<AssertableApplicationContext> match(boolean expected) {
return (context) -> {
if (expected) {
assertThat(context).hasBean("foo");
}
else {
assertThat(context).doesNotHaveBean("foo");
}
};
} }
@Configuration @Configuration
......
...@@ -18,9 +18,7 @@ package org.springframework.boot.autoconfigure.condition; ...@@ -18,9 +18,7 @@ package org.springframework.boot.autoconfigure.condition;
import org.junit.Test; import org.junit.Test;
import org.springframework.boot.test.context.assertj.AssertableApplicationContext;
import org.springframework.boot.test.context.runner.ApplicationContextRunner; import org.springframework.boot.test.context.runner.ApplicationContextRunner;
import org.springframework.boot.test.context.runner.ContextConsumer;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
...@@ -39,31 +37,19 @@ public class ConditionalOnExpressionTests { ...@@ -39,31 +37,19 @@ public class ConditionalOnExpressionTests {
@Test @Test
public void expressionIsTrue() { public void expressionIsTrue() {
this.contextRunner.withUserConfiguration(BasicConfiguration.class) this.contextRunner.withUserConfiguration(BasicConfiguration.class)
.run(match(true)); .run((context) -> assertThat(context.getBean("foo")).isEqualTo("foo"));
} }
@Test @Test
public void expressionIsFalse() { public void expressionIsFalse() {
this.contextRunner.withUserConfiguration(MissingConfiguration.class) this.contextRunner.withUserConfiguration(MissingConfiguration.class)
.run(match(false)); .run((context) -> assertThat(context).doesNotHaveBean("foo"));
} }
@Test @Test
public void expressionIsNull() { public void expressionIsNull() {
this.contextRunner.withUserConfiguration(NullConfiguration.class) this.contextRunner.withUserConfiguration(NullConfiguration.class)
.run(match(false)); .run((context) -> assertThat(context).doesNotHaveBean("foo"));
}
private ContextConsumer<AssertableApplicationContext> match(boolean expected) {
return (context) -> {
if (expected) {
assertThat(context).hasBean("foo");
assertThat(context.getBean("foo")).isEqualTo("foo");
}
else {
assertThat(context).doesNotHaveBean("foo");
}
};
} }
@Configuration @Configuration
......
/* /*
* Copyright 2012-2016 the original author or authors. * Copyright 2012-2017 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
...@@ -29,9 +29,7 @@ import org.junit.Test; ...@@ -29,9 +29,7 @@ import org.junit.Test;
import org.springframework.boot.autoconfigure.condition.ConditionalOnJava.JavaVersion; import org.springframework.boot.autoconfigure.condition.ConditionalOnJava.JavaVersion;
import org.springframework.boot.autoconfigure.condition.ConditionalOnJava.Range; import org.springframework.boot.autoconfigure.condition.ConditionalOnJava.Range;
import org.springframework.boot.test.context.assertj.AssertableApplicationContext;
import org.springframework.boot.test.context.runner.ApplicationContextRunner; import org.springframework.boot.test.context.runner.ApplicationContextRunner;
import org.springframework.boot.test.context.runner.ContextConsumer;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.util.ReflectionUtils; import org.springframework.util.ReflectionUtils;
...@@ -52,17 +50,20 @@ public class ConditionalOnJavaTests { ...@@ -52,17 +50,20 @@ public class ConditionalOnJavaTests {
@Test @Test
public void doesNotMatchIfBetterVersionIsRequired() { public void doesNotMatchIfBetterVersionIsRequired() {
this.contextRunner.withUserConfiguration(Java9Required.class).run(match(false)); this.contextRunner.withUserConfiguration(Java9Required.class)
.run((context) -> assertThat(context).doesNotHaveBean(String.class));
} }
@Test @Test
public void doesNotMatchIfLowerIsRequired() { public void doesNotMatchIfLowerIsRequired() {
this.contextRunner.withUserConfiguration(Java7Required.class).run(match(false)); this.contextRunner.withUserConfiguration(Java7Required.class)
.run((context) -> assertThat(context).doesNotHaveBean(String.class));
} }
@Test @Test
public void matchesIfVersionIsInRange() { public void matchesIfVersionIsInRange() {
this.contextRunner.withUserConfiguration(Java8Required.class).run(match(true)); this.contextRunner.withUserConfiguration(Java8Required.class)
.run((context) -> assertThat(context).hasSingleBean(String.class));
} }
@Test @Test
...@@ -105,10 +106,8 @@ public class ConditionalOnJavaTests { ...@@ -105,10 +106,8 @@ public class ConditionalOnJavaTests {
private String getJavaVersion(Class<?>... hiddenClasses) throws Exception { private String getJavaVersion(Class<?>... hiddenClasses) throws Exception {
URL[] urls = ((URLClassLoader) getClass().getClassLoader()).getURLs(); URL[] urls = ((URLClassLoader) getClass().getClassLoader()).getURLs();
URLClassLoader classLoader = new ClassHidingClassLoader(urls, hiddenClasses); URLClassLoader classLoader = new ClassHidingClassLoader(urls, hiddenClasses);
Class<?> javaVersionClass = classLoader Class<?> javaVersionClass = classLoader
.loadClass(ConditionalOnJava.JavaVersion.class.getName()); .loadClass(ConditionalOnJava.JavaVersion.class.getName());
Method getJavaVersionMethod = ReflectionUtils.findMethod(javaVersionClass, Method getJavaVersionMethod = ReflectionUtils.findMethod(javaVersionClass,
"getJavaVersion"); "getJavaVersion");
Object javaVersion = ReflectionUtils.invokeMethod(getJavaVersionMethod, null); Object javaVersion = ReflectionUtils.invokeMethod(getJavaVersionMethod, null);
...@@ -123,17 +122,6 @@ public class ConditionalOnJavaTests { ...@@ -123,17 +122,6 @@ public class ConditionalOnJavaTests {
assertThat(outcome.isMatch()).as(outcome.getMessage()).isEqualTo(expected); assertThat(outcome.isMatch()).as(outcome.getMessage()).isEqualTo(expected);
} }
private ContextConsumer<AssertableApplicationContext> match(boolean expected) {
return (context) -> {
if (expected) {
assertThat(context).hasSingleBean(String.class);
}
else {
assertThat(context).doesNotHaveBean(String.class);
}
};
}
private final class ClassHidingClassLoader extends URLClassLoader { private final class ClassHidingClassLoader extends URLClassLoader {
private final List<Class<?>> hiddenClasses; private final List<Class<?>> hiddenClasses;
......
...@@ -27,9 +27,7 @@ import org.junit.Test; ...@@ -27,9 +27,7 @@ import org.junit.Test;
import org.springframework.boot.autoconfigure.jndi.JndiPropertiesHidingClassLoader; import org.springframework.boot.autoconfigure.jndi.JndiPropertiesHidingClassLoader;
import org.springframework.boot.autoconfigure.jndi.TestableInitialContextFactory; import org.springframework.boot.autoconfigure.jndi.TestableInitialContextFactory;
import org.springframework.boot.test.context.assertj.AssertableApplicationContext;
import org.springframework.boot.test.context.runner.ApplicationContextRunner; import org.springframework.boot.test.context.runner.ApplicationContextRunner;
import org.springframework.boot.test.context.runner.ContextConsumer;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.core.type.AnnotatedTypeMetadata; import org.springframework.core.type.AnnotatedTypeMetadata;
...@@ -77,22 +75,26 @@ public class ConditionalOnJndiTests { ...@@ -77,22 +75,26 @@ public class ConditionalOnJndiTests {
@Test @Test
public void jndiNotAvailable() { public void jndiNotAvailable() {
this.contextRunner.withUserConfiguration(JndiAvailableConfiguration.class, this.contextRunner
JndiConditionConfiguration.class).run(match(false)); .withUserConfiguration(JndiAvailableConfiguration.class,
JndiConditionConfiguration.class)
.run((context) -> assertThat(context).doesNotHaveBean(String.class));
} }
@Test @Test
public void jndiAvailable() { public void jndiAvailable() {
setupJndi(); setupJndi();
this.contextRunner.withUserConfiguration(JndiAvailableConfiguration.class, this.contextRunner
JndiConditionConfiguration.class).run(match(true)); .withUserConfiguration(JndiAvailableConfiguration.class,
JndiConditionConfiguration.class)
.run((context) -> assertThat(context).hasSingleBean(String.class));
} }
@Test @Test
public void jndiLocationNotBound() { public void jndiLocationNotBound() {
setupJndi(); setupJndi();
this.contextRunner.withUserConfiguration(JndiConditionConfiguration.class) this.contextRunner.withUserConfiguration(JndiConditionConfiguration.class)
.run(match(false)); .run((context) -> assertThat(context).doesNotHaveBean(String.class));
} }
@Test @Test
...@@ -100,7 +102,7 @@ public class ConditionalOnJndiTests { ...@@ -100,7 +102,7 @@ public class ConditionalOnJndiTests {
setupJndi(); setupJndi();
TestableInitialContextFactory.bind("java:/FooManager", new Object()); TestableInitialContextFactory.bind("java:/FooManager", new Object());
this.contextRunner.withUserConfiguration(JndiConditionConfiguration.class) this.contextRunner.withUserConfiguration(JndiConditionConfiguration.class)
.run(match(true)); .run((context) -> assertThat(context).hasSingleBean(String.class));
} }
@Test @Test
...@@ -124,17 +126,6 @@ public class ConditionalOnJndiTests { ...@@ -124,17 +126,6 @@ public class ConditionalOnJndiTests {
TestableInitialContextFactory.class.getName()); TestableInitialContextFactory.class.getName());
} }
private ContextConsumer<AssertableApplicationContext> match(boolean expected) {
return (context) -> {
if (expected) {
assertThat(context).hasSingleBean(String.class);
}
else {
assertThat(context).doesNotHaveBean(String.class);
}
};
}
private AnnotatedTypeMetadata mockMetaData(String... value) { private AnnotatedTypeMetadata mockMetaData(String... value) {
AnnotatedTypeMetadata metadata = mock(AnnotatedTypeMetadata.class); AnnotatedTypeMetadata metadata = mock(AnnotatedTypeMetadata.class);
Map<String, Object> attributes = new HashMap<>(); Map<String, Object> attributes = new HashMap<>();
......
...@@ -292,20 +292,13 @@ abstract class AbstractApplicationContextRunner<SELF extends AbstractApplication ...@@ -292,20 +292,13 @@ abstract class AbstractApplicationContextRunner<SELF extends AbstractApplication
consumer.accept(context); consumer.accept(context);
} }
catch (Throwable ex) { catch (Throwable ex) {
AnyThrow.throwUnchecked(ex); rethrow(ex);
} }
} }
private static class AnyThrow { @SuppressWarnings("unchecked")
private <E extends Throwable> void rethrow(Throwable e) throws E {
static void throwUnchecked(Throwable e) { throw (E) e;
AnyThrow.throwAny(e);
}
@SuppressWarnings("unchecked")
private static <E extends Throwable> void throwAny(Throwable e) throws E {
throw (E) e;
}
} }
} }
/* /*
* Copyright 2012-2016 the original author or authors. * Copyright 2012-2017 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
......
...@@ -130,11 +130,16 @@ public final class TestPropertyValues { ...@@ -130,11 +130,16 @@ public final class TestPropertyValues {
return call.call(); return call.call();
} }
catch (Exception ex) { catch (Exception ex) {
AnyThrow.throwUnchecked(ex); rethrow(ex);
return null; // never reached throw new IllegalStateException("Original cause not rethrown", ex);
} }
} }
@SuppressWarnings("unchecked")
private <E extends Throwable> void rethrow(Throwable e) throws E {
throw (E) e;
}
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private void addToSources(MutablePropertySources sources, Type type, String name) { private void addToSources(MutablePropertySources sources, Type type, String name) {
if (sources.contains(name)) { if (sources.contains(name)) {
...@@ -310,16 +315,4 @@ public final class TestPropertyValues { ...@@ -310,16 +315,4 @@ public final class TestPropertyValues {
} }
private static class AnyThrow {
static void throwUnchecked(Throwable e) {
AnyThrow.throwAny(e);
}
@SuppressWarnings("unchecked")
private static <E extends Throwable> void throwAny(Throwable e) throws E {
throw (E) e;
}
}
} }
...@@ -153,7 +153,8 @@ public abstract class AbstractApplicationContextRunnerTests<T extends AbstractAp ...@@ -153,7 +153,8 @@ public abstract class AbstractApplicationContextRunnerTests<T extends AbstractAp
new HidePackagesClassLoader(Gson.class.getPackage().getName())) new HidePackagesClassLoader(Gson.class.getPackage().getName()))
.run((context) -> { .run((context) -> {
try { try {
ClassUtils.forName(Gson.class.getName(), context.getClassLoader()); ClassUtils.forName(Gson.class.getName(),
context.getClassLoader());
fail("Should have thrown a ClassNotFoundException"); fail("Should have thrown a ClassNotFoundException");
} }
catch (ClassNotFoundException e) { catch (ClassNotFoundException e) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment