[pulsar] branch master updated: [Flaky test] Fix flaky RabbitMQSourceTest (#9869)

[mmerli]

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

mmerli 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 6ff3244  [Flaky test] Fix flaky RabbitMQSourceTest (#9869)
6ff3244 is described below

commit 6ff3244ccd963984053e669c32c289675ffeeaa3
Author: Lari Hotari <lhot...@users.noreply.github.com>
AuthorDate: Wed Mar 10 21:32:27 2021 +0200

    [Flaky test] Fix flaky RabbitMQSourceTest (#9869)
---
 .../java/org/apache/pulsar/io/rabbitmq/sink/RabbitMQSinkTest.java     | 2 +-
 .../java/org/apache/pulsar/io/rabbitmq/source/RabbitMQSourceTest.java | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git 
a/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/sink/RabbitMQSinkTest.java
 
b/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/sink/RabbitMQSinkTest.java
index 33642f0..375d391 100644
--- 
a/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/sink/RabbitMQSinkTest.java
+++ 
b/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/sink/RabbitMQSinkTest.java
@@ -66,7 +66,7 @@ public class RabbitMQSinkTest {
 
         // open should success
         // rabbitmq service may need time to initialize
-        Awaitility.await().untilAsserted(() -> sink.open(configs, null));
+        Awaitility.await().ignoreExceptions().untilAsserted(() -> 
sink.open(configs, null));
 
         // write should success
         Record<byte[]> record = build("test-topic", "fakeKey", "fakeValue", 
"fakeRoutingKey");
diff --git 
a/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/source/RabbitMQSourceTest.java
 
b/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/source/RabbitMQSourceTest.java
index 8983e36..40e1a28 100644
--- 
a/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/source/RabbitMQSourceTest.java
+++ 
b/pulsar-io/rabbitmq/src/test/java/org/apache/pulsar/io/rabbitmq/source/RabbitMQSourceTest.java
@@ -20,6 +20,7 @@ package org.apache.pulsar.io.rabbitmq.source;
 
 import org.apache.pulsar.io.rabbitmq.RabbitMQBrokerManager;
 import org.apache.pulsar.io.rabbitmq.RabbitMQSource;
+import org.awaitility.Awaitility;
 import org.testng.annotations.AfterMethod;
 import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.Test;
@@ -64,7 +65,8 @@ public class RabbitMQSourceTest {
         RabbitMQSource source = new RabbitMQSource();
 
         // open should success
-        source.open(configs, null);
+        // rabbitmq service may need time to initialize
+        Awaitility.await().ignoreExceptions().untilAsserted(() -> 
source.open(configs, null));
     }
 
 }