Spring启动测试 - 使用2个数据库时EntityManager不会保留

时间:2017-09-18 08:08:14

标签: java spring hibernate spring-boot spring-boot-test

我有控制器的Spring Boot测试:

@RunWith(SpringRunner.class)
@SpringBootTest
@AutoConfigureMockMvc
public class ExampleTest {

    @Autowired
    private MockMvc mockMvc;

    @Autowired
    private ExampleDAO;

    @Test
    public void someTest(){

        exampleDAO.save(someEntity);

        // call to endpoint
    }
} 

我有两个数据库配置:

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(entityManagerFactoryRef = "primaryManagerFactory", basePackages = { "com.example.repository.primary" })
public class PrimaryDbConfig {

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(entityManagerFactoryRef = "anotherManagerFactory", basePackages = { "com.example.repository.another" })
public class AnotherDbConfig {

测试属性片段:

spring.primary.datasource.url=jdbc:h2:mem:primary;DB_CLOSE_ON_EXIT=FALSE
spring.primary.datasource.username=sa
spring.primary.datasource.password=
spring.primary.datasource.driverClassName=org.h2.Driver

spring.another.datasource.url=jdbc:h2:mem:another;DB_CLOSE_ON_EXIT=FALSE
spring.another.datasource.username=sa
spring.another.datasource.password=
spring.another.datasource.driverClassName=org.h2.Driver

spring.jpa.properties.hibernate.show_sql=true

GenericDAOImpl片段:

@PersistenceContext(unitName = "another-persistence-unit")
protected EntityManager entityManager;

@Override
@Transactional
public void save(T entity) {
    entityManager.persist(entity);
}

当我运行测试时,exampleDAO.save(someEntity)方法不会将任何SQL保存或输出到日志。 exampleDAO不是主要数据源。

如果我改为:

@Override
@Transactional
public void save(T entity) {
    entityManager.unwrap(Session.class).save(entity);
}

效果很好,可以保留数据。但是,与主数据源DAO相同的方法会使实体持续EntityManager

为什么用Session保存实体而不用EntityManager保存实体?

1 个答案:

答案 0 :(得分:5)

如果使用两个DB,应创建单独的TransactionManager-s并指定

@Override
@Transactional("transactionManagerName")
public void save(T entity) {
   entityManager.persist(entity);
}