From 4f22a560667d9794efe9f5b941f52280ed0d172d Mon Sep 17 00:00:00 2001
From: Clebert Suconic <clebertsuconic@apache.org>
Date: Fri, 5 May 2017 14:45:21 -0400
Subject: [PATCH] NO-JIRA Trivial test fix

---
 .../activemq/artemis/tests/integration/client/SessionTest.java  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/client/SessionTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/client/SessionTest.java
index 2242cfcf6e..9954a4e617 100644
--- a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/client/SessionTest.java
+++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/client/SessionTest.java
@@ -38,6 +38,7 @@ import org.apache.activemq.artemis.core.settings.impl.AddressSettings;
 import org.apache.activemq.artemis.spi.core.protocol.RemotingConnection;
 import org.apache.activemq.artemis.tests.util.ActiveMQTestBase;
 import org.apache.activemq.artemis.tests.util.CountDownSessionFailureListener;
+import org.apache.activemq.artemis.tests.util.Wait;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
@@ -482,6 +483,7 @@ public class SessionTest extends ActiveMQTestBase {
       Assert.assertNotNull(m);
       m.acknowledge();
       clientSession.rollback();
+      Wait.waitFor(() -> getMessageCount(q) == 10);
       Assert.assertEquals(10, getMessageCount(q));
       clientSession.close();
       sendSession.close();