java 查找元素是否存在_java - WebDriver：检查元素是否存在？

我扩展了Selenium WebDriver实现，在我的案例中是HtmlUnitDriver来公开一个方法

public boolean isElementPresent(By by){}

像这样：

检查页面是否在超时期限内加载。

加载页面后，我将WebDriver的隐式等待时间降低到几毫秒，在我的情况下为100毫秒，可能也应该使用0工厂。

调用findElements(By)，WebDriver即使找不到元素也只会等待上面的时间量。

提升未来页面加载的隐含等待时间

这是我的代码：

import java.util.concurrent.TimeUnit;

import org.openqa.selenium.By;

import org.openqa.selenium.JavascriptExecutor;

import org.openqa.selenium.WebDriver;

import org.openqa.selenium.htmlunit.HtmlUnitDriver;

import org.openqa.selenium.support.ui.ExpectedCondition;

import org.openqa.selenium.support.ui.WebDriverWait;

public class CustomHtmlUnitDriver extends HtmlUnitDriver {

public static final long DEFAULT_TIMEOUT_SECONDS = 30;

private long timeout = DEFAULT_TIMEOUT_SECONDS;

public long getTimeout() {

return timeout;

}

public void setTimeout(long timeout) {

this.timeout = timeout;

}

public boolean isElementPresent(By by) {

boolean isPresent = true;

waitForLoad();

//search for elements and check if list is empty

if (this.findElements(by).isEmpty()) {

isPresent = false;

}

//rise back implicitly wait time

this.manage().timeouts().implicitlyWait(timeout, TimeUnit.SECONDS);

return isPresent;

}

public void waitForLoad() {

ExpectedCondition pageLoadCondition = new ExpectedCondition() {

public Boolean apply(WebDriver wd) {

//this will tel if page is loaded

return "complete".equals(((JavascriptExecutor) wd).executeScript("return document.readyState"));

}

};

WebDriverWait wait = new WebDriverWait(this, timeout);

//wait for page complete

wait.until(pageLoadCondition);

//lower implicitly wait time

this.manage().timeouts().implicitlyWait(100, TimeUnit.MILLISECONDS);

}

}

用法：

CustomHtmlUnitDriver wd = new CustomHtmlUnitDriver();

wd.get("http://example.org");

if (wd.isElementPresent(By.id("Accept"))) {

wd.findElement(By.id("Accept")).click();

}

else {

System.out.println("Accept button not found on page");

}