DefaultPageObjectInitialiser.java

package net.thucydides.core.webdriver;

import com.google.common.base.Predicate;
import net.thucydides.core.annotations.locators.SmartElementProxyCreator;
import net.thucydides.core.pages.PageObject;
import org.openqa.selenium.WebDriver;

/**
 * Initializes WebElement and WebElementFacade fields in a page object.
 */
public class DefaultPageObjectInitialiser implements Predicate<PageObject> {

    private final WebDriver driver;
    private final int ajaxTimeoutInMilliseconds;
    private final ElementProxyCreator elementProxyCreator;

    public DefaultPageObjectInitialiser(WebDriver driver, int ajaxTimeoutInMilliseconds) {
        this.driver = driver;
        this.ajaxTimeoutInMilliseconds = ajaxTimeoutInMilliseconds;
        this.elementProxyCreator = new SmartElementProxyCreator();
    }

    protected int ajaxTimeoutInSecondsWithAtLeast1Second() {
        if (ajaxTimeoutInMilliseconds > 1000) {
            return ajaxTimeoutInMilliseconds / 1000;
        } else {
            return 1;
        }
    }

    public boolean apply(PageObject page) {
        page.setWaitForTimeout(ajaxTimeoutInMilliseconds);
        elementProxyCreator.proxyElements(page, driver, ajaxTimeoutInSecondsWithAtLeast1Second());
        return true;
    }
}