具有多个 DataSource bean 的 AutoConfigureTestDatabase答案

【问题标题】：AutoConfigureTestDatabase with multiple DataSource beans具有多个 DataSource bean 的 AutoConfigureTestDatabase
【发布时间】：2019-10-25 13:48:34
【问题描述】：

我的应用程序中有两个 DataSource，它们在连接到实时环境中的数据库时可以正常工作。但是，在为此应用程序编写单元测试时，我的 bean 定义遇到了问题。

这是我的主要数据源配置：

@Configuration
@EnableJpaRepositories(
        basePackages = "foo.bar.repository.primary",
        entityManagerFactoryRef = "primaryEntityManager",
        transactionManagerRef = "primaryTransactionManager")
public class PrimaryDataSourceConfig {

    @Bean
    @Primary
    @ConfigurationProperties("primary.datasource")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @Primary
    public LocalContainerEntityManagerFactoryBean primaryEntityManager() {

        LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
        em.setDataSource(primaryDataSource());
        em.setPackagesToScan("foo.bar.domain.entity");

        HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
        em.setJpaVendorAdapter(vendorAdapter);

        Map<String, String> properties = new HashMap<>();
        properties.put("hibernate.implicit_naming_strategy",
                "org.springframework.boot.orm.jpa.hibernate.SpringImplicitNamingStrategy");
        properties.put("hibernate.physical_naming_strategy",
                "org.springframework.boot.orm.jpa.hibernate.SpringPhysicalNamingStrategy");
        em.setJpaPropertyMap(properties);

        return em;
    }

    @Bean
    @Primary
    public PlatformTransactionManager primaryTransactionManager() {

        JpaTransactionManager transactionManager = new JpaTransactionManager();
        transactionManager.setEntityManagerFactory(primaryEntityManager().getObject());
        return transactionManager;
    }
}

我有一个测试，我想做的就是启动整个应用程序上下文，看起来像这样。

@AutoConfigureTestDatabase
@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class ErpApplicationTest {

    @Test
    public void test() {
        // Application started
    }
}

但是，当我运行此测试时，我收到以下错误：

org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'primaryEntityManager' defined in class path resource [foo/bar/primaryDataSourceConfig.class]: Bean instantiation via factory method failed; nested exception is org.springframework.beans.BeanInstantiationException: Failed to instantiate [org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean]: Factory method 'primaryEntityManager' threw exception; nested exception is java.lang.IllegalArgumentException: No visible constructors in class org.springframework.boot.test.autoconfigure.jdbc.TestDatabaseAutoConfiguration$EmbeddedDataSourceFactoryBean

是什么导致了这个错误，我可以做些什么来纠正它？ Flyway 能够连接到注释生成的嵌入式 H2 实例并运行其迁移，但主 DataSource 及其 EntityManager 在创建时失败。

【问题讨论】：

标签： java spring-boot datasource h2

【解决方案1】：

我通过以下操作解决了这个问题：

首先我将 AutoConfigureTestDatabase-annotation 更改为不替换任何数据源：

@AutoConfigureTestDatabase(
        replace = AutoConfigureTestDatabase.Replace.NONE)

我现在遇到了数据库超时的错误。由于 AutoConfig 默认为内存中的 h2，我认为以前的数据库连接设置不正确。我将 application.properties 文件更改为包含以下内容：

primary.datasource.driver-class-name=org.h2.Driver
primary.datasource.jdbc-url=jdbc:h2:~;MODE=MYSQL
primary.datasource.username=
primary.datasource.password=

secondary.datasource.driver-class-name=org.h2.Driver
secondary.datasource.jdbc-url=jdbc:h2:~;MODE=MYSQL
secondary.datasource.username=
secondary.datasource.password=

可能发生此错误是因为自动配置无法正确替换我的自定义数据源的 entityManager。通过确保未替换 DataSource 并确保连接 url 正确，测试按预期进行。

【讨论】：