Disabling receiveNoWait for ActiveMQ test (#755)

Author: eyalkoren

apm-agent-plugins/apm-jms-plugin/src/test/java/co/elastic/apm/agent/jms/ActiveMqArtemisFacade.java

@@ -151,6 +151,11 @@ public Message receive(Destination destination, long timeout) {
         return context.createConsumer(destination).receive(timeout);
     }
 
+    @Override
+    public boolean shouldTestReceiveNoWait() {
+        return true;
+    }
+
     @Override
     public Message receiveNoWait(Destination destination) {
         return context.createConsumer(destination).receiveNoWait();

apm-agent-plugins/apm-jms-plugin/src/test/java/co/elastic/apm/agent/jms/ActiveMqFacade.java

@@ -143,6 +143,11 @@ public Message receive(Destination destination, long timeout) throws JMSExceptio
         return session.createConsumer(destination).receive(timeout);
     }
 
+    @Override
+    public boolean shouldTestReceiveNoWait() {
+        return false;
+    }
+
     @Override
     public Message receiveNoWait(Destination destination) throws JMSException {
         return session.createConsumer(destination).receiveNoWait();

apm-agent-plugins/apm-jms-plugin/src/test/java/co/elastic/apm/agent/jms/BrokerFacade.java

@@ -58,5 +58,7 @@ interface BrokerFacade {
 
     Message receive(Destination destination, long timeout) throws Exception;
 
+    boolean shouldTestReceiveNoWait();
+
     Message receiveNoWait(Destination destination) throws Exception;
 }

apm-agent-plugins/apm-jms-plugin/src/test/java/co/elastic/apm/agent/jms/JmsInstrumentationIT.java

@@ -118,6 +118,9 @@ public void testQueueSendReceiveOnTracedThread() throws Exception {
 
     @Test
     public void testQueueSendReceiveNoWaitOnTracedThread() throws Exception {
+        if (!brokerFacade.shouldTestReceiveNoWait()) {
+            return;
+        }
         final Queue queue = createQueue();
         testQueueSendReceiveOnTracedThread(() -> brokerFacade.receiveNoWait(queue), queue, true);
     }
@@ -130,6 +133,9 @@ public void testQueueSendReceiveOnNonTracedThread() throws Exception {
 
     @Test
     public void testQueueSendReceiveNoWaitOnNonTracedThread() throws Exception {
+        if (!brokerFacade.shouldTestReceiveNoWait()) {
+            return;
+        }
         final Queue queue = createQueue();
         testQueueSendReceiveOnNonTracedThread(() -> brokerFacade.receiveNoWait(queue), queue, true);
     }