[fix][test]Fix flaky test V1_ProducerConsumerTest.testConcurrentConsu…

…merReceiveWhileReconnect (apache#24019) (cherry picked from commit 2c686d0) (cherry picked from commit c8d8b6f)
datastax · Feb 28, 2025 · ee7c76c · ee7c76c
1 parent 1ce9b02
commit ee7c76c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/client/api/v1/V1_ProducerConsumerTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/client/api/v1/V1_ProducerConsumerTest.java
@@ -589,9 +589,11 @@ public Void call() throws Exception {
         restartBroker();
 
         // The available permits should be 10 and num messages in the queue should be 90
-        Awaitility.await().untilAsserted(() ->
-                Assert.assertEquals(consumerImpl.getAvailablePermits(), numConsumersThreads));
-        Assert.assertEquals(consumerImpl.numMessagesInQueue(), recvQueueSize - numConsumersThreads);
+        Awaitility.await().untilAsserted(() -> {
+            Assert.assertEquals(consumerImpl.getAvailablePermits(), numConsumersThreads);
+            Assert.assertEquals(consumerImpl.numMessagesInQueue(), recvQueueSize - numConsumersThreads);
+        });
+
         consumer.close();
     }