fix flakiness

cdapio · Jul 13, 2023 · 3821128 · 3821128
1 parent c69c460
commit 3821128
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/e2e-test/java/io/cdap/cdap/ui/stepsdesign/PipelineEdit.java b/src/e2e-test/java/io/cdap/cdap/ui/stepsdesign/PipelineEdit.java
@@ -195,6 +195,8 @@ public void draftShouldNotExist(String pipelineName) {
 
   @Then("Generated runtime arguments should be empty")
   public void checkRuntimeArgsEmpty() {
+    WaitHelper.waitForElementToBePresent(
+      By.cssSelector(Helper.getCssSelectorByDataTestId("generated-runtimeargs-count")));
     Assert.assertTrue(Helper.locateElementByTestId("generated-runtimeargs-count").getText().charAt(0) == '0');
     ElementHelper.clickOnElement(Helper.locateElementByTestId("generated-runtimeargs"));
     List<WebElement> keys = Helper.locateElementsByTestId("runtimeargs-key");
@@ -209,6 +211,8 @@ public void checkRuntimeArgsEmpty() {
 
   @Then("Generated runtime arguments should not be empty")
   public void checkRuntimeArgsNotEmpty() {
+    WaitHelper.waitForElementToBePresent(
+      By.cssSelector(Helper.getCssSelectorByDataTestId("generated-runtimeargs-count")));
     Assert.assertFalse(Helper.locateElementByTestId("generated-runtimeargs-count").getText().charAt(0) == '0');
     ElementHelper.clickOnElement(Helper.locateElementByTestId("generated-runtimeargs"));
     List<WebElement> keys = Helper.locateElementsByTestId("runtimeargs-key");