使用 Spring 测试数据库

Question

我需要在数据库（连接，关闭数据库..）错误期间对应用程序行为进行 Spring 测试。 有没有办法从 Spring 单元测试中关闭/杀死或启动 H2 内存数据库？

Answer 1

如果测试数据库连接，这里是旧的，但仍然很好blog on the topic。您也应该能够使用 TestContainers，以便更简单地创建数据库。

如果您只需要在某些确切操作上测试一些失败（例如在存储库保存时），您可以简单地模拟存储库以进行测试运行：

var mockedBean = Mockito.mock(MyRepository.class);
var originalBean = ReflectionTestUtils.getField(articleService, fieldName);
Mockito.when(mockedBean.save(Mockito.any(MyEntity.class))).thenThrow(new RuntimeException("My test exception"));
ReflectionTestUtils.setField(myService, fieldName, mockedBean);
...
// test here
...
// set bean back for other test cases
ReflectionTestUtils.setField(myService, fieldName, originalBean);

Answer 2

使用 h2 测试数据库运行 Junit 测试时，数据库实例将在测试套件开始运行时启动，并在测试完成时停止（假设您已使用 bean 配置数据库）。

如果您使用的是 spring-boot，那么您可以像这样配置您的 h2 测试数据库：

src/test/resources/application-test.yml:

spring:
  datasource:
    driverClassName: org.h2.Driver
    url: jdbc:h2:mem:testdb
    username: sa
    password:

然后像这样配置你的测试：

@RunWith(SpringRunner.class)
@SpringBootTest
@ActiveProfiles(profiles = "test")
public class SignupControllerTest {
    @Autowired
    private WebApplicationContext webApplicationContext;

    private MockMvc mockMvc;

    @Autowired
    private RealityKeeperRepository myrepo;

如果您使用的是常规 spring，那么您可以使用 @Profile("test") 注释配置 DataSource bean：

@Bean
@Profile("test")
public DataSource devDataSource() {
    return new EmbeddedDatabaseBuilder()
        .setType(EmbeddedDatabaseType.HSQL)
        .build();
}

并像这样配置您的测试：

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = {AppConfig.class})
@ActiveProfiles(profiles = "test")
public class MyTest {
    @Autowired
    ...

您的 AppConfig 类应使用 @Configuration 注释，包含上面提到的 devDataSource bean。

Answer 3

是的，如果您有 H2 依赖项并且没有对其进行设置，那么 Spring boot 将为您启动嵌入式 H2 DB。

这篇文章应该有用：http://www.springboottutorial.com/spring-boot-and-h2-in-memory-database