From dd4291154b3cf3c19aaa014c1b726a02a996436e Mon Sep 17 00:00:00 2001
From: Igor Ohrimenko <iokhrime@redhat.com>
Date: Mon, 12 Mar 2018 14:39:30 +0200
Subject: [PATCH] Add new common methods to the 'SeleniumWebDriverHelper' class
 (#9065)

---
 .../pageobject/SeleniumWebDriverHelper.java   | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/selenium/che-selenium-test/src/main/java/org/eclipse/che/selenium/pageobject/SeleniumWebDriverHelper.java b/selenium/che-selenium-test/src/main/java/org/eclipse/che/selenium/pageobject/SeleniumWebDriverHelper.java
index 0977b58a3e..abdd3e5615 100644
--- a/selenium/che-selenium-test/src/main/java/org/eclipse/che/selenium/pageobject/SeleniumWebDriverHelper.java
+++ b/selenium/che-selenium-test/src/main/java/org/eclipse/che/selenium/pageobject/SeleniumWebDriverHelper.java
@@ -13,6 +13,7 @@ package org.eclipse.che.selenium.pageobject;
 import static java.util.Arrays.asList;
 import static java.util.stream.Collectors.toList;
 import static org.eclipse.che.selenium.core.constant.TestTimeoutsConstants.LOAD_PAGE_TIMEOUT_SEC;
+import static org.openqa.selenium.support.ui.ExpectedConditions.frameToBeAvailableAndSwitchToIt;
 import static org.openqa.selenium.support.ui.ExpectedConditions.invisibilityOfAllElements;
 import static org.openqa.selenium.support.ui.ExpectedConditions.invisibilityOfElementLocated;
 import static org.openqa.selenium.support.ui.ExpectedConditions.presenceOfAllElementsLocatedBy;
@@ -730,4 +731,43 @@ public class SeleniumWebDriverHelper {
       return false;
     }
   }
+
+  /**
+   * Waits during {@code timeout} until frame which defined by {@code frameLocator} is available and
+   * switches to it.
+   *
+   * @param frameLocator locator of the frame to which should be switched
+   * @param timeout waiting time in seconds
+   */
+  public void waitAndSwitchToFrame(By frameLocator, int timeout) {
+    webDriverWaitFactory.get(timeout).until(frameToBeAvailableAndSwitchToIt(frameLocator));
+  }
+
+  /**
+   * Waits until frame which defined by {@code frameLocator} is available and switches to it.
+   *
+   * @param frameLocator locator of the frame to which should be switched
+   */
+  public void waitAndSwitchToFrame(By frameLocator) {
+    waitAndSwitchToFrame(frameLocator, DEFAULT_TIMEOUT);
+  }
+
+  /**
+   * Waits during {@code timeout} until {@code frame} is available and switches to it.
+   *
+   * @param frame web element which defines frame
+   * @param timeout waiting time in seconds
+   */
+  public void waitAndSwitchToFrame(WebElement frame, int timeout) {
+    webDriverWaitFactory.get(timeout).until(frameToBeAvailableAndSwitchToIt(frame));
+  }
+
+  /**
+   * Waits until {@code frame} is available and switches to it.
+   *
+   * @param frame web element which defines frame
+   */
+  public void waitAndSwitchToFrame(WebElement frame) {
+    waitAndSwitchToFrame(frame, DEFAULT_TIMEOUT);
+  }
 }