diff --git a/spring-boot-test-autoconfigure/src/main/java/org/springframework/boot/test/autoconfigure/orm/jpa/TestDatabaseAutoConfiguration.java b/spring-boot-test-autoconfigure/src/main/java/org/springframework/boot/test/autoconfigure/orm/jpa/TestDatabaseAutoConfiguration.java index 33df6947e71..3e018a5c06f 100644 --- a/spring-boot-test-autoconfigure/src/main/java/org/springframework/boot/test/autoconfigure/orm/jpa/TestDatabaseAutoConfiguration.java +++ b/spring-boot-test-autoconfigure/src/main/java/org/springframework/boot/test/autoconfigure/orm/jpa/TestDatabaseAutoConfiguration.java @@ -101,15 +101,20 @@ public class TestDatabaseAutoConfiguration { ConfigurableListableBeanFactory beanFactory) { BeanDefinitionHolder holder = getDataSourceBeanDefinition(beanFactory); if (holder != null) { - logger.info("Replacing '" + holder.getBeanName() - + "' DataSource bean with embedded version"); - registry.registerBeanDefinition(holder.getBeanName(), - createEmbeddedBeanDefinition()); + String beanName = holder.getBeanName(); + boolean primary = holder.getBeanDefinition().isPrimary(); + logger.info("Replacing '" + beanName + "' DataSource bean with " + + (primary ? "primary " : "") + "embedded version"); + registry.registerBeanDefinition(beanName, + createEmbeddedBeanDefinition(primary)); } } - private BeanDefinition createEmbeddedBeanDefinition() { - return new RootBeanDefinition(EmbeddedDataSourceFactoryBean.class); + private BeanDefinition createEmbeddedBeanDefinition(boolean primary) { + BeanDefinition beanDefinition = new RootBeanDefinition( + EmbeddedDataSourceFactoryBean.class); + beanDefinition.setPrimary(primary); + return beanDefinition; } private BeanDefinitionHolder getDataSourceBeanDefinition( diff --git a/spring-boot-test-autoconfigure/src/test/java/org/springframework/boot/test/autoconfigure/orm/jpa/AutoConfigureTestDatabaseWithMultipleDatasourcesIntegrationTests.java b/spring-boot-test-autoconfigure/src/test/java/org/springframework/boot/test/autoconfigure/orm/jpa/AutoConfigureTestDatabaseWithMultipleDatasourcesIntegrationTests.java new file mode 100644 index 00000000000..5428b5da9c3 --- /dev/null +++ b/spring-boot-test-autoconfigure/src/test/java/org/springframework/boot/test/autoconfigure/orm/jpa/AutoConfigureTestDatabaseWithMultipleDatasourcesIntegrationTests.java @@ -0,0 +1,92 @@ +/* + * Copyright 2012-2016 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.boot.test.autoconfigure.orm.jpa; + +import javax.sql.DataSource; + +import org.junit.Test; +import org.junit.runner.RunWith; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.autoconfigure.EnableAutoConfiguration; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.context.annotation.Primary; +import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder; +import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType; +import org.springframework.test.context.junit4.SpringRunner; + +import static org.assertj.core.api.Assertions.assertThat; + +/** + * Integration tests for {@link AutoConfigureTestDatabase} when there are multiple + * datasources. + * + * @author Greg Potter + */ +@RunWith(SpringRunner.class) +@DataJpaTest +@AutoConfigureTestDatabase +public class AutoConfigureTestDatabaseWithMultipleDatasourcesIntegrationTests { + + @Autowired + private TestEntityManager entities; + + @Autowired + private ExampleRepository repository; + + @Autowired + private DataSource dataSource; + + @Test + public void testRepository() throws Exception { + this.entities.persist(new ExampleEntity("boot", "124")); + this.entities.flush(); + ExampleEntity found = this.repository.findByReference("124"); + assertThat(found.getName()).isEqualTo("boot"); + } + + @Test + public void replacesDefinedDataSourceWithExplicit() throws Exception { + // Look that the datasource is replaced with an H2 DB. + String product = this.dataSource.getConnection().getMetaData() + .getDatabaseProductName(); + assertThat(product).startsWith("H2"); + } + + @Configuration + @EnableAutoConfiguration + static class Config { + + @Bean + @Primary + public DataSource dataSource() { + EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder() + .setType(EmbeddedDatabaseType.HSQL); + return builder.build(); + } + + @Bean + public DataSource secondaryDataSource() { + EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder() + .setType(EmbeddedDatabaseType.HSQL); + return builder.build(); + } + + } + +}