测试中使用EntityManager批量清空所有表数据的解决办法？

问题根源 #

你之前的代码错误在于用executeUpdate()执行了SELECT查询——executeUpdate()仅支持DDL/DML语句（比如TRUNCATE、INSERT），不能用于查询数据。要批量截断表，得先查出所有表名，再逐个执行TRUNCATE。

修正后的代码 #

@RunWith(SpringJUnit4ClassRunner.class)
@TestPropertySource(...)
@ContextConfiguration(...)
public class MyTest {
    @PersistenceContext
    private EntityManager em;
    
    @After
    public void after() {
        // 替换为你的测试数据库名称，避免误操作系统表
        String testDbName = "your_test_db";
        
        // 1. 查询当前数据库下所有业务表名
        Query tableQuery = em.createNativeQuery(
            "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ?"
        );
        tableQuery.setParameter(1, testDbName);
        List<String> tableNames = tableQuery.getResultList();
        
        // 2. 临时禁用外键检查（InnoDB表必备，否则有外键关联的表无法截断）
        em.createNativeQuery("SET FOREIGN_KEY_CHECKS = 0").executeUpdate();
        
        // 3. 逐个截断表
        for (String tableName : tableNames) {
            // 跳过不需要截断的系统/元数据表，比如迁移表、序列表
            if (tableName.startsWith("flyway_") || tableName.equals("hibernate_sequence")) {
                continue;
            }
            em.createNativeQuery("TRUNCATE TABLE " + tableName).executeUpdate();
        }
        
        // 4. 恢复外键检查
        em.createNativeQuery("SET FOREIGN_KEY_CHECKS = 1").executeUpdate();
    }
}

关键注意事项 #

• 指定数据库名：必须通过TABLE_SCHEMA过滤，不然会遍历服务器上所有数据库的表，造成误操作。
• 跳过系统表：根据你的环境跳过元数据表（比如Hibernate序列、Flyway迁移日志），避免破坏测试环境的基础配置。
• 外键处理：InnoDB表有外键关联时，必须先禁用外键检查，否则截断会失败，执行完记得恢复。
• 移除@Transactional：TRUNCATE是DDL语句，执行时会自动提交事务，保留@Transactional可能导致事务行为异常，且你提到当前测试套件不支持全事务化测试。

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