[pulsar] branch master updated: Fix flaky test ProducerCloseTest.brokerCloseTopicTest. (#16644)

[technoboy]

This is an automated email from the ASF dual-hosted git repository.

technoboy pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/pulsar.git


The following commit(s) were added to refs/heads/master by this push:
     new 888898c030e Fix flaky test ProducerCloseTest.brokerCloseTopicTest. 
(#16644)
888898c030e is described below

commit 888898c030e5dab4fcfc726c718ba547243a7c22
Author: Jiwei Guo <techno...@apache.org>
AuthorDate: Mon Jul 18 19:52:10 2022 +0800

    Fix flaky test ProducerCloseTest.brokerCloseTopicTest. (#16644)
---
 .../src/test/java/org/apache/pulsar/client/impl/ProducerCloseTest.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git 
a/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ProducerCloseTest.java
 
b/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ProducerCloseTest.java
index 040bfe5813f..5366937cd28 100644
--- 
a/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ProducerCloseTest.java
+++ 
b/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ProducerCloseTest.java
@@ -28,6 +28,7 @@ import org.apache.pulsar.client.api.PulsarClientException;
 import org.apache.pulsar.client.api.TypedMessageBuilder;
 import org.apache.pulsar.common.api.proto.CommandSuccess;
 import org.apache.pulsar.common.naming.TopicName;
+import org.awaitility.Awaitility;
 import org.testng.Assert;
 import org.testng.annotations.AfterMethod;
 import org.testng.annotations.BeforeMethod;
@@ -137,7 +138,7 @@ public class ProducerCloseTest extends ProducerConsumerBase 
{
                 
.getTopicReference(TopicName.get(topic).getPartitionedTopicName());
         Assert.assertTrue(topicOptional.isPresent());
         topicOptional.get().close(true).get();
-        Assert.assertEquals(producer.getState(), 
HandlerState.State.Connecting);
+        Awaitility.await().untilAsserted(() -> 
Assert.assertEquals(producer.getState(), HandlerState.State.Connecting));
         if (isAsyncSend) {
             producer.newMessage().value("test".getBytes()).sendAsync().get();
         } else {