如何让Gradle等待CompletableFuture回调完成，同时保持测试并行？

问题场景 #

你测试返回CompletableFuture的方法时，遇到两个核心问题：

• 直接调用whenComplete时，测试主线程会在异步任务和回调执行完成前退出，导致回调从未执行
• 用join()等待任务完成会阻塞当前线程，担心影响测试并行性

解决方案 #

其实JUnit的并行测试是测试方法级别的并行——每个测试方法在独立线程中执行，单个测试内的线程阻塞不会影响其他测试的并行运行。你只需要让当前测试的主线程等待自身的异步任务和回调完成即可，以下是两种可行方式：

方式1：利用whenComplete返回的CompletableFuture

whenComplete会返回一个新的CompletableFuture，该Future会在回调执行完毕后才进入完成状态。直接对这个Future调用join()，就能等待回调执行完成：

@Test
// 可选：添加超时防止无限等待
@Timeout(value = 5, unit = TimeUnit.SECONDS)
void testFoo() {
    foo()
        .whenComplete((result, throwable) -> {
            System.out.println(result);
            // 在这里添加断言逻辑
            assertEquals("ABC", result);
            assertNull(throwable);
        })
        .join(); // 等待回调执行完成
}

方式2：使用CountDownLatch

如果需要更灵活的等待控制（比如同时等待多个回调），可以用CountDownLatch：

@Test
@Timeout(value = 5, unit = TimeUnit.SECONDS)
void testFoo() {
    CountDownLatch latch = new CountDownLatch(1);

    foo().whenComplete((result, throwable) -> {
        System.out.println(result);
        assertEquals("ABC", result);
        latch.countDown(); // 回调执行完成后计数减1
    });

    try {
        // 等待计数归零，超时则判定测试失败
        if (!latch.await(5, TimeUnit.SECONDS)) {
            fail("回调未在规定时间内执行");
        }
    } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
        fail("等待过程被中断");
    }
}

关键说明 #

• 测试框架的并行性是多个测试方法同时运行，每个测试方法的线程是独立的。单个测试内的join()或await只会阻塞当前测试的线程，不会干扰其他测试的执行。
• 添加@Timeout注解是最佳实践，避免因异步任务异常导致测试无限挂起。

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