[ZEPPELIN-6088] Fix some flaky tests in zeppelin-integration (apache#…

…4826) * Fix flaky SparkParagraphIT.testCancelPyspark * Fix flaky PersonalizeActionsIT.testGraphAction
seung-00 · Sep 14, 2024 · 7ae8980 · 7ae8980
1 parent 6a71af0
commit 7ae8980
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/zeppelin-integration/src/test/java/org/apache/zeppelin/integration/PersonalizeActionsIT.java b/zeppelin-integration/src/test/java/org/apache/zeppelin/integration/PersonalizeActionsIT.java
@@ -19,6 +19,7 @@
 import org.apache.zeppelin.AbstractZeppelinIT;
 import org.apache.zeppelin.MiniZeppelinServer;
 import org.apache.zeppelin.WebDriverManager;
+import org.apache.zeppelin.ZeppelinITUtils;
 import org.apache.zeppelin.test.DownloadUtils;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.AfterEach;
@@ -239,6 +240,7 @@ void testGraphAction() throws Exception {
 
       pollingWait(By.xpath(getParagraphXPath(1) +
           "//button[contains(@uib-tooltip, 'Table')]"), MAX_BROWSER_TIMEOUT_SEC).click();
+      ZeppelinITUtils.sleep(1000, false);
       assertEquals("fa fa-table", manager.getWebDriver().findElement(By.xpath(getParagraphXPath(1)
           + "//button[contains(@class," +
           "'btn btn-default btn-sm ng-binding ng-scope active')]//i")).getAttribute("class"));

diff --git a/zeppelin-integration/src/test/java/org/apache/zeppelin/integration/SparkParagraphIT.java b/zeppelin-integration/src/test/java/org/apache/zeppelin/integration/SparkParagraphIT.java
@@ -188,6 +188,7 @@ void testCancelPyspark() throws Exception {
       ZeppelinITUtils.sleep(2000, false);
       cancelParagraph(1);
       waitForParagraph(1, "ABORT");
+      ZeppelinITUtils.sleep(1000, false);
 
       assertEquals("ABORT", getParagraphStatus(1),
           "First paragraph status is " + getParagraphStatus(1));