com.axonivy.ivy.supplements.primeui.tester.AjaxHelper.java Source code

Introduction

Here is the source code for com.axonivy.ivy.supplements.primeui.tester.AjaxHelper.java

Source

/*
 * Copyright (C) 2016 AXON IVY AG
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.axonivy.ivy.supplements.primeui.tester;

import java.util.concurrent.TimeUnit;

import org.openqa.selenium.TimeoutException;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;

/**
 * An API using a {@link org.openqa.selenium.WebDriver WebDriver} to wait for a
 * {@link org.openqa.selenium.WebElement WebElement} to reach an
 * {@link org.openqa.selenium.support.ui.ExpectedCondition ExpectedCondition}.
 * 
 * @author Denis Huelin
 * @since 6.0.0
 */
public class AjaxHelper {
    private WebDriver driver;

    public AjaxHelper(WebDriver driver) {
        this.driver = driver;
    }

    public <T> T await(ExpectedCondition<T> condition) {
        return waitAtMost(10, TimeUnit.SECONDS, condition);
    }

    public <T> T waitAtMost(long waitingTime, TimeUnit timeUnit, ExpectedCondition<T> condition) {
        driver.manage().timeouts().implicitlyWait(1, TimeUnit.SECONDS);
        try {
            return new WebDriverWait(driver, timeUnit.toSeconds(waitingTime)).until(condition);
        } catch (TimeoutException ex) {
            System.out.println("Timed out while " + condition + " in source code: " + driver.getPageSource());
            throw ex;
        } finally {
            driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);
        }
    }
}