This commit is contained in:
Timothy Bish 2020-04-13 12:12:32 -04:00
commit 0a4ec3cbff
2 changed files with 4 additions and 3 deletions

View File

@ -19,6 +19,7 @@ package org.apache.activemq.artemis.tests.integration.amqp;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import org.apache.activemq.artemis.core.server.Queue; import org.apache.activemq.artemis.core.server.Queue;
import org.apache.activemq.artemis.utils.Wait;
import org.apache.activemq.transport.amqp.client.AmqpClient; import org.apache.activemq.transport.amqp.client.AmqpClient;
import org.apache.activemq.transport.amqp.client.AmqpConnection; import org.apache.activemq.transport.amqp.client.AmqpConnection;
import org.apache.activemq.transport.amqp.client.AmqpMessage; import org.apache.activemq.transport.amqp.client.AmqpMessage;
@ -52,7 +53,7 @@ public class AmqpDeliveryAnnotationsTest extends AmqpClientTestSupport {
receiver.flow(1); receiver.flow(1);
Queue queue = getProxyToQueue(getQueueName()); Queue queue = getProxyToQueue(getQueueName());
assertEquals(1, queue.getMessageCount()); Wait.assertEquals(1L, queue::getMessageCount, 5000L, 25L);
AmqpMessage received = receiver.receive(5, TimeUnit.SECONDS); AmqpMessage received = receiver.receive(5, TimeUnit.SECONDS);
assertNotNull(received); assertNotNull(received);

View File

@ -124,8 +124,8 @@ public class AmqpReceiverPriorityTest extends AmqpClientTestSupport {
for (int i = 0; i < 2; i++) { for (int i = 0; i < 2; i++) {
AmqpMessage message1 = receiver1.receiveNoWait(); AmqpMessage message1 = receiver1.receive(3000, TimeUnit.MILLISECONDS);
assertNotNull("did not receive message first time", message1); assertNotNull("did not receive message" + i, message1);
assertEquals("MessageID:" + i, message1.getMessageId()); assertEquals("MessageID:" + i, message1.getMessageId());
message1.accept(); message1.accept();
} }