1 files changed, 99 insertions, 0 deletions
diff --git a/vid-automation/src/main/java/vid/automation/test/infra/Wait.java b/vid-automation/src/main/java/vid/automation/test/infra/Wait.java
new file mode 100644
index 000000000..fa6ce4f38
--- /dev/null
+++ b/vid-automation/src/main/java/vid/automation/test/infra/Wait.java
@@ -0,0 +1,99 @@
+package vid.automation.test.infra;
+
+import org.openecomp.sdc.ci.tests.utilities.GeneralUIUtils;
+import org.openqa.selenium.JavascriptExecutor;
+import org.openqa.selenium.NoSuchElementException;
+import vid.automation.test.Constants;
+
+import java.util.concurrent.TimeUnit;
+import java.util.function.Predicate;
+
+public class Wait {
+    public static boolean byText(String text) {
+        return GeneralUIUtils.findAndWaitByText(text, Constants.generalTimeout);
+    }
+
+    public static <T> boolean waitFor(Predicate<T> predicate, T input, int numOfRetries, int interval, TimeUnit intervalUnit) {
+        for (int i=0; i<numOfRetries; i++) {
+            try {
+                if (predicate.test(input)) {
+                    return true;
+                }
+            }
+            catch (Throwable t) {
+                System.out.println("a retry failed duo to:" +t.getMessage());
+            }
+            try {
+                intervalUnit.sleep(interval);
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
+        return false;
+    }
+
+    public static <T> boolean waitFor(Predicate<T> predicate, T input, int numOfRetries, int interval) {
+        return waitFor(predicate, input, numOfRetries, interval, TimeUnit.SECONDS);
+    }
+
+    public static boolean waitByClassAndText(String className, String text, int timeoutInSeconds) {
+        return waitFor((x->Get.byClassAndText(className,text)!=null),null, timeoutInSeconds, 1);
+    }
+
+    public static boolean waitByTestId(String dataTestId,  int timeoutInSeconds) {
+        return waitFor((x->Get.byTestId(dataTestId)!=null),null, timeoutInSeconds, 1);
+    }
+
+    public static void angularHttpRequestsLoaded() {
+        JavascriptExecutor js = (JavascriptExecutor) GeneralUIUtils.getDriver();
+        for (int i=0; i<Constants.generalRetries; i++) {
+            Object result = js.executeScript("return window.angular.element('body').injector().get('$http').pendingRequests.length;");
+            if(result.toString().equals("0")) {
+                break;
+            } else {
+                try {
+                    Thread.sleep(1000);
+                } catch (InterruptedException e) {
+                    e.printStackTrace();
+                }
+            }
+        }
+    }
+
+    public static void modalToDisappear() {
+        for (int i=0; i<Constants.generalRetries; i++) {
+            try {
+                Object modalElement =  Get.byCssSelector(Constants.Modals.modalClass);
+                if(modalElement == null) {
+                    break;
+                } else {
+                    Thread.sleep(1000);
+                }
+            } catch (NoSuchElementException e) {
+                break;
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
+    public static void modalToBeDisplayed() {
+        for (int i=0; i<Constants.generalRetries; i++) {
+            try {
+                Object modalElement =  Get.byCssSelector(Constants.Modals.modalClass);
+                if(modalElement == null) {
+                    Thread.sleep(1000);
+                } else {
+                    break;
+                }
+            } catch (Exception e) {
+                try {
+                    Thread.sleep(1000);
+                } catch (InterruptedException e1) {
+                    e1.printStackTrace();
+                }
+                e.printStackTrace();
+            }
+        }
+    }
+}