保持春季上下文处于活动状态,直到使用 JMS 消息

2022-09-03 15:51:53

我有一个相当标准的设置与 - 和.它工作正常,直到我试图做一个简单的集成测试。经过一些调查,我发现Spring上下文和嵌入式代理在使用第一条JMS消息后都关闭了,无论在消耗期间,都会触发另一个事件。我能够通过在测试设置中添加连接选项来解决代理问题,即JMSSpring BootActiveMQuseShutdownHook=false

spring.activemq.broker-url = vm://broker?async=false&broker.persistent=false&broker.useShutdownHook=false

我正在寻找的基本上是一种强制测试“保持活动”直到所有JMS消息都被消耗掉的方法(在这种情况下,它们只有两个)。我了解整个设置的异步性质,但在测试期间,仍然有助于获得生成和使用这些消息的所有结果。

以下是我的设置,尽管它相当简单。

@EnableJms
public class ActiveMqConfig {

    @Bean
    public JmsTemplate jmsTemplate(ConnectionFactory connectionFactory, MessageConverter messageConverter) {
        JmsTemplate jmsTemplate = new JmsTemplate(connectionFactory);
        jmsTemplate.setMessageConverter(messageConverter);
        return jmsTemplate;
    }

    @Bean
    public MessageConverter messageConverter() {
        MappingJackson2MessageConverter messageConverter = new MappingJackson2MessageConverter();
        messageConverter.setTargetType(MessageType.TEXT);
        messageConverter.setTypeIdPropertyName("_type");
        return messageConverter;
    }
}

然后,我有一个消息驱动的POJO来侦听给定的事件:

@JmsListener(destination = "events")
public void applicationSubmitted(MyType event) {
    // do some work with the event here

    jmsTemplate.convertAndSend("commands", mymessage);
}

还有一个:

@JmsListener(destination = "commands")
public void onCommand(TextMessage textMessage) {

}

我尝试过的一件事是添加延迟,即 发送邮件后。但是,这是非常不可靠的,并且也会减慢测试速度,因为执行可能需要不到50ms。下面是测试本身。除非等待被取消注释,否则我永远不会到达第二个事件侦听器,因为应用程序上下文关闭,测试结束并且消息被“遗忘”。sleep(200)

@SpringBootTest
class MyEventIntegrationTest extends Specification {

    @Autowired
    JmsTemplate jmsTemplate

    def "My event is successfully handled"() {

        given:
        def event = new MyEvent()

        when:
        jmsTemplate.convertAndSend("events", event)
        // sleep(200)

        then:
        1 == 1
    }
}

答案 1

好吧,这是测试基于异步消息交换的系统时的标准问题。通常,它会在您跳过的测试部分 - 该部分中解决。then

问题是,在你的测试中,你通常期望系统做一些有用的事情,例如,在数据库中进行更改,向另一个系统发送一个rest调用,在另一个队列中发送消息等。我们可以通过不断检查结果来等待一段时间,直到它发生 - 如果结果是在我们设置的时间窗口内实现的 - 那么我们可以假设测试已经通过。

此方法的伪代码如下:

for (i to MAX_RETRIES; i++) {
   checkThatTheChangesInDBHasBeenMade();
   checkThatTheRestCallHasBeenMade();
   checkThatTheMessageIsPostedInAnotherQueue();

   Thread.sleep(50ms);
}

这样,在最佳情况下,您的测试将在50ms内通过。在最坏的情况下,它将失败,这将需要MAX_RETRIES * 50ms时间来执行测试。

另外,我应该提到,有一个很好的工具叫做 awaitility,它提供了很好的 API(顺便说一句,它支持时髦的 DSL)来处理异步世界中的这类问题:

await().atMost(5, SECONDS).until(customerStatusIsUpdated());

答案 2

我认为问题的根源是异步事件处理。发送事件后,测试即刚刚结束。当然,这将导致Spring上下文和经纪人关闭。JMS 侦听器正在另一个线程中运行。你必须找到一种方法来等待他们。否则,您的线程(这是您的测试用例)刚刚完成。

我们在上一个项目中遇到了类似的问题,并编写了一个小实用程序,这对我们帮助很大。JMS提供了“浏览”队列并查看它是否为空的功能:

public final class JmsUtil {

    private static final int MAX_TRIES = 5000;
    private final JmsTemplate jmsTemplate;

    public JmsUtil(JmsTemplate jmsTemplate) {
        this.jmsTemplate = jmsTemplate;
    }

    private int getMessageCount(String queueName) {
        return jmsTemplate.browseSelected(queueName, "true = true", (s, qb) -> Collections.list(qb.getEnumeration()).size());
    }

    public void waitForAll(String queueName) {
        int i = 0;
        while (i <= MAX_TRIES) {
            if (getMessageCount(queueName) == 0) {
                return;
            }
            i++;
        }
}

使用此实用程序,您可以执行如下操作:

def "My event is successfully handled"() {

        given:
        def event = new MyEvent()

        when:
        jmsTemplate.convertAndSend("events", event)
        jmsUtility.waitForAll("events"); // wait until the event has been consumed
        jmsUtility.waitForAll("commands"); // wait until the command has been consumed

        then:
        1 == 1
    }

注: 此实用程序假定您将 JMS 消息发送到队列。通过浏览队列,我们可以检查它是否为空。如果是主题,您可能需要进行另一次检查。所以要注意这一点!


推荐