JUnit单元测试中如何等待ExecutorChannel执行完成接收消息

ExecutorChannel 是异步处理通道，消息的消费逻辑会提交到关联的线程池用独立线程执行，测试主线程调用send()完成消息投递后会立刻继续执行，跑完测试方法就直接终止运行，此时异步消费线程还没来得及执行你写的订阅输出逻辑，自然看不到打印内容。

方案1：CountDownLatch 同步等待（无额外依赖，通用实现） #

用JDK自带的CountDownLatch做线程同步，让主线程阻塞到消费逻辑收到消息后再继续执行，记得设置超时时间避免测试异常时永久挂死：

import org.junit.jupiter.api.Test;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import static org.junit.jupiter.api.Assertions.assertTrue;

@Test
public void testOnePojo() throws Exception {
    ExecutorChannel orderSendChannel = context.getBean("validationChannel", ExecutorChannel.class);
    ExecutorChannel orderReceiveChannel = context.getBean("auditChannel", ExecutorChannel.class);
    
    // 单条消息等待，计数器初始值设为1
    CountDownLatch receiveLatch = new CountDownLatch(1);
    orderReceiveChannel.subscribe(t -> {
         System.out.println(t);
         // 收到消息后计数器减1，唤醒阻塞的主线程
         receiveLatch.countDown();
    }); 
    
    orderSendChannel.send(getMessageMessage());
    // 最多阻塞等待3秒，超时自动放行
    boolean isReceived = receiveLatch.await(3, TimeUnit.SECONDS);
    // 加断言校验消息确实被接收，避免超时导致测试假通过
    assertTrue(isReceived, "指定超时时间内未接收到auditChannel的响应消息");
}

方案2：Spring Integration 官方测试组件（更贴合生态的规范写法） #

如果你的测试已经集成了Spring Integration测试包，可以直接用内置的MessageCollector自动收集通道消息，不需要手动写订阅和同步逻辑：

import org.junit.jupiter.api.Test;
import org.springframework.integration.test.support.MessageCollector;
import org.springframework.messaging.Message;
import java.util.concurrent.TimeUnit;
import static org.junit.jupiter.api.Assertions.assertNotNull;

@Autowired
private MessageCollector messageCollector;

@Test
public void testOnePojo() throws Exception {
    ExecutorChannel orderSendChannel = context.getBean("validationChannel", ExecutorChannel.class);
    ExecutorChannel orderReceiveChannel = context.getBean("auditChannel", ExecutorChannel.class);
    
    orderSendChannel.send(getMessageMessage());
    // 直接从收集器轮询消息，内置超时等待逻辑
    Message<?> receivedMsg = messageCollector.forChannel(orderReceiveChannel)
            .poll(3, TimeUnit.SECONDS);
    
    assertNotNull(receivedMsg, "未接收到auditChannel的消息");
    System.out.println(receivedMsg);
}

• 所有等待逻辑必须设置合理的超时时间，不要用无参的await()或者无限阻塞的轮询，避免消息投递链路异常时测试进程永久卡住
• 不要用Thread.sleep()硬等固定时长，既会拖慢测试执行速度，在机器性能波动时还容易出现偶发测试失败

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