JUnit 单元测试中如何覆盖回调逻辑（Callback）代码

在使用 junit 测试含异步回调（如 `soapactioncallback`）的方法时，需通过同步化手段捕获回调执行结果；推荐使用 `completablefuture` 等机制等待回调完成，并对回调参数或副作用进行断言验证。

在实际开发中，许多基于 Spring-WS 或类似框架的 SOAP 调用会采用回调模式（如 SoapActionCallback）来定制请求消息（MyMessageClass）。这类设计使核心逻辑与消息构建解耦，但也给单元测试带来挑战——因为回调代码不会立即执行，而是由框架在后续流程中异步调用，导致常规断言无法直接覆盖。

解决思路是：将异步回调“同步化”。CompletableFuture 是 Java 8+ 提供的轻量级异步协调工具，非常适合用于“等待回调触发并获取其输入对象”。其关键优势在于无需修改被测代码（即不侵入业务逻辑），仅需在测试中监听回调行为即可。

以下是一个典型、可复用的测试模板：

@Test
public void testMarshallWithCallback() throws Exception {
    // 1. 创建 CompletableFuture 用于接收回调参数
    final CompletableFuture callbackFuture = new CompletableFuture<>();

    // 2. 执行被测方法，并在回调中完成 future
    JAXBElement result = (JAXBElement) template.marshall(
        "some string",
        new SoapActionCallback("some string") {
    
        @Override
            public void doWithMessage(MyMessageClass message) {
                // ✅ 回调触发时，将 message 传递给 future
                callbackFuture.complete(message);
            }
        }
    );

    // 3. 同步等待回调完成（带超时，避免死等）
    MyMessageClass message = callbackFuture.get(5, TimeUnit.SECONDS);

    // 4. 对回调接收到的 message 进行断言（例如检查 header、payload 或自定义属性）
    assertNotNull(message);
    assertEquals("expected soap action", "some string", message.getSoapAction());
    // 其他业务相关断言...
}

⚠️ 注意事项：

• 务必设置超时（如 get(5, TimeUnit.SECONDS)）：防止因回调未触发导致测试无限挂起；
• 避免在回调中抛出异常：CompletableFuture.complete() 不传播异常；若需验证异常场景，应改用 callbackFuture.completeExceptionally(e) 并配合 assertThrows()；
• 若回调逻辑涉及状态变更（如修改外部对象、调用 mock 方法），也可结合 ArgumentCaptor 或 Mockito.verify() 进行行为验证，而非仅依赖返回值；
• 该方案适用于所有基于回调的异步/延迟执行场景（如 ResponseExtractor、WebServiceMessageCallback 等），具备良好通用性。

总结：覆盖回调代码的本质不是“模拟回调”，而是“捕获回调的执行时机与输入数据”。CompletableFuture 以最小侵入代价实现了测试可控性，是 JUnit 中处理回调逻辑的推荐实践。