java - 具有多个 DataSource bean 的 AutoConfigureTestDatabase

Question

我的应用程序中有两个数据源，它们在连接到实时环境中的数据库时按预期工作。但是，在为此应用程序编写单元测试时，我的 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 实例并运行其迁移，但主数据源及其 EntityManager 在创建时失败。

Answer 1

我通过执行以下操作解决了该问题:

首先，我更改了 AutoConfigureTestDatabase 注释以不替换任何数据源:

@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。通过确保数据源未被替换，并确保连接 URL 正确，测试按预期进行。