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,17 +87,19 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -87,17 +87,19 @@ 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
.withSystemProperties(
"spring.application.admin.jmx-name=" + customJmxName) "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());
}); });
} }
...@@ -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,7 +130,6 @@ public class SpringApplicationAdminJmxAutoConfigurationTests { ...@@ -128,7 +130,6 @@ 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
...@@ -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() });
} }
......
...@@ -77,8 +77,8 @@ public class RabbitAutoConfigurationTests { ...@@ -77,8 +77,8 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testDefaultRabbitConfiguration() { public void testDefaultRabbitConfiguration() {
this.contextRunner this.contextRunner.withUserConfiguration(TestConfiguration.class)
.withUserConfiguration(TestConfiguration.class).run((context) -> { .run((context) -> {
RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class); RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
RabbitMessagingTemplate messagingTemplate = context RabbitMessagingTemplate messagingTemplate = context
.getBean(RabbitMessagingTemplate.class); .getBean(RabbitMessagingTemplate.class);
...@@ -86,15 +86,19 @@ public class RabbitAutoConfigurationTests { ...@@ -86,15 +86,19 @@ public class RabbitAutoConfigurationTests {
.getBean(CachingConnectionFactory.class); .getBean(CachingConnectionFactory.class);
DirectFieldAccessor dfa = new DirectFieldAccessor(connectionFactory); DirectFieldAccessor dfa = new DirectFieldAccessor(connectionFactory);
RabbitAdmin amqpAdmin = context.getBean(RabbitAdmin.class); RabbitAdmin amqpAdmin = context.getBean(RabbitAdmin.class);
assertThat(rabbitTemplate.getConnectionFactory()).isEqualTo(connectionFactory); assertThat(rabbitTemplate.getConnectionFactory())
.isEqualTo(connectionFactory);
assertThat(getMandatory(rabbitTemplate)).isFalse(); assertThat(getMandatory(rabbitTemplate)).isFalse();
assertThat(messagingTemplate.getRabbitTemplate()).isEqualTo(rabbitTemplate); assertThat(messagingTemplate.getRabbitTemplate())
.isEqualTo(rabbitTemplate);
assertThat(amqpAdmin).isNotNull(); assertThat(amqpAdmin).isNotNull();
assertThat(connectionFactory.getHost()).isEqualTo("localhost"); assertThat(connectionFactory.getHost()).isEqualTo("localhost");
assertThat(dfa.getPropertyValue("publisherConfirms")).isEqualTo(false); assertThat(dfa.getPropertyValue("publisherConfirms"))
.isEqualTo(false);
assertThat(dfa.getPropertyValue("publisherReturns")).isEqualTo(false); assertThat(dfa.getPropertyValue("publisherReturns")).isEqualTo(false);
assertThat(context.containsBean("rabbitListenerContainerFactory")) assertThat(context.containsBean("rabbitListenerContainerFactory"))
.as("Listener container factory should be created by default").isTrue(); .as("Listener container factory should be created by default")
.isTrue();
}); });
} }
...@@ -105,7 +109,8 @@ public class RabbitAutoConfigurationTests { ...@@ -105,7 +109,8 @@ public class RabbitAutoConfigurationTests {
"spring.rabbitmq.port:9000", "spring.rabbitmq.username:alice", "spring.rabbitmq.port:9000", "spring.rabbitmq.username:alice",
"spring.rabbitmq.password:secret", "spring.rabbitmq.password:secret",
"spring.rabbitmq.virtual_host:/vhost", "spring.rabbitmq.virtual_host:/vhost",
"spring.rabbitmq.connection-timeout:123").run((context) -> { "spring.rabbitmq.connection-timeout:123")
.run((context) -> {
CachingConnectionFactory connectionFactory = context CachingConnectionFactory connectionFactory = context
.getBean(CachingConnectionFactory.class); .getBean(CachingConnectionFactory.class);
assertThat(connectionFactory.getHost()).isEqualTo("remote-server"); assertThat(connectionFactory.getHost()).isEqualTo("remote-server");
...@@ -164,7 +169,8 @@ public class RabbitAutoConfigurationTests { ...@@ -164,7 +169,8 @@ public class RabbitAutoConfigurationTests {
public void testConnectionFactoryPublisherSettings() { public void testConnectionFactoryPublisherSettings() {
this.contextRunner.withUserConfiguration(TestConfiguration.class) this.contextRunner.withUserConfiguration(TestConfiguration.class)
.withPropertyValues("spring.rabbitmq.publisher-confirms=true", .withPropertyValues("spring.rabbitmq.publisher-confirms=true",
"spring.rabbitmq.publisher-returns=true").run(context -> { "spring.rabbitmq.publisher-returns=true")
.run(context -> {
CachingConnectionFactory connectionFactory = context CachingConnectionFactory connectionFactory = context
.getBean(CachingConnectionFactory.class); .getBean(CachingConnectionFactory.class);
RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class); RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
...@@ -177,8 +183,8 @@ public class RabbitAutoConfigurationTests { ...@@ -177,8 +183,8 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testRabbitTemplateMessageConverters() { public void testRabbitTemplateMessageConverters() {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(MessageConvertersConfiguration.class)
MessageConvertersConfiguration.class).run((context) -> { .run((context) -> {
RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class); RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
assertThat(rabbitTemplate.getMessageConverter()) assertThat(rabbitTemplate.getMessageConverter())
.isSameAs(context.getBean("myMessageConverter")); .isSameAs(context.getBean("myMessageConverter"));
...@@ -196,8 +202,8 @@ public class RabbitAutoConfigurationTests { ...@@ -196,8 +202,8 @@ public class RabbitAutoConfigurationTests {
"spring.rabbitmq.template.retry.multiplier:1.5", "spring.rabbitmq.template.retry.multiplier:1.5",
"spring.rabbitmq.template.retry.maxInterval:5000", "spring.rabbitmq.template.retry.maxInterval:5000",
"spring.rabbitmq.template.receiveTimeout:123", "spring.rabbitmq.template.receiveTimeout:123",
"spring.rabbitmq.template.replyTimeout:456" "spring.rabbitmq.template.replyTimeout:456")
).run((context) -> { .run((context) -> {
RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class); RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
DirectFieldAccessor dfa = new DirectFieldAccessor(rabbitTemplate); DirectFieldAccessor dfa = new DirectFieldAccessor(rabbitTemplate);
assertThat(dfa.getPropertyValue("receiveTimeout")).isEqualTo(123L); assertThat(dfa.getPropertyValue("receiveTimeout")).isEqualTo(123L);
...@@ -240,12 +246,13 @@ public class RabbitAutoConfigurationTests { ...@@ -240,12 +246,13 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testConnectionFactoryBackOff() { public void testConnectionFactoryBackOff() {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(TestConfiguration2.class)
TestConfiguration2.class).run((context) -> { .run((context) -> {
RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class); RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
CachingConnectionFactory connectionFactory = context CachingConnectionFactory connectionFactory = context
.getBean(CachingConnectionFactory.class); .getBean(CachingConnectionFactory.class);
assertThat(connectionFactory).isEqualTo(rabbitTemplate.getConnectionFactory()); assertThat(connectionFactory)
.isEqualTo(rabbitTemplate.getConnectionFactory());
assertThat(connectionFactory.getHost()).isEqualTo("otherserver"); assertThat(connectionFactory.getHost()).isEqualTo("otherserver");
assertThat(connectionFactory.getPort()).isEqualTo(8001); assertThat(connectionFactory.getPort()).isEqualTo(8001);
}); });
...@@ -257,14 +264,17 @@ public class RabbitAutoConfigurationTests { ...@@ -257,14 +264,17 @@ public class RabbitAutoConfigurationTests {
.withPropertyValues("spring.rabbitmq.cache.channel.size=23", .withPropertyValues("spring.rabbitmq.cache.channel.size=23",
"spring.rabbitmq.cache.channel.checkoutTimeout=1000", "spring.rabbitmq.cache.channel.checkoutTimeout=1000",
"spring.rabbitmq.cache.connection.mode=CONNECTION", "spring.rabbitmq.cache.connection.mode=CONNECTION",
"spring.rabbitmq.cache.connection.size=2").run((context) -> { "spring.rabbitmq.cache.connection.size=2")
.run((context) -> {
CachingConnectionFactory connectionFactory = context CachingConnectionFactory connectionFactory = context
.getBean(CachingConnectionFactory.class); .getBean(CachingConnectionFactory.class);
DirectFieldAccessor dfa = new DirectFieldAccessor(connectionFactory); DirectFieldAccessor dfa = new DirectFieldAccessor(connectionFactory);
assertThat(dfa.getPropertyValue("channelCacheSize")).isEqualTo(23); assertThat(dfa.getPropertyValue("channelCacheSize")).isEqualTo(23);
assertThat(dfa.getPropertyValue("cacheMode")).isEqualTo(CacheMode.CONNECTION); assertThat(dfa.getPropertyValue("cacheMode"))
.isEqualTo(CacheMode.CONNECTION);
assertThat(dfa.getPropertyValue("connectionCacheSize")).isEqualTo(2); assertThat(dfa.getPropertyValue("connectionCacheSize")).isEqualTo(2);
assertThat(dfa.getPropertyValue("channelCheckoutTimeout")).isEqualTo(1000L); assertThat(dfa.getPropertyValue("channelCheckoutTimeout"))
.isEqualTo(1000L);
}); });
} }
...@@ -280,11 +290,12 @@ public class RabbitAutoConfigurationTests { ...@@ -280,11 +290,12 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testRabbitMessagingTemplateBackOff() { public void testRabbitMessagingTemplateBackOff() {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(TestConfiguration4.class)
TestConfiguration4.class).run((context) -> { .run((context) -> {
RabbitMessagingTemplate messagingTemplate = context RabbitMessagingTemplate messagingTemplate = context
.getBean(RabbitMessagingTemplate.class); .getBean(RabbitMessagingTemplate.class);
assertThat(messagingTemplate.getDefaultDestination()).isEqualTo("fooBar"); assertThat(messagingTemplate.getDefaultDestination())
.isEqualTo("fooBar");
}); });
} }
...@@ -292,7 +303,8 @@ public class RabbitAutoConfigurationTests { ...@@ -292,7 +303,8 @@ public class RabbitAutoConfigurationTests {
public void testStaticQueues() { public void testStaticQueues() {
this.contextRunner.withUserConfiguration(TestConfiguration.class) this.contextRunner.withUserConfiguration(TestConfiguration.class)
.withPropertyValues("spring.rabbitmq.dynamic:false").run((context) -> { .withPropertyValues("spring.rabbitmq.dynamic:false").run((context) -> {
// There should NOT be an AmqpAdmin bean when dynamic is switch to false // There should NOT be an AmqpAdmin bean when dynamic is switch to
// false
this.thrown.expect(NoSuchBeanDefinitionException.class); this.thrown.expect(NoSuchBeanDefinitionException.class);
this.thrown.expectMessage("No qualifying bean of type"); this.thrown.expectMessage("No qualifying bean of type");
this.thrown.expectMessage(AmqpAdmin.class.getName()); this.thrown.expectMessage(AmqpAdmin.class.getName());
...@@ -302,8 +314,8 @@ public class RabbitAutoConfigurationTests { ...@@ -302,8 +314,8 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testEnableRabbitCreateDefaultContainerFactory() { public void testEnableRabbitCreateDefaultContainerFactory() {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(EnableRabbitConfiguration.class)
EnableRabbitConfiguration.class).run((context) -> { .run((context) -> {
RabbitListenerContainerFactory<?> rabbitListenerContainerFactory = context RabbitListenerContainerFactory<?> rabbitListenerContainerFactory = context
.getBean("rabbitListenerContainerFactory", .getBean("rabbitListenerContainerFactory",
RabbitListenerContainerFactory.class); RabbitListenerContainerFactory.class);
...@@ -314,14 +326,15 @@ public class RabbitAutoConfigurationTests { ...@@ -314,14 +326,15 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testRabbitListenerContainerFactoryBackOff() { public void testRabbitListenerContainerFactoryBackOff() {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(TestConfiguration5.class)
TestConfiguration5.class).run((context) -> { .run((context) -> {
SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory = context SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory = context
.getBean("rabbitListenerContainerFactory", .getBean("rabbitListenerContainerFactory",
SimpleRabbitListenerContainerFactory.class); SimpleRabbitListenerContainerFactory.class);
rabbitListenerContainerFactory.setTxSize(10); rabbitListenerContainerFactory.setTxSize(10);
verify(rabbitListenerContainerFactory).setTxSize(10); verify(rabbitListenerContainerFactory).setTxSize(10);
DirectFieldAccessor dfa = new DirectFieldAccessor(rabbitListenerContainerFactory); DirectFieldAccessor dfa = new DirectFieldAccessor(
rabbitListenerContainerFactory);
Advice[] adviceChain = (Advice[]) dfa.getPropertyValue("adviceChain"); Advice[] adviceChain = (Advice[]) dfa.getPropertyValue("adviceChain");
assertThat(adviceChain).isNull(); assertThat(adviceChain).isNull();
}); });
...@@ -329,9 +342,10 @@ public class RabbitAutoConfigurationTests { ...@@ -329,9 +342,10 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testSimpleRabbitListenerContainerFactoryWithCustomSettings() { public void testSimpleRabbitListenerContainerFactoryWithCustomSettings() {
this.contextRunner.withUserConfiguration(MessageConvertersConfiguration.class, this.contextRunner
MessageRecoverersConfiguration.class).withPropertyValues( .withUserConfiguration(MessageConvertersConfiguration.class,
"spring.rabbitmq.listener.simple.retry.enabled:true", MessageRecoverersConfiguration.class)
.withPropertyValues("spring.rabbitmq.listener.simple.retry.enabled:true",
"spring.rabbitmq.listener.simple.retry.maxAttempts:4", "spring.rabbitmq.listener.simple.retry.maxAttempts:4",
"spring.rabbitmq.listener.simple.retry.initialInterval:2000", "spring.rabbitmq.listener.simple.retry.initialInterval:2000",
"spring.rabbitmq.listener.simple.retry.multiplier:1.5", "spring.rabbitmq.listener.simple.retry.multiplier:1.5",
...@@ -343,13 +357,16 @@ public class RabbitAutoConfigurationTests { ...@@ -343,13 +357,16 @@ public class RabbitAutoConfigurationTests {
"spring.rabbitmq.listener.simple.prefetch:40", "spring.rabbitmq.listener.simple.prefetch:40",
"spring.rabbitmq.listener.simple.defaultRequeueRejected:false", "spring.rabbitmq.listener.simple.defaultRequeueRejected:false",
"spring.rabbitmq.listener.simple.idleEventInterval:5", "spring.rabbitmq.listener.simple.idleEventInterval:5",
"spring.rabbitmq.listener.simple.transactionSize:20").run((context) -> { "spring.rabbitmq.listener.simple.transactionSize:20")
.run((context) -> {
SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory = context SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory = context
.getBean("rabbitListenerContainerFactory", .getBean("rabbitListenerContainerFactory",
SimpleRabbitListenerContainerFactory.class); SimpleRabbitListenerContainerFactory.class);
DirectFieldAccessor dfa = new DirectFieldAccessor(rabbitListenerContainerFactory); DirectFieldAccessor dfa = new DirectFieldAccessor(
rabbitListenerContainerFactory);
assertThat(dfa.getPropertyValue("concurrentConsumers")).isEqualTo(5); assertThat(dfa.getPropertyValue("concurrentConsumers")).isEqualTo(5);
assertThat(dfa.getPropertyValue("maxConcurrentConsumers")).isEqualTo(10); assertThat(dfa.getPropertyValue("maxConcurrentConsumers"))
.isEqualTo(10);
assertThat(dfa.getPropertyValue("txSize")).isEqualTo(20); assertThat(dfa.getPropertyValue("txSize")).isEqualTo(20);
checkCommonProps(context, dfa); checkCommonProps(context, dfa);
}); });
...@@ -357,9 +374,10 @@ public class RabbitAutoConfigurationTests { ...@@ -357,9 +374,10 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void testDirectRabbitListenerContainerFactoryWithCustomSettings() { public void testDirectRabbitListenerContainerFactoryWithCustomSettings() {
this.contextRunner.withUserConfiguration(MessageConvertersConfiguration.class, this.contextRunner
MessageRecoverersConfiguration.class).withPropertyValues( .withUserConfiguration(MessageConvertersConfiguration.class,
"spring.rabbitmq.listener.type:direct", MessageRecoverersConfiguration.class)
.withPropertyValues("spring.rabbitmq.listener.type:direct",
"spring.rabbitmq.listener.direct.retry.enabled:true", "spring.rabbitmq.listener.direct.retry.enabled:true",
"spring.rabbitmq.listener.direct.retry.maxAttempts:4", "spring.rabbitmq.listener.direct.retry.maxAttempts:4",
"spring.rabbitmq.listener.direct.retry.initialInterval:2000", "spring.rabbitmq.listener.direct.retry.initialInterval:2000",
...@@ -370,11 +388,13 @@ public class RabbitAutoConfigurationTests { ...@@ -370,11 +388,13 @@ public class RabbitAutoConfigurationTests {
"spring.rabbitmq.listener.direct.consumers-per-queue:5", "spring.rabbitmq.listener.direct.consumers-per-queue:5",
"spring.rabbitmq.listener.direct.prefetch:40", "spring.rabbitmq.listener.direct.prefetch:40",
"spring.rabbitmq.listener.direct.defaultRequeueRejected:false", "spring.rabbitmq.listener.direct.defaultRequeueRejected:false",
"spring.rabbitmq.listener.direct.idleEventInterval:5").run((context) -> { "spring.rabbitmq.listener.direct.idleEventInterval:5")
.run((context) -> {
DirectRabbitListenerContainerFactory rabbitListenerContainerFactory = context DirectRabbitListenerContainerFactory rabbitListenerContainerFactory = context
.getBean("rabbitListenerContainerFactory", .getBean("rabbitListenerContainerFactory",
DirectRabbitListenerContainerFactory.class); DirectRabbitListenerContainerFactory.class);
DirectFieldAccessor dfa = new DirectFieldAccessor(rabbitListenerContainerFactory); DirectFieldAccessor dfa = new DirectFieldAccessor(
rabbitListenerContainerFactory);
assertThat(dfa.getPropertyValue("consumersPerQueue")).isEqualTo(5); assertThat(dfa.getPropertyValue("consumersPerQueue")).isEqualTo(5);
checkCommonProps(context, dfa); checkCommonProps(context, dfa);
}); });
...@@ -387,7 +407,8 @@ public class RabbitAutoConfigurationTests { ...@@ -387,7 +407,8 @@ public class RabbitAutoConfigurationTests {
"spring.rabbitmq.listener.simple.maxConcurrency:10", "spring.rabbitmq.listener.simple.maxConcurrency:10",
"spring.rabbitmq.listener.simple.prefetch:40", "spring.rabbitmq.listener.simple.prefetch:40",
"spring.rabbitmq.listener.direct.consumers-per-queue:5", "spring.rabbitmq.listener.direct.consumers-per-queue:5",
"spring.rabbitmq.listener.direct.prefetch:40").run((context) -> { "spring.rabbitmq.listener.direct.prefetch:40")
.run((context) -> {
assertThat(context).hasSingleBean( assertThat(context).hasSingleBean(
SimpleRabbitListenerContainerFactoryConfigurer.class); SimpleRabbitListenerContainerFactoryConfigurer.class);
assertThat(context).hasSingleBean( assertThat(context).hasSingleBean(
...@@ -401,9 +422,11 @@ public class RabbitAutoConfigurationTests { ...@@ -401,9 +422,11 @@ public class RabbitAutoConfigurationTests {
.withPropertyValues("spring.rabbitmq.listener.type:direct", .withPropertyValues("spring.rabbitmq.listener.type:direct",
"spring.rabbitmq.listener.simple.concurrency:5", "spring.rabbitmq.listener.simple.concurrency:5",
"spring.rabbitmq.listener.simple.maxConcurrency:10", "spring.rabbitmq.listener.simple.maxConcurrency:10",
"spring.rabbitmq.listener.simple.prefetch:40").run((context) -> { "spring.rabbitmq.listener.simple.prefetch:40")
.run((context) -> {
SimpleRabbitListenerContainerFactoryConfigurer configurer = context SimpleRabbitListenerContainerFactoryConfigurer configurer = context
.getBean(SimpleRabbitListenerContainerFactoryConfigurer.class); .getBean(
SimpleRabbitListenerContainerFactoryConfigurer.class);
SimpleRabbitListenerContainerFactory factory = mock( SimpleRabbitListenerContainerFactory factory = mock(
SimpleRabbitListenerContainerFactory.class); SimpleRabbitListenerContainerFactory.class);
configurer.configure(factory, mock(ConnectionFactory.class)); configurer.configure(factory, mock(ConnectionFactory.class));
...@@ -418,9 +441,11 @@ public class RabbitAutoConfigurationTests { ...@@ -418,9 +441,11 @@ public class RabbitAutoConfigurationTests {
this.contextRunner.withUserConfiguration(TestConfiguration.class) this.contextRunner.withUserConfiguration(TestConfiguration.class)
.withPropertyValues("spring.rabbitmq.listener.type:simple", .withPropertyValues("spring.rabbitmq.listener.type:simple",
"spring.rabbitmq.listener.direct.consumers-per-queue:5", "spring.rabbitmq.listener.direct.consumers-per-queue:5",
"spring.rabbitmq.listener.direct.prefetch:40").run((context) -> { "spring.rabbitmq.listener.direct.prefetch:40")
.run((context) -> {
DirectRabbitListenerContainerFactoryConfigurer configurer = context DirectRabbitListenerContainerFactoryConfigurer configurer = context
.getBean(DirectRabbitListenerContainerFactoryConfigurer.class); .getBean(
DirectRabbitListenerContainerFactoryConfigurer.class);
DirectRabbitListenerContainerFactory factory = mock( DirectRabbitListenerContainerFactory factory = mock(
DirectRabbitListenerContainerFactory.class); DirectRabbitListenerContainerFactory.class);
configurer.configure(factory, mock(ConnectionFactory.class)); configurer.configure(factory, mock(ConnectionFactory.class));
...@@ -468,8 +493,8 @@ public class RabbitAutoConfigurationTests { ...@@ -468,8 +493,8 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void enableRabbitAutomatically() throws Exception { public void enableRabbitAutomatically() throws Exception {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(NoEnableRabbitConfiguration.class)
NoEnableRabbitConfiguration.class).run((context) -> { .run((context) -> {
assertThat(context).hasBean( assertThat(context).hasBean(
RabbitListenerConfigUtils.RABBIT_LISTENER_ANNOTATION_PROCESSOR_BEAN_NAME); RabbitListenerConfigUtils.RABBIT_LISTENER_ANNOTATION_PROCESSOR_BEAN_NAME);
assertThat(context).hasBean( assertThat(context).hasBean(
...@@ -482,7 +507,8 @@ public class RabbitAutoConfigurationTests { ...@@ -482,7 +507,8 @@ public class RabbitAutoConfigurationTests {
this.contextRunner.withUserConfiguration(TestConfiguration.class) this.contextRunner.withUserConfiguration(TestConfiguration.class)
.withPropertyValues("spring.rabbitmq.requestedHeartbeat:20") .withPropertyValues("spring.rabbitmq.requestedHeartbeat:20")
.run((context) -> { .run((context) -> {
com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory = getTargetConnectionFactory(context); com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory = getTargetConnectionFactory(
context);
assertThat(rabbitConnectionFactory.getRequestedHeartbeat()) assertThat(rabbitConnectionFactory.getRequestedHeartbeat())
.isEqualTo(20); .isEqualTo(20);
}); });
...@@ -490,9 +516,10 @@ public class RabbitAutoConfigurationTests { ...@@ -490,9 +516,10 @@ public class RabbitAutoConfigurationTests {
@Test @Test
public void noSslByDefault() { public void noSslByDefault() {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(TestConfiguration.class)
TestConfiguration.class).run((context) -> { .run((context) -> {
com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory = getTargetConnectionFactory(context); com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory = getTargetConnectionFactory(
context);
assertThat(rabbitConnectionFactory.getSocketFactory()) assertThat(rabbitConnectionFactory.getSocketFactory())
.as("Must use default SocketFactory") .as("Must use default SocketFactory")
.isEqualTo(SocketFactory.getDefault()); .isEqualTo(SocketFactory.getDefault());
...@@ -503,9 +530,11 @@ public class RabbitAutoConfigurationTests { ...@@ -503,9 +530,11 @@ public class RabbitAutoConfigurationTests {
public void enableSsl() { public void enableSsl() {
this.contextRunner.withUserConfiguration(TestConfiguration.class) this.contextRunner.withUserConfiguration(TestConfiguration.class)
.withPropertyValues("spring.rabbitmq.ssl.enabled:true").run((context) -> { .withPropertyValues("spring.rabbitmq.ssl.enabled:true").run((context) -> {
com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory = getTargetConnectionFactory(context); com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory = getTargetConnectionFactory(
context);
assertThat(rabbitConnectionFactory.getSocketFactory()) assertThat(rabbitConnectionFactory.getSocketFactory())
.as("SocketFactory must use SSL").isInstanceOf(SSLSocketFactory.class); .as("SocketFactory must use SSL")
.isInstanceOf(SSLSocketFactory.class);
}); });
} }
...@@ -517,10 +546,12 @@ public class RabbitAutoConfigurationTests { ...@@ -517,10 +546,12 @@ public class RabbitAutoConfigurationTests {
"spring.rabbitmq.ssl.keyStore=foo", "spring.rabbitmq.ssl.keyStore=foo",
"spring.rabbitmq.ssl.keyStorePassword=secret", "spring.rabbitmq.ssl.keyStorePassword=secret",
"spring.rabbitmq.ssl.trustStore=bar", "spring.rabbitmq.ssl.trustStore=bar",
"spring.rabbitmq.ssl.trustStorePassword=secret").run((context) -> { "spring.rabbitmq.ssl.trustStorePassword=secret")
.run((context) -> {
assertThat(context).hasFailed(); assertThat(context).hasFailed();
assertThat(context).getFailure().hasMessageContaining("foo"); assertThat(context).getFailure().hasMessageContaining("foo");
assertThat(context).getFailure().hasMessageContaining("does not exist"); assertThat(context).getFailure()
.hasMessageContaining("does not exist");
}); });
} }
......
...@@ -86,8 +86,8 @@ public class BatchAutoConfigurationTests { ...@@ -86,8 +86,8 @@ public class BatchAutoConfigurationTests {
EmbeddedDataSourceConfiguration.class).run((context) -> { EmbeddedDataSourceConfiguration.class).run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
assertThat(context).hasSingleBean(JobExplorer.class); assertThat(context).hasSingleBean(JobExplorer.class);
assertThat(context.getBean(BatchProperties.class) assertThat(context.getBean(BatchProperties.class).getInitializer()
.getInitializer().isEnabled()).isTrue(); .isEnabled()).isTrue();
assertThat(new JdbcTemplate(context.getBean(DataSource.class)) assertThat(new JdbcTemplate(context.getBean(DataSource.class))
.queryForList("select * from BATCH_JOB_EXECUTION")).isEmpty(); .queryForList("select * from BATCH_JOB_EXECUTION")).isEmpty();
}); });
...@@ -95,8 +95,8 @@ public class BatchAutoConfigurationTests { ...@@ -95,8 +95,8 @@ public class BatchAutoConfigurationTests {
@Test @Test
public void testNoDatabase() throws Exception { public void testNoDatabase() throws Exception {
this.contextRunner.withUserConfiguration( this.contextRunner.withUserConfiguration(TestCustomConfiguration.class)
TestCustomConfiguration.class).run((context) -> { .run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
JobExplorer explorer = context.getBean(JobExplorer.class); JobExplorer explorer = context.getBean(JobExplorer.class);
assertThat(explorer.getJobInstances("job", 0, 100)).isEmpty(); assertThat(explorer.getJobInstances("job", 0, 100)).isEmpty();
...@@ -118,17 +118,18 @@ public class BatchAutoConfigurationTests { ...@@ -118,17 +118,18 @@ public class BatchAutoConfigurationTests {
EmbeddedDataSourceConfiguration.class).run((context) -> { EmbeddedDataSourceConfiguration.class).run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
context.getBean(JobLauncherCommandLineRunner.class).run(); context.getBean(JobLauncherCommandLineRunner.class).run();
assertThat(context.getBean(JobRepository.class).getLastJobExecution("job", assertThat(context.getBean(JobRepository.class)
new JobParameters())).isNotNull(); .getLastJobExecution("job", new JobParameters())).isNotNull();
}); });
} }
@Test @Test
public void testDefinesAndLaunchesNamedJob() throws Exception { public void testDefinesAndLaunchesNamedJob() throws Exception {
this.contextRunner.withUserConfiguration( this.contextRunner
NamedJobConfigurationWithRegisteredJob.class, .withUserConfiguration(NamedJobConfigurationWithRegisteredJob.class,
EmbeddedDataSourceConfiguration.class).withPropertyValues( EmbeddedDataSourceConfiguration.class)
"spring.batch.job.names:discreteRegisteredJob").run((context) -> { .withPropertyValues("spring.batch.job.names:discreteRegisteredJob")
.run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
context.getBean(JobLauncherCommandLineRunner.class).run(); context.getBean(JobLauncherCommandLineRunner.class).run();
assertThat(context.getBean(JobRepository.class).getLastJobExecution( assertThat(context.getBean(JobRepository.class).getLastJobExecution(
...@@ -138,21 +139,25 @@ public class BatchAutoConfigurationTests { ...@@ -138,21 +139,25 @@ public class BatchAutoConfigurationTests {
@Test @Test
public void testDefinesAndLaunchesLocalJob() throws Exception { public void testDefinesAndLaunchesLocalJob() throws Exception {
this.contextRunner.withUserConfiguration(NamedJobConfigurationWithLocalJob.class, this.contextRunner
EmbeddedDataSourceConfiguration.class).withPropertyValues( .withUserConfiguration(NamedJobConfigurationWithLocalJob.class,
"spring.batch.job.names:discreteLocalJob").run((context) -> { EmbeddedDataSourceConfiguration.class)
.withPropertyValues("spring.batch.job.names:discreteLocalJob")
.run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
context.getBean(JobLauncherCommandLineRunner.class).run(); context.getBean(JobLauncherCommandLineRunner.class).run();
assertThat(context.getBean(JobRepository.class).getLastJobExecution( assertThat(context.getBean(JobRepository.class)
"discreteLocalJob", new JobParameters())).isNotNull(); .getLastJobExecution("discreteLocalJob", new JobParameters()))
.isNotNull();
}); });
} }
@Test @Test
public void testDisableLaunchesJob() throws Exception { public void testDisableLaunchesJob() throws Exception {
this.contextRunner.withUserConfiguration(JobConfiguration.class, this.contextRunner
EmbeddedDataSourceConfiguration.class).withPropertyValues( .withUserConfiguration(JobConfiguration.class,
"spring.batch.job.enabled:false").run((context) -> { EmbeddedDataSourceConfiguration.class)
.withPropertyValues("spring.batch.job.enabled:false").run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
assertThat(context).doesNotHaveBean(CommandLineRunner.class); assertThat(context).doesNotHaveBean(CommandLineRunner.class);
}); });
...@@ -160,16 +165,18 @@ public class BatchAutoConfigurationTests { ...@@ -160,16 +165,18 @@ public class BatchAutoConfigurationTests {
@Test @Test
public void testDisableSchemaLoader() throws Exception { public void testDisableSchemaLoader() throws Exception {
this.contextRunner.withUserConfiguration(TestConfiguration.class, this.contextRunner
EmbeddedDataSourceConfiguration.class).withPropertyValues( .withUserConfiguration(TestConfiguration.class,
"spring.datasource.generate-unique-name=true", EmbeddedDataSourceConfiguration.class)
"spring.batch.initializer.enabled:false").run((context) -> { .withPropertyValues("spring.datasource.generate-unique-name=true",
"spring.batch.initializer.enabled:false")
.run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
assertThat(context.getBean(BatchProperties.class).getInitializer() assertThat(context.getBean(BatchProperties.class).getInitializer()
.isEnabled()).isFalse(); .isEnabled()).isFalse();
this.expected.expect(BadSqlGrammarException.class); this.expected.expect(BadSqlGrammarException.class);
new JdbcTemplate(context.getBean(DataSource.class)).queryForList( new JdbcTemplate(context.getBean(DataSource.class))
"select * from BATCH_JOB_EXECUTION"); .queryForList("select * from BATCH_JOB_EXECUTION");
}); });
} }
...@@ -180,65 +187,74 @@ public class BatchAutoConfigurationTests { ...@@ -180,65 +187,74 @@ public class BatchAutoConfigurationTests {
HibernateJpaAutoConfiguration.class).run((context) -> { HibernateJpaAutoConfiguration.class).run((context) -> {
PlatformTransactionManager transactionManager = context PlatformTransactionManager transactionManager = context
.getBean(PlatformTransactionManager.class); .getBean(PlatformTransactionManager.class);
// It's a lazy proxy, but it does render its target if you ask for toString(): // It's a lazy proxy, but it does render its target if you ask for
assertThat(transactionManager.toString().contains("JpaTransactionManager")) // toString():
.isTrue(); assertThat(transactionManager.toString()
.contains("JpaTransactionManager")).isTrue();
assertThat(context).hasSingleBean(EntityManagerFactory.class); assertThat(context).hasSingleBean(EntityManagerFactory.class);
// Ensure the JobRepository can be used (no problem with isolation level) // Ensure the JobRepository can be used (no problem with isolation
assertThat(context.getBean(JobRepository.class).getLastJobExecution("job", // level)
new JobParameters())).isNull(); assertThat(context.getBean(JobRepository.class)
.getLastJobExecution("job", new JobParameters())).isNull();
}); });
} }
@Test @Test
public void testRenamePrefix() throws Exception { public void testRenamePrefix() throws Exception {
this.contextRunner.withUserConfiguration(TestConfiguration.class, this.contextRunner
.withUserConfiguration(TestConfiguration.class,
EmbeddedDataSourceConfiguration.class, EmbeddedDataSourceConfiguration.class,
HibernateJpaAutoConfiguration.class).withPropertyValues( HibernateJpaAutoConfiguration.class)
"spring.datasource.generate-unique-name=true", .withPropertyValues("spring.datasource.generate-unique-name=true",
"spring.batch.schema:classpath:batch/custom-schema-hsql.sql", "spring.batch.schema:classpath:batch/custom-schema-hsql.sql",
"spring.batch.tablePrefix:PREFIX_").run((context) -> { "spring.batch.tablePrefix:PREFIX_")
.run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
assertThat(context.getBean(BatchProperties.class).getInitializer() assertThat(context.getBean(BatchProperties.class).getInitializer()
.isEnabled()).isTrue(); .isEnabled()).isTrue();
assertThat(new JdbcTemplate(context.getBean(DataSource.class)) assertThat(new JdbcTemplate(context.getBean(DataSource.class))
.queryForList("select * from PREFIX_JOB_EXECUTION")).isEmpty(); .queryForList("select * from PREFIX_JOB_EXECUTION"))
.isEmpty();
JobExplorer jobExplorer = context.getBean(JobExplorer.class); JobExplorer jobExplorer = context.getBean(JobExplorer.class);
assertThat(jobExplorer.findRunningJobExecutions("test")).isEmpty(); assertThat(jobExplorer.findRunningJobExecutions("test")).isEmpty();
JobRepository jobRepository = context.getBean(JobRepository.class); JobRepository jobRepository = context.getBean(JobRepository.class);
assertThat(jobRepository.getLastJobExecution("test", new JobParameters())) assertThat(jobRepository.getLastJobExecution("test",
.isNull(); new JobParameters())).isNull();
}); });
} }
@Test @Test
public void testCustomTablePrefixWithDefaultSchemaDisablesInitializer() public void testCustomTablePrefixWithDefaultSchemaDisablesInitializer()
throws Exception { throws Exception {
this.contextRunner.withUserConfiguration(TestConfiguration.class, this.contextRunner
.withUserConfiguration(TestConfiguration.class,
EmbeddedDataSourceConfiguration.class, EmbeddedDataSourceConfiguration.class,
HibernateJpaAutoConfiguration.class).withPropertyValues( HibernateJpaAutoConfiguration.class)
"spring.datasource.generate-unique-name=true", .withPropertyValues("spring.datasource.generate-unique-name=true",
"spring.batch.tablePrefix:PREFIX_").run((context) -> { "spring.batch.tablePrefix:PREFIX_")
.run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
assertThat(context.getBean(BatchProperties.class).getInitializer() assertThat(context.getBean(BatchProperties.class).getInitializer()
.isEnabled()).isFalse(); .isEnabled()).isFalse();
this.expected.expect(BadSqlGrammarException.class); this.expected.expect(BadSqlGrammarException.class);
new JdbcTemplate(context.getBean(DataSource.class)).queryForList( new JdbcTemplate(context.getBean(DataSource.class))
"select * from BATCH_JOB_EXECUTION"); .queryForList("select * from BATCH_JOB_EXECUTION");
}); });
} }
@Test @Test
public void testCustomizeJpaTransactionManagerUsingProperties() throws Exception { public void testCustomizeJpaTransactionManagerUsingProperties() throws Exception {
this.contextRunner.withUserConfiguration(TestConfiguration.class, this.contextRunner
.withUserConfiguration(TestConfiguration.class,
EmbeddedDataSourceConfiguration.class, EmbeddedDataSourceConfiguration.class,
HibernateJpaAutoConfiguration.class).withPropertyValues( HibernateJpaAutoConfiguration.class)
"spring.transaction.default-timeout:30", .withPropertyValues("spring.transaction.default-timeout:30",
"spring.transaction.rollback-on-commit-failure:true").run((context) -> { "spring.transaction.rollback-on-commit-failure:true")
.run((context) -> {
assertThat(context).hasSingleBean(BatchConfigurer.class); assertThat(context).hasSingleBean(BatchConfigurer.class);
JpaTransactionManager transactionManager = JpaTransactionManager.class.cast( JpaTransactionManager transactionManager = JpaTransactionManager.class
context.getBean(BatchConfigurer.class).getTransactionManager()); .cast(context.getBean(BatchConfigurer.class)
.getTransactionManager());
assertThat(transactionManager.getDefaultTimeout()).isEqualTo(30); assertThat(transactionManager.getDefaultTimeout()).isEqualTo(30);
assertThat(transactionManager.isRollbackOnCommitFailure()).isTrue(); assertThat(transactionManager.isRollbackOnCommitFailure()).isTrue();
}); });
...@@ -247,13 +263,16 @@ public class BatchAutoConfigurationTests { ...@@ -247,13 +263,16 @@ public class BatchAutoConfigurationTests {
@Test @Test
public void testCustomizeDataSourceTransactionManagerUsingProperties() public void testCustomizeDataSourceTransactionManagerUsingProperties()
throws Exception { throws Exception {
this.contextRunner.withUserConfiguration(TestConfiguration.class, this.contextRunner
EmbeddedDataSourceConfiguration.class).withPropertyValues( .withUserConfiguration(TestConfiguration.class,
"spring.transaction.default-timeout:30", EmbeddedDataSourceConfiguration.class)
"spring.transaction.rollback-on-commit-failure:true").run((context) -> { .withPropertyValues("spring.transaction.default-timeout:30",
"spring.transaction.rollback-on-commit-failure:true")
.run((context) -> {
assertThat(context).hasSingleBean(BatchConfigurer.class); assertThat(context).hasSingleBean(BatchConfigurer.class);
DataSourceTransactionManager transactionManager = DataSourceTransactionManager.class DataSourceTransactionManager transactionManager = DataSourceTransactionManager.class
.cast(context.getBean(BatchConfigurer.class).getTransactionManager()); .cast(context.getBean(BatchConfigurer.class)
.getTransactionManager());
assertThat(transactionManager.getDefaultTimeout()).isEqualTo(30); assertThat(transactionManager.getDefaultTimeout()).isEqualTo(30);
assertThat(transactionManager.isRollbackOnCommitFailure()).isTrue(); assertThat(transactionManager.isRollbackOnCommitFailure()).isTrue();
}); });
......
...@@ -54,42 +54,45 @@ public class BatchAutoConfigurationWithoutJpaTests { ...@@ -54,42 +54,45 @@ 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)
.withPropertyValues("spring.datasource.generate-unique-name=true")
.run((context) -> {
assertThat(context).hasSingleBean(JobLauncher.class); assertThat(context).hasSingleBean(JobLauncher.class);
assertThat(context).hasSingleBean(JobExplorer.class); assertThat(context).hasSingleBean(JobExplorer.class);
assertThat(context).hasSingleBean(JobRepository.class); assertThat(context).hasSingleBean(JobRepository.class);
assertThat(context).hasSingleBean(PlatformTransactionManager.class); assertThat(context).hasSingleBean(PlatformTransactionManager.class);
assertThat(context.getBean(PlatformTransactionManager.class).toString())
.contains("DataSourceTransactionManager");
assertThat( assertThat(
context.getBean(BatchProperties.class).getInitializer().isEnabled()) context.getBean(PlatformTransactionManager.class).toString())
.isTrue(); .contains("DataSourceTransactionManager");
assertThat(context.getBean(BatchProperties.class).getInitializer()
.isEnabled()).isTrue();
assertThat(new JdbcTemplate(context.getBean(DataSource.class)) assertThat(new JdbcTemplate(context.getBean(DataSource.class))
.queryForList("select * from BATCH_JOB_EXECUTION")).isEmpty(); .queryForList("select * from BATCH_JOB_EXECUTION")).isEmpty();
assertThat( assertThat(context.getBean(JobExplorer.class)
context.getBean(JobExplorer.class).findRunningJobExecutions("test")) .findRunningJobExecutions("test")).isEmpty();
.isEmpty(); assertThat(context.getBean(JobRepository.class)
assertThat(context.getBean(JobRepository.class).getLastJobExecution("test", .getLastJobExecution("test", new JobParameters())).isNull();
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)
.withPropertyValues("spring.datasource.generate-unique-name=true",
"spring.batch.schema:classpath:batch/custom-schema-hsql.sql", "spring.batch.schema:classpath:batch/custom-schema-hsql.sql",
"spring.batch.tablePrefix:PREFIX_").run((context) -> { "spring.batch.tablePrefix:PREFIX_")
.run((context) -> {
assertThat(new JdbcTemplate(context.getBean(DataSource.class)) assertThat(new JdbcTemplate(context.getBean(DataSource.class))
.queryForList("select * from PREFIX_JOB_EXECUTION")).isEmpty(); .queryForList("select * from PREFIX_JOB_EXECUTION"))
assertThat(
context.getBean(JobExplorer.class).findRunningJobExecutions("test"))
.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();
}); });
} }
......
...@@ -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,8 +50,9 @@ public class CassandraAutoConfigurationTests { ...@@ -50,8 +50,9 @@ 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")
.run((context) -> {
assertThat(context).hasSingleBean(Cluster.class); assertThat(context).hasSingleBean(Cluster.class);
assertThat(context.getBean(Cluster.class).getClusterName()) assertThat(context.getBean(Cluster.class).getClusterName())
.isEqualTo("testcluster"); .isEqualTo("testcluster");
...@@ -60,8 +61,8 @@ public class CassandraAutoConfigurationTests { ...@@ -60,8 +61,8 @@ public class CassandraAutoConfigurationTests {
@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);
}); });
...@@ -70,7 +71,8 @@ public class CassandraAutoConfigurationTests { ...@@ -70,7 +71,8 @@ public class CassandraAutoConfigurationTests {
@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")
.run((context) -> {
assertThat(context).hasSingleBean(Cluster.class); assertThat(context).hasSingleBean(Cluster.class);
assertThat(context.getBean(Cluster.class).getClusterName()) assertThat(context.getBean(Cluster.class).getClusterName())
.isEqualTo("overridden-name"); .isEqualTo("overridden-name");
...@@ -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")
.run((context) -> {
assertThat(context).hasSingleBean(Cluster.class); assertThat(context).hasSingleBean(Cluster.class);
PoolingOptions poolingOptions = context.getBean(Cluster.class) PoolingOptions poolingOptions = context.getBean(Cluster.class)
.getConfiguration().getPoolingOptions(); .getConfiguration().getPoolingOptions();
assertThat(poolingOptions.getIdleTimeoutSeconds()).isEqualTo(42); assertThat(poolingOptions.getIdleTimeoutSeconds()).isEqualTo(42);
assertThat(poolingOptions.getPoolTimeoutMillis()).isEqualTo(52); assertThat(poolingOptions.getPoolTimeoutMillis()).isEqualTo(52);
assertThat(poolingOptions.getHeartbeatIntervalSeconds()).isEqualTo(62); assertThat(poolingOptions.getHeartbeatIntervalSeconds())
.isEqualTo(62);
assertThat(poolingOptions.getMaxQueueSize()).isEqualTo(72); 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,20 +102,21 @@ public class ConditionalOnBeanTests { ...@@ -101,20 +102,21 @@ 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")
...@@ -130,14 +132,11 @@ public class ConditionalOnBeanTests { ...@@ -130,14 +132,11 @@ public class ConditionalOnBeanTests {
}); });
} }
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)
.run((context) -> {
assertThat(context).doesNotHaveBean("bar"); assertThat(context).doesNotHaveBean("bar");
assertThat(context).hasBean("foo"); assertThat(context).hasBean("foo");
assertThat(context.getBean("foo")).isEqualTo("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 {
static void throwUnchecked(Throwable e) {
AnyThrow.throwAny(e);
}
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private static <E extends Throwable> void throwAny(Throwable e) throws E { private <E extends Throwable> void rethrow(Throwable e) throws E {
throw (E) 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