本文介绍使用 Mockito Spying 结合 CountDownLatch 和 AtomicReference，精准验证 CompletableFuture.runAsync 触发的异步方法是否被调用、调用次数及执行线程是否符合预期。
本文介绍使用 mockito spying 结合 countdownlatch 和 atomicreference，精准验证 `completablefuture.runasync` 触发的异步方法是否被调用、调用次数及执行线程是否符合预期。
在 Java 单元测试中验证异步行为（如 CompletableFuture.runAsync）是一项常见但易出错的任务。直接使用 verify(mock, times(1)).methodB(...) 会失败，因为 methodB 并非同步执行，测试主线程通常在异步任务完成前就已结束断言。解决方案是对被测对象（SUT）进行部分模拟（spy），并配合同步机制捕获异步执行上下文。
关键思路如下：
- 使用
spy()包装真实ClassA实例，保留其全部逻辑，仅对目标方法（methodB）增强可观测性； - 利用
doAnswer().when(spy).methodB(...)拦截调用，在真实执行前记录当前线程（Thread.currentThread()），并在执行后触发CountDownLatch通知主线程； - 主线程调用
methodA后，通过latch.await(timeout)阻塞等待异步任务完成； - 最终通过
verify()确认调用次数，并用assertThat(...).isNotSameAs(Thread.currentThread())验证确实在独立线程中执行。
以下是完整可运行的 JUnit 5 测试示例（需引入 Mockito 4+ 和 AssertJ）：
Java注释规范Skills
下载
当用户要求为 Java 代码添加注释、修改注释、审查注释、整理注释格式，或要求按照代码注释规范处理 Java 文件时，使用本技能。本技能用于统一生成和维护 Java 类注释模板、按 Git 对比 release 判断版本号、补充每行代码上方中文块注释、调整注释空行规则，并保持 Java 文件 UTF-8 无 BOM 编码。典型触发场景包括：补充类注释、更新 @version、检查 @author 来源、整理行尾注释、把注释放到代码上一行、检查注释是否符合规范、构建前检查 Java 注释风格。
import static org.mockito.Mockito.*;
import static org.assertj.core.api.Assertions.*;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicReference;
import org.junit.jupiter.api.Test;
import org.mockito.Spy;
class ClassATest {
@Test
void testMethodA_InvokesMethodBAsynchronously() throws InterruptedException {
// Given: 构造被测对象（注入真实依赖，如 executor）
ExecutionContextExecutor executor = Executors.newSingleThreadExecutor();
ClassA sut = new ClassA(executor);
sut = spy(sut); // 创建 spy
Object x = "x", y = "y";
CountDownLatch latch = new CountDownLatch(1);
AtomicReference<Thread> asyncThreadRef = new AtomicReference<>();
// When: 拦截 methodB 调用，记录线程并释放 latch
doAnswer(invocation -> {
try {
asyncThreadRef.set(Thread.currentThread());
return invocation.callRealMethod(); // 执行真实 methodB
} finally {
latch.countDown();
}
}).when(sut).methodB(eq(x), eq(y));
// 触发异步流程
sut.methodA(x, y);
// 等待异步执行完成（带超时避免死锁）
boolean completed = latch.await(500, TimeUnit.MILLISECONDS);
assertThat(completed).isTrue(); // 确保未超时
// Then: 验证调用次数与线程隔离性
verify(sut, times(1)).methodB(eq(x), eq(y));
assertThat(asyncThreadRef.get())
.as("methodB must run in a different thread")
.isNotNull()
.isNotSameAs(Thread.currentThread());
// 清理资源（可选）
executor.shutdown();
}
}⚠️ 注意事项：
立即学习“Java免费学习笔记（深入）”；
-
CountDownLatch的超时值（如500ms）应根据methodB实际耗时合理设置，过短导致误报，过长拖慢测试； - 若
methodB抛出异常，callRealMethod()会传播异常，建议在doAnswer中添加日志或异常处理以提升调试性； -
spy()对 final 方法、private 方法或静态方法无效，确保methodB是 public/protected 且非 final； - 生产环境中的
ExecutionContextExecutor应通过依赖注入传入，测试中可使用Executors.newSingleThreadExecutor()或Mockito.mock()替代，但需注意线程可见性； - 避免在测试中使用
Thread.sleep()替代CountDownLatch——它不可靠且低效。
通过该模式，你不仅能验证“是否调用”，还能验证“是否异步调用”，使单元测试真正覆盖并发逻辑的关键契约。
