如何在JUnit中等待@JMSListener注释方法完成

Question

因此，我试图对基于Spring（v4.1.6）的JMS处理进行一些集成测试。

这是一个非常标准的Spring设置，带有带@JmsListener注释的方法，DefaultMessageListenerContainer带有的concurrency设置为1，因此只允许一个侦听线程。

现在，我利用ActiveMQ的嵌入式代理不依赖任何外部jms代理来使测试随时随地运行（我应该从事市场营销工作）。

所以所有的电线都很好，然后我进行了JUnit测试：

@Test
public void test() {
    sendSomeMessage();
    //how to wait here for the @JMSListener method to complete
    verify();
}

我发送了消息，但随后我需要以某种方式等待带@JMSListener注释的方法完成。我该怎么做呢？

Answer 1

好吧，我希望可以以某种方式挂接到消息驱动的Pojos生命周期中，以实现此目的，但是在解决有关异步代码的其他SO问题时，我想出了一个基于
CountDownLatch

1. 所有工作完成后，带@JMSListener注释的方法应调用countDown()CountDownLatch：

        @JmsListener(destination = "dest", containerFactory = "cf")
    public void processMessage(TextMessage message) throws JMSException {
        //do the actual processing
        actualProcessing(message);
        //if there's countDownLatch call the countdown.
        if(countDownLatch != null) {
            countDownLatch.countDown();
        }
    }

1. 在testMethod中

        @Test
    public void test() throws InterruptedException {
        //initialize the countDownLatch and set in on the processing class
        CountDownLatch countDownLatch = new CountDownLatch(1);
        messageProcessor.setCountDownLatch(countDownLatch);
        //sendthemessage
        sendSomeMessage();
        //wait for the processing method to call countdown()
        countDownLatch.await();
        verify();
    }

该解决方案的缺点是您必须@JMSListener针对集成测试专门更改带注释的方法