Spring Boot测试后数据库未清理问题排查求助

一、排查findDatabaseTableNames无法获取H2表名的原因 #

• 检查H2元数据查询逻辑
  H2的系统表查询需要明确指定schema和表类型，默认schema是PUBLIC，如果你的查询语句没做过滤，可能拿不到业务表。正确的查询应该是：

  SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES 
  WHERE TABLE_SCHEMA = 'PUBLIC' AND TABLE_TYPE = 'TABLE'
  

  之前的代码如果省略了TABLE_SCHEMA或TABLE_TYPE的过滤，会返回系统表或空结果。

• 确认连接获取的时机与来源

  • 如果你是手动新建JDBC连接而不是用Spring注入的DataSource，很可能连接到了另一个H2内存实例（H2内存库的实例由JDBC URL决定，比如jdbc:h2:mem:testdb，不同URL对应不同实例），自然看不到业务表。必须通过Spring容器获取DataSource来获取连接。
  • JUnit5 Extension的执行时机要注意：如果在Spring上下文初始化前就调用findDatabaseTableNames，此时JPA还没完成自动建表，表自然不存在。要确保在测试方法执行后、Spring上下文完全就绪的状态下执行表名查询。

二、更可靠的数据清理替代方案 #

自己写Extension容易踩坑，推荐用Spring Boot测试生态自带的方案：

• 用@Transactional自动回滚
  在测试类或方法上添加@Transactional，Spring会在测试结束后自动回滚事务，完全不会留下数据残留。注意：如果测试中手动提交了事务（比如调用transactionManager.commit()），这个机制会失效。示例：

  @SpringBootTest
  @Transactional
  class MemberTest {
      @Test
      void registMemberTestIfOauthPlatformHasInvalidValue() {
          // 第一次无效参数测试
          assertThrows(IllegalArgumentException.class, () -> memberService.regist(...));
          // 第二次测试，事务已回滚，无数据残留
          assertThrows(IllegalArgumentException.class, () -> memberService.regist(...));
      }
  }
  

• 用@Sql执行清理脚本
  如果测试涉及多事务、异步操作，无法用事务回滚，可以在测试后执行清理SQL：

  @SpringBootTest
  @Sql(executionPhase = Sql.ExecutionPhase.AFTER_TEST_METHOD, scripts = "classpath:cleanup.sql")
  class MemberTest {
      // 测试方法
  }
  

  cleanup.sql内容示例（H2语法）：

  SET REFERENTIAL_INTEGRITY FALSE;
  TRUNCATE TABLE member RESTART IDENTITY;
  TRUNCATE TABLE oauth_info RESTART IDENTITY;
  SET REFERENTIAL_INTEGRITY TRUE;
  

  先禁用外键约束避免删除顺序问题，清理后再恢复。

• 自定义Spring TestExecutionListener
  比手写JUnit Extension更贴合Spring生态，能直接利用容器里的DataSource：

  public class DataCleanupListener extends AbstractTestExecutionListener {
      @Autowired
      private DataSource dataSource;
  
      @Override
      public void afterTestMethod(TestContext testContext) throws Exception {
          testContext.getApplicationContext().getAutowireCapableBeanFactory().autowireBean(this);
          try (Connection conn = dataSource.getConnection()) {
              DatabaseMetaData metaData = conn.getMetaData();
              try (ResultSet rs = metaData.getTables(null, "PUBLIC", "%", new String[]{"TABLE"})) {
                  List<String> tableNames = new ArrayList<>();
                  while (rs.next()) {
                      tableNames.add(rs.getString("TABLE_NAME"));
                  }
                  try (Statement stmt = conn.createStatement()) {
                      stmt.execute("SET REFERENTIAL_INTEGRITY FALSE");
                      for (String table : tableNames) {
                          stmt.execute(String.format("TRUNCATE TABLE %s RESTART IDENTITY", table));
                      }
                      stmt.execute("SET REFERENTIAL_INTEGRITY TRUE");
                  }
              }
          }
      }
  }
  

  测试类上添加：

  @SpringBootTest
  @TestExecutionListeners(listeners = DataCleanupListener.class, mergeMode = MergeMode.MERGE_WITH_DEFAULTS)
  class MemberTest {
      // ...
  }
  

三、测试用例本身的潜在问题 #

• 检查第二个assertThrows的触发条件：确认业务逻辑中，第二次测试的参数确实会抛出异常，有没有可能是第一次残留的数据改变了业务判断逻辑？比如某个唯一约束被占用，导致第二次测试的错误类型和预期不符。
• 确保测试独立性：每个测试方法应该不依赖其他测试的执行结果，即使没有全局数据清理，单个测试方法也应该在内部完成必要的初始化/清理，避免互相干扰。

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