[pulsar] branch master updated: [Flaky test] Fix flaky test PulsarFunctionsTest.testExclamationFunction (#9864)

[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 c6ea5ba  [Flaky test] Fix flaky test 
PulsarFunctionsTest.testExclamationFunction (#9864)
c6ea5ba is described below

commit c6ea5bad506b1ab77d08a19ee018c14e7504a30e
Author: Lari Hotari <lhot...@users.noreply.github.com>
AuthorDate: Wed Mar 10 21:40:26 2021 +0200

    [Flaky test] Fix flaky test PulsarFunctionsTest.testExclamationFunction 
(#9864)
    
    Fixes #9863
---
 .../tests/integration/functions/PulsarFunctionsTest.java     | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git 
a/tests/integration/src/test/java/org/apache/pulsar/tests/integration/functions/PulsarFunctionsTest.java
 
b/tests/integration/src/test/java/org/apache/pulsar/tests/integration/functions/PulsarFunctionsTest.java
index 0845bf1..93a1800 100644
--- 
a/tests/integration/src/test/java/org/apache/pulsar/tests/integration/functions/PulsarFunctionsTest.java
+++ 
b/tests/integration/src/test/java/org/apache/pulsar/tests/integration/functions/PulsarFunctionsTest.java
@@ -24,7 +24,6 @@ import static org.testng.Assert.assertEquals;
 import static org.testng.Assert.assertTrue;
 import static org.testng.Assert.fail;
 import com.google.gson.Gson;
-
 import java.time.Duration;
 import java.util.Collections;
 import java.util.HashSet;
@@ -35,7 +34,6 @@ import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
-
 import lombok.Cleanup;
 import lombok.extern.slf4j.Slf4j;
 import net.jodah.failsafe.Failsafe;
@@ -93,6 +91,7 @@ import org.apache.pulsar.tests.integration.io.SourceTester;
 import org.apache.pulsar.tests.integration.topologies.FunctionRuntimeType;
 import org.apache.pulsar.tests.integration.topologies.PulsarCluster;
 import org.assertj.core.api.Assertions;
+import org.awaitility.Awaitility;
 import org.testcontainers.containers.GenericContainer;
 import org.testcontainers.shaded.com.google.common.collect.Sets;
 import org.testng.annotations.Test;
@@ -1647,8 +1646,13 @@ public abstract class PulsarFunctionsTest extends 
PulsarFunctionsTestBase {
         // update parallelism
         updateFunctionParallelism(functionName, 2);
 
-        //get function status
-        getFunctionStatus(functionName, 0, true, 2);
+        Awaitility.await()
+                .pollInterval(Duration.ofMillis(500L))
+                .ignoreExceptions()
+                .untilAsserted(() ->
+                //get function status
+                getFunctionStatus(functionName, 0, true, 2)
+        );
 
         // delete function
         deleteFunction(functionName);