Selenium 自动化测试中，经常会遇到因网络波动、服务器响应慢等原因导致页面加载失败的情况，表现为浏览器打开一个空白页面。如果每次打开新页面都单独处理重试逻辑，代码会变得冗余且难以维护。本文将介绍一种全局性的重试机制，无需修改每个打开页面的方法，即可自动处理页面加载空白的情况。
实现原理
该机制的核心思想是利用 WebDriverWait 和 JavascriptExecutor 动态检查页面加载状态。具体步骤如下：
页面加载策略： 确保 Selenium 使用合适的页面加载策略。normal 策略会等待所有资源加载完成，而 eager 策略只等待初始 HTML 文档加载和解析完成。根据实际情况选择合适的策略。
-
document.readyState 检查： 使用 JavaScript 代码 document.readyState 获取当前页面的加载状态。该属性有以下几种可能的值：
- loading: 页面正在加载。
- interactive: 页面已经加载完成，文档已被解析，但诸如图像、样式表和框架之类的子资源仍在加载。
- complete: 页面完全加载。
只有当 document.readyState 的值为 complete 时，才认为页面加载完成。
WebDriverWait 等待： 使用 WebDriverWait 结合 ExpectedCondition，循环检查 document.readyState 的值。如果在指定时间内 document.readyState 的值变为 complete，则认为页面加载成功，否则超时。
重试机制： 如果 WebDriverWait 超时，说明页面加载失败，可以刷新当前页面并重新执行上述步骤，直到页面加载成功或达到最大重试次数。
代码示例 (Java)
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;
public class PageLoadHelper {
/**
* 等待页面加载完成.
*
* @param driver WebDriver 实例
* @param timeoutInSeconds 超时时间，单位秒
* @param maxRetries 最大重试次数
*/
public static void waitForLoad(WebDriver driver, int timeoutInSeconds, int maxRetries) {
int retryCount = 0;
boolean loaded = false;
while (retryCount < maxRetries && !loaded) {
try {
new WebDriverWait(driver, timeoutInSeconds).until((ExpectedCondition) wd ->
((JavascriptExecutor) wd).executeScript("return document.readyState").equals("complete"));
loaded = true; // 页面加载成功
} catch (Exception e) {
System.err.println("页面加载失败，正在重试... (第 " + (retryCount + 1) + " 次)");
driver.navigate().refresh(); // 刷新页面
retryCount++;
}
}
if (!loaded) {
System.err.println("页面加载失败，已达到最大重试次数。");
// 可以抛出异常或进行其他处理
throw new RuntimeException("页面加载失败，已达到最大重试次数。");
}
}
} 使用方法:
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
public class ExampleTest {
public static void main(String[] args) {
// 设置 ChromeDriver 路径
System.setProperty("webdriver.chrome.driver", "/path/to/chromedriver");
WebDriver driver = new ChromeDriver();
driver.get("https://www.example.com");
// 等待页面加载完成，超时时间30秒，最大重试次数3次
PageLoadHelper.waitForLoad(driver, 30, 3);
// 页面加载完成后，可以继续执行其他操作
System.out.println("页面加载完成！");
driver.quit();
}
}代码解释:
- waitForLoad 方法接收 WebDriver 实例、超时时间和最大重试次数作为参数。
- 使用 while 循环进行重试，直到页面加载成功或达到最大重试次数。
- WebDriverWait 用于等待 document.readyState 变为 complete。
- 如果 WebDriverWait 超时，则刷新页面并增加重试次数。
- 如果达到最大重试次数仍然加载失败，则抛出异常。
注意事项
- 页面加载策略： 根据实际情况选择合适的页面加载策略，例如 normal 或 eager。可以通过 ChromeOptions 或 FirefoxOptions 设置页面加载策略。
- 超时时间： 设置合理的超时时间，避免无限等待。
- 最大重试次数： 设置最大重试次数，避免陷入无限循环。
- 异常处理： 在达到最大重试次数仍然加载失败时，进行适当的异常处理，例如抛出异常或记录日志。
- JavaScript 错误： 页面加载空白也可能是由于 JavaScript 错误导致的。需要检查浏览器的开发者工具，查看是否存在 JavaScript 错误。
总结
本文介绍了一种通用的 Selenium 页面加载空白重试机制，通过动态函数结合页面加载策略和 document.readyState 状态检查，实现了自动刷新页面并重试加载的功能。该机制可以有效提高自动化测试的稳定性和可靠性，避免因网络环境不稳定或服务器响应缓慢导致的页面加载失败问题。在实际应用中，可以根据具体情况调整超时时间和最大重试次数，以达到最佳效果。
