From cd8640b6b94538e28e5e642fa28b8616e198e19e Mon Sep 17 00:00:00 2001 From: fengyubiao Date: Tue, 25 Feb 2025 14:48:55 +0800 Subject: [PATCH] [fix][test]Fix flaky test V1_ProducerConsumerTest.testConcurrentConsumerReceiveWhileReconnect (#24019) (cherry picked from commit 2c686d0ab1ea3f697dea4f5ce7938282b9b06f88) --- .../pulsar/client/api/v1/V1_ProducerConsumerTest.java | 8 +++++--- 1 file changed, 5 insertions(+), 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 index e126d963a88f5..d25041e1dfcbd 100644 --- 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(); }