Spring Boot整合Cucumber时如何在每个场景执行后回滚事务重置数据库

问题根因 #

你遇到的@Transactional不生效是因为Spring测试的默认事务生命周期绑定了JUnit的@Test方法执行周期，而Cucumber的场景（Scenario）执行走的是自己的生命周期逻辑，不会触发Spring的自动事务开启、回滚逻辑。@DirtiesContext慢是因为每次执行都会销毁并重建整个Spring上下文，开销极高。 #

方案1：Cucumber Hook绑定事务（优先推荐） #

这个方案完全对齐Cucumber场景的生命周期管理事务，回滚速度和原生Spring测试的事务回滚一致，无额外开销。

操作步骤 #

• 确保你已经引入了版本匹配的cucumber-spring依赖，用于打通Cucumber和Spring的上下文集成
• 编写全局事务Hook类，放到Cucumber配置的glue扫描路径下，示例代码如下：

import io.cucumber.java.Before;
import io.cucumber.java.After;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;

public class TransactionHook {
    private TransactionStatus transactionStatus;
    private final PlatformTransactionManager transactionManager;

    // Spring会自动注入上下文内的事务管理器
    public TransactionHook(PlatformTransactionManager transactionManager) {
        this.transactionManager = transactionManager;
    }

    // 每个场景执行前开启事务
    @Before
    public void openTransaction() {
        transactionStatus = transactionManager.getTransaction(new DefaultTransactionDefinition());
    }

    // 每个场景执行后强制回滚事务
    @After
    public void rollbackTransaction() {
        if (transactionStatus != null && !transactionStatus.isCompleted()) {
            transactionManager.rollback(transactionStatus);
        }
    }
}

• 确认你的Cucumber上下文配置类标注了@SpringBootTest，如果有多数据源的情况，指定你需要回滚的事务管理器即可。

方案2：H2数据库快速重置（适合有事务无法覆盖的操作场景） #

如果你的测试流程里包含DDL语句、触发器调用等事务无法回滚的数据库操作，可以用H2自带的快速重置能力，速度远高于重建Spring上下文。

操作步骤 #

• 在测试配置的application-test.yml中配置H2为内存持久化模式：

spring:
  datasource:
    url: jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1
    driver-class-name: org.h2.Driver

• 编写数据库重置Hook，在每个场景执行前清空数据库并重新加载基础数据：

import io.cucumber.java.Before;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;
import org.springframework.core.io.ClassPathResource;
import javax.sql.DataSource;

public class DbResetHook {
    private final JdbcTemplate jdbcTemplate;
    private final DataSource dataSource;

    public DbResetHook(JdbcTemplate jdbcTemplate, DataSource dataSource) {
        this.jdbcTemplate = jdbcTemplate;
        this.dataSource = dataSource;
    }

    @Before
    public void resetH2Db() {
        // 清空当前库所有对象
        jdbcTemplate.execute("DROP ALL OBJECTS");
        // 重新执行你的基础数据初始化脚本
        new ResourceDatabasePopulator(
            new ClassPathResource("schema.sql"),
            new ClassPathResource("test-data.sql")
        ).execute(dataSource);
    }
}

如果你的基础数据是代码动态生成的，可以在第一次上下文启动完成后生成一次，然后用H2的BACKUP TO 'mem:testdb-backup'命令拍快照，每次场景前直接恢复快照，速度会更快。 #

注意事项 #

• 所有Hook类必须放到Cucumber启动类的glue属性指定的扫描包下，否则不会生效
• 如果你的测试用到了@Modifying注解的JPA方法，确保配置了clearAutomatically = true，避免一级缓存和数据库状态不一致的问题

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