diff --git a/artemis-junit/src/main/java/org/apache/activemq/artemis/junit/Wait.java b/artemis-junit/src/main/java/org/apache/activemq/artemis/junit/Wait.java index e37539a3a6..c0aa55d40c 100644 --- a/artemis-junit/src/main/java/org/apache/activemq/artemis/junit/Wait.java +++ b/artemis-junit/src/main/java/org/apache/activemq/artemis/junit/Wait.java @@ -52,7 +52,11 @@ public class Wait { } public static void assertEquals(long size, LongCondition condition, long timeout) throws Exception { - boolean result = waitFor(() -> condition.getCount() == size, timeout); + assertEquals(size, condition, timeout, SLEEP_MILLIS); + } + + public static void assertEquals(Long size, LongCondition condition, long timeout, long sleepMillis) throws Exception { + boolean result = waitFor(() -> condition.getCount() == size, timeout, sleepMillis); if (!result) { Assert.fail(size + " != " + condition.getCount()); diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/QueueControlTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/QueueControlTest.java index 0faa055f53..123f581248 100644 --- a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/QueueControlTest.java +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/QueueControlTest.java @@ -2632,25 +2632,25 @@ public class QueueControlTest extends ManagementTestBase { Supplier durableCount, Supplier durableSize) throws Exception { //make sure count stat equals message count - Assert.assertTrue(Wait.waitFor(() -> count.get().longValue() == messageCount, 3, 100)); + Assert.assertTrue(Wait.waitFor(() -> count.get().longValue() == messageCount, 3 * 1000, 100)); if (messageCount > 0) { //verify size stat greater than 0 - Assert.assertTrue(Wait.waitFor(() -> size.get().longValue() > 0, 3, 100)); + Assert.assertTrue(Wait.waitFor(() -> size.get().longValue() > 0, 3 * 1000, 100)); //If durable then make sure durable count and size are correct if (durable) { - Assert.assertTrue(Wait.waitFor(() -> durableCount.get().longValue() == messageCount, 3, 100)); - Assert.assertTrue(Wait.waitFor(() -> durableSize.get().longValue() > 0, 3, 100)); + Wait.assertEquals(messageCount, () -> durableCount.get().longValue(), 3 * 1000, 100); + Assert.assertTrue(Wait.waitFor(() -> durableSize.get().longValue() > 0, 3 * 1000, 100)); } else { - Assert.assertTrue(Wait.waitFor(() -> durableCount.get().longValue() == 0, 3, 100)); - Assert.assertTrue(Wait.waitFor(() -> durableSize.get().longValue() == 0, 3, 100)); + Wait.assertEquals(0L, () -> durableCount.get().longValue(), 3 * 1000, 100); + Wait.assertEquals(0L, () -> durableSize.get().longValue(), 3 * 1000, 100); } } else { - Assert.assertTrue(Wait.waitFor(() -> count.get().longValue() == 0, 3, 100)); - Assert.assertTrue(Wait.waitFor(() -> durableCount.get().longValue() == 0, 3, 100)); - Assert.assertTrue(Wait.waitFor(() -> size.get().longValue() == 0, 3, 100)); - Assert.assertTrue(Wait.waitFor(() -> durableSize.get().longValue() == 0, 3, 100)); + Wait.assertEquals(0L, () -> count.get().longValue(), 3 * 1000, 100); + Wait.assertEquals(0L, () -> durableCount.get().longValue(), 3 * 1000, 100); + Wait.assertEquals(0L, () -> size.get().longValue(), 3 * 1000, 100); + Wait.assertEquals(0L, () -> durableSize.get().longValue(), 3 * 1000, 100); } } }