Spring Boot中数据库连接断开时仍可运行JUnit测试的解决方案

这个问题我太有共鸣了——之前做Spring Boot测试时也被真实数据库绑定搞得头大，断个网测试全挂，排除配置又跑不起来。给你几个实战好用的方案，按需挑：

1. 用内存数据库替代真实数据源 #

这是最常用的方案，测试时自动启动一个内存数据库（比如H2），完全不依赖外部真实库。步骤很简单：

• 先给项目加H2依赖，注意设为test scope，只在测试环境生效：

  <!-- Maven 示例 -->
  <dependency>
      <groupId>com.h2database</groupId>
      <artifactId>h2</artifactId>
      <scope>test</scope>
  </dependency>
  

• 新建测试专用的配置文件application-test.yml，配置H2数据源：

  spring:
    datasource:
      url: jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1  # 内存模式，关闭时不销毁数据
      driver-class-name: org.h2.Driver
      username: sa
      password:
    jpa:
      hibernate:
        ddl-auto: create-drop  # 测试时自动建表，结束后销毁
      show-sql: true
  

• 测试类上激活这个测试配置：

  @SpringBootTest
  @ActiveProfiles("test")  // 加载application-test.yml
  public class YourServiceTest {
      // 测试逻辑正常写，用的是内存库，和真实库断开无关
  }
  

2. Mock数据源相关Bean，完全跳过数据库连接 #

如果你的测试根本不需要执行真实的数据库操作（比如只测试业务逻辑，DAO层已经单独测过），直接Mock掉数据源或者相关的持久化Bean就行：

@SpringBootTest
public class YourServiceTest {
    // Mock掉数据源，Spring就不会去初始化真实连接了
    @MockBean
    private DataSource dataSource;

    // 如果用的是JdbcTemplate，也可以直接Mock它
    // @MockBean
    // private JdbcTemplate jdbcTemplate;

    @Autowired
    private YourService yourService;

    @Test
    public void testBusinessLogic() {
        // 这里调用yourService的方法，用到的数据源是Mock的，不会触发真实连接
    }
}

3. 给真实数据源配置加条件，测试时禁用 #

给你的持久化上下文类加条件注解，只有特定配置生效时才加载真实数据源，测试时关闭这个开关：

• 修改你的持久化配置类：

  @Configuration
  // 只有当spring.datasource.enabled为true时才加载，默认开启
  @ConditionalOnProperty(name = "spring.datasource.enabled", havingValue = "true", matchIfMissing = true)
  public class PersistenceContext {
      // 原来的数据源配置代码
  }
  

• 在application-test.yml里禁用真实数据源：

  spring:
    datasource:
      enabled: false
  

• 同时你可以在测试配置里提供一个Mock的数据源，或者继续用内存库，保证Spring上下文能正常初始化。

4. 用切片测试注解，只加载必要的上下文 #

如果是测试DAO层（比如Repository），用@DataJpaTest切片注解，它会自动配置内存数据库，并且只加载JPA相关的上下文，不会加载整个应用的配置，效率更高：

@DataJpaTest
// 强制用内存库替代任何真实数据源配置
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.ANY)
public class YourRepositoryTest {
    @Autowired
    private YourRepository yourRepository;

    @Test
    public void testRepositoryMethod() {
        // 测试DAO方法，完全用内存库，和真实库无关
    }
}

总结一下：需要真实数据库操作逻辑的选内存库；纯业务逻辑测试选Mock；需要灵活切换数据源的选条件化配置；专注DAO层的用切片测试。按需组合就行～

内容的提问来源于stack exchange，提问作者Shanto George