如何避免SpringBootTest在异步方法完成前停止服务器？

你的问题很典型：当控制器用CompletableFuture启动异步任务后立即返回响应，Spring Boot集成测试会在请求处理完成后直接关闭上下文，导致异步任务还没执行完就被终止，没法验证后续逻辑。而且你提到的MvcResult.getAsyncResult()确实不适用——它只针对Spring MVC原生的异步请求（比如返回Callable/DeferredResult），而你自己启动的异步线程不在这个范畴里。

下面给你几个实用的解决方案，按场景推荐：

方案1：用任务跟踪器手动等待异步任务完成（推荐，不影响生产代码） #

这个思路是在测试环境中专门跟踪所有异步任务，测试时等待它们全部执行完毕。

步骤1：创建测试环境专属的任务跟踪器 #

@Component
@Profile("integration_tests") // 只在集成测试环境生效
public class AsyncTaskTracker {
    private final List<CompletableFuture<Void>> trackedTasks = new CopyOnWriteArrayList<>();

    public void trackTask(CompletableFuture<Void> task) {
        trackedTasks.add(task);
    }

    public void waitForAllTasks() throws InterruptedException, ExecutionException {
        // 等待所有跟踪的任务完成
        CompletableFuture.allOf(trackedTasks.toArray(new CompletableFuture[0])).get();
        trackedTasks.clear(); // 清理，避免影响下一个测试
    }
}

步骤2：修改控制器，跟踪异步任务 #

@Autowired(required = false) // 生产环境无需此Bean，避免注入失败
private AsyncTaskTracker taskTracker;

@RequestMapping(value = "/create", method = RequestMethod.POST)
public Result createModel(DataModel model) {
    CompletableFuture<Void> asyncTask = CompletableFuture.runAsync(() -> doSomeThing(model));
    if (taskTracker != null) {
        taskTracker.trackTask(asyncTask); // 仅在测试环境跟踪任务
    }
    return new Result("your request has been recieved successfully");
}

步骤3：在测试中等待任务完成 #

@Autowired
private AsyncTaskTracker taskTracker;

@Test
public void testCreateModel() throws Exception {
    // 发送请求
    this.mockMvc.perform(post("/create"))
            .andExpect(status().isOk());

    // 等待所有异步任务执行完毕
    taskTracker.waitForAllTasks();

    // 现在验证数据库操作
    assertEquals(1, JdbcTestUtil.countRowsInTable(jdbcTemplate, "data_model_table"));
}

方案2：把异步线程池替换为同步执行（适合复杂异步链路） #

如果你的异步逻辑链路很长，跟踪每个任务太麻烦，可以在测试环境把异步线程池改成同步执行——这样所有异步任务都会在请求线程里立即执行，控制器返回时任务已经完成。

步骤1：创建测试环境的异步配置 #

@Configuration
@Profile("integration_tests")
public class TestAsyncConfiguration implements AsyncConfigurer {
    @Override
    public Executor getAsyncExecutor() {
        // 返回一个同步执行的Executor，所有任务直接在当前线程跑
        return Runnable::run;
    }
}

步骤2：修改控制器，使用Spring管理的线程池 #

确保你的异步任务用的是Spring的TaskExecutor，而不是默认的ForkJoinPool：

@Autowired
private TaskExecutor taskExecutor;

@RequestMapping(value = "/create", method = RequestMethod.POST)
public Result createModel(DataModel model) {
    // 指定用Spring的线程池执行异步任务
    CompletableFuture.runAsync(() -> doSomeThing(model), taskExecutor);
    return new Result("your request has been recieved successfully");
}

这样在测试环境中，任务会同步执行，测试代码不需要额外修改，直接验证即可。

方案3：用CountDownLatch临时测试单个场景（快速验证） #

如果只是临时测试某个接口，不想改太多代码，可以用CountDownLatch来等待异步任务完成：

@Test
public void testCreateModel() throws Exception {
    CountDownLatch latch = new CountDownLatch(1);

    // 用Mockito Spy包装你的业务Bean，在任务完成时触发latch
    DataService dataService = spy(wac.getBean(DataService.class));
    doAnswer(invocation -> {
        // 先执行真实方法
        Object result = invocation.callRealMethod();
        // 任务完成，倒计时减1
        latch.countDown();
        return result;
    }).when(dataService).doSomeThing(any(DataModel.class));

    // 发送请求
    this.mockMvc.perform(post("/create"))
            .andExpect(status().isOk());

    // 等待最多10秒，避免无限阻塞
    assertTrue("异步任务超时未完成", latch.await(10, TimeUnit.SECONDS));

    // 验证数据库
    assertEquals(1, JdbcTestUtil.countRowsInTable(jdbcTemplate, "data_model_table"));
}

这个方案适合快速验证，但如果异步链路有多个步骤，需要多个CountDownLatch，维护起来比较麻烦。

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