From 11865d17413aa2d821e0d67ae3a36dec4c2146f4 Mon Sep 17 00:00:00 2001 From: fengyubiao Date: Mon, 24 Feb 2025 21:52:35 +0800 Subject: [PATCH] [fix][test]Fix flaky test V1_ProducerConsumerTest.testConcurrentConsumerReceiveWhileReconnect --- .../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 0b3ff345acfc3..255591ea7da77 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(); }