如何在Spring中测试回滚场景？含服务代码与集成测试框架

没问题，我来帮你完成这个回滚场景的集成测试。Spring的测试框架提供了非常方便的事务控制能力，结合@Transactional注解和TestTransaction工具类，我们可以轻松验证回滚逻辑。

首先，先明确两个核心测试场景：

1. 验证Spring测试框架的自动回滚行为（默认行为）
2. 验证业务异常触发的事务回滚

下面是完整的测试实现，我会逐段解释：

完整测试类代码 #

@Transactional
@RunWith(SpringRunner.class)
@SpringBootTest(classes = JpaApplication.class)
public class MyServiceIntegrationTest {

    @Autowired
    private MyService2 myService2;

    @Autowired
    private Entity2Repo entity2Repo;

    // 测试场景1：验证Spring测试的自动回滚机制
    @Test
    public void testCreateWithAutomaticRollback() {
        Long entity1Id = 1L;

        // 调用服务创建Entity2
        myService2.create(entity1Id);

        // 事务内验证数据已保存
        Optional<Entity2> savedEntity = entity2Repo.findByEntity1Id(entity1Id);
        assertThat(savedEntity).isPresent();

        // 测试结束后，Spring会自动回滚这个事务，数据库中不会留下这条数据
    }

    // 测试场景2：验证异常触发的事务回滚（用TestTransaction手动控制）
    @Test
    public void testRollbackOnBusinessException() {
        Long entity1Id = 2L;

        try {
            // 启动事务（@Transactional已自动启动，这里显式调用也可以）
            TestTransaction.start();

            // 执行创建操作
            myService2.create(entity1Id);

            // 模拟业务逻辑中抛出异常
            throw new RuntimeException("模拟业务校验失败");
        } catch (RuntimeException e) {
            // 标记当前事务需要回滚
            TestTransaction.flagForRollback();
        }

        // 结束事务，触发回滚
        TestTransaction.end();

        // 验证数据未被持久化到数据库
        Optional<Entity2> savedEntity = entity2Repo.findByEntity1Id(entity1Id);
        assertThat(savedEntity).isNotPresent();
    }

    // 测试场景3：如果你的create方法本身会抛出异常（比如后续修改后）
    @Test
    public void testCreateWithInvalidInputTriggersRollback() {
        Long invalidEntity1Id = -1L;

        // 调用会抛出异常的方法
        assertThrows(IllegalArgumentException.class, () -> {
            myService2.create(invalidEntity1Id);
        });

        // 验证数据未被保存
        Optional<Entity2> savedEntity = entity2Repo.findByEntity1Id(invalidEntity1Id);
        assertThat(savedEntity).isNotPresent();
    }
}

关键知识点解释 #

• @Transactional注解：在测试类上添加这个注解后，Spring会为每个测试方法创建独立的事务，默认在测试方法执行完毕后自动回滚，避免测试数据污染数据库。
• TestTransaction工具类：这是Spring Test提供的专用工具，让我们可以手动控制事务的生命周期（启动、标记回滚、结束），非常适合精准测试回滚场景。
• Spring事务回滚规则：默认情况下，只有当抛出**未检查异常（RuntimeException/Error）**时，事务才会自动回滚；如果是检查异常，需要在@Transactional中配置rollbackFor属性来指定触发回滚的异常类型。

依赖提示 #

确保你的项目中引入了Spring Test和AssertJ的依赖（如果用Maven）：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>

这样就能完整覆盖回滚场景的测试了，你可以根据实际业务需求调整测试逻辑~

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