Spring Boot 单元测试覆盖 TransactionTemplate 编程式事务

业务代码

@Service
public class MyService {

    @Resource
    TransactionTemplate transactionTemplate;
    
    public void doMyMethod(String value) {
        transactionTemplate.executeWithoutResult(status -> {
            // 业务逻辑
        });
    }
}

单元测试

class MyUnitTest {

    @Mock
    TransactionTemplate transactionTemplate;

    @InjectMocks
    MyService myService;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
    }

    @Test
    void testMyMethod() {
        // 模拟 transactionTemplate.executeWithoutResult 的行为
        Mockito.doAnswer(invocation -> {
            // 获取传入的 TransactionCallbackWithoutResult 对象
            Consumer<TransactionStatus> callback  = invocation.getArgument(0);
            // 执行 executeWithoutResult 方法
            callback.accept(null);
            return null;
        }).when(transactionTemplate).executeWithoutResult(Mockito.any());

        // 调用被测试的方法
        myService.doMyMethod("myValue");

        // 验证 transactionTemplate.executeWithoutResult 是否被调用
        Mockito.verify(transactionTemplate).executeWithoutResult(Mockito.any());
    }

}