主页 / coding / 问题 / 79126672
Accepted
snej0keee
Asked: 2024-10-26 01:02:33 +0800 CST

@Transactional 不能用 @DataJpaTest 进行测试

  • 772

我被方法周围的单元测试困住了@Transactional。问题是我无法测试内部的回滚机制@DataJpaTest。
这是我的简化示例应用程序：

@RestController
public class MyController {

    private final MyRepository myRepository;
    private final MyService    myService;

    public MyController(MyRepository myRepository, MyService myService) {
        this.myRepository = myRepository;
        this.myService = myService;
    }

    @GetMapping
    @Transactional
    public String justForTest() {
        var myEntity = new MyEntity().setStatus("NEW");
        myRepository.save(myEntity);
        try {
            myService.throwsAnRuntimeException();
            // on service succeed
            myEntity.setStatus("SUCCESS");
        } catch (Exception e) {
            // on service fail
            myEntity.setStatus("FAIL");
            throw e;
        } finally {
            myRepository.save(myEntity);
        }
        return "this is the end";
    }
}

@Data
@Entity
@Accessors(chain = true)
public class MyEntity {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String status;
}

@Repository
public interface MyRepository extends JpaRepository<MyEntity, Long> {
}

@Service
public class MyService {
    public void throwsAnRuntimeException() {
        throw new RuntimeException("Rollbacks not working!");
    }
}

因此，当我以通常的方式（使用本地 postgresql 数据库）运行此代码时，它会按预期工作：当我在 localhost:8080 上调用查询时，响应为 500，并且我的数据库中的表完全为空，正如预期的那样，因为当方法@Transactional以 终止时RuntimeException，它会执行rollback整个事务（即使最终块有repository.save()）。但是当我尝试使用（测试范围的类路径中的 h2 数据库）
测试此代码时：@DataJpaTest

@DataJpaTest
class MyControllerTest {

    @Autowired
    private MyRepository repository;

    @Test
    void justForTest() {
        var controller = new MyController(repository, new MyService());
        assertThrows(RuntimeException.class, controller::justForTest);
        var afterTestEntities = repository.findAll();
        System.out.println(afterTestEntities.getFirst());
        assertEquals(0, afterTestEntities.size());
    }
}

我收到测试失败，并且调试 println 显示我的实体状态为失败MyEntity(id=1, status=FAIL)。但我认为，就像在实际运行中一样，由于事务回滚，我应该有一个空的存储库。

所以，我的主要问题是如何@Transactional在单元测试中测试该机制？

java

2 个回答

  1. 主要问题是@DataJpaTest没有处理控制器方法上的 @Transactional 注释，因为控制器在您的测试设置中不受 Spring 管理，请尝试添加事务测试配置：

    @Configuration
@EnableTransactionManagement
public class TransactionalTestConfig {
    @Bean
    public PlatformTransactionManager transactionManager(EntityManagerFactory emf) {
        return new JpaTransactionManager(emf);
    }
}

    然后在你的测试课中：

    
@DataJpaTest
@Import({TransactionalTestConfig.class, MyService.class})
class MyControllerTest {

 @Autowired
    private MyRepository repository;
    
    @Autowired
    private PlatformTransactionManager transactionManager;
    
    @Autowired
    private MyService myService;

   @Test
    void testTransactionalRollback() {
        // Create a transaction template
        TransactionTemplate transactionTemplate = new TransactionTemplate(transactionManager);
        
        // Execute the test within a transaction
        assertThrows(RuntimeException.class, () -> {
            transactionTemplate.execute(status -> {
                var controller = new MyController(repository, myService);
                controller.justForTest();
                return null;
            });
        });

        // Verify the rollback
        var afterTestEntities = repository.findAll();
        assertEquals(0, afterTestEntities.size());
    }

}

    在您的财产中：

    spring.jpa.show-sql=true
spring.jpa.properties.hibernate.format_sql=true
logging.level.org.springframework.transaction=TRACE
    • 0
  2. Best Answer

    发生这种情况是因为@DataJpaTest测试默认是事务性的。事务在每个测试之前启动，并在测试完成后回滚。显然，您无法在事务内部测试事务回滚。您需要使用@Transactional测试上的注释禁用事务传播。

    @DataJpaTest
@Transactional(propagation = Propagation.NOT_SUPPORTED)
class MyNonTransactionalTests {

    // ...

}

    有关更多信息，请参阅Spring Boot 参考。

    此外，如果控制器不是 Spring Bean，@Transactional则无法工作。因此，您必须将其添加到测试中@Import，然后使用 将其注入测试@Autowired。

    @DataJpaTest
@Import({MyController.class})
@Transactional(propagation = Propagation.NOT_SUPPORTED)
public class MyControllerTest {

    @Autowired
    private MyController myController;
 
    // ... 
}
    • 0

相关问题