Selenium中实现健壮的元素查找与重试机制

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.TimeoutException;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

import java.time.Duration; // For Selenium 4+

public class ElementFinder {

    /**
     * 在指定重试次数内查找并返回可见的WebElement。
     * 如果在所有重试次数后仍未找到元素，则抛出TimeoutException。
     *
     * @param driver     WebDriver实例。
     * @param by         用于定位元素的By策略。
     * @param retryCount 最大重试次数。
     * @param waitTimeoutPerAttempt 每次尝试的显式等待超时时间（秒）。
     * @return 找到的WebElement。
     * @throws TimeoutException 如果在所有重试后仍未找到元素。
     */
    public static WebElement findElementWithRetry(WebDriver driver, By by, int retryCount, int waitTimeoutPerAttempt) {
        for (int i = 1; i <= retryCount; i++) {
            try {
                // 为每次尝试创建一个新的WebDriverWait实例，设置其超时时间
                WebDriverWait wait = new WebDriverWait(driver, Duration.ofSeconds(waitTimeoutPerAttempt));
                WebElement element = wait.until(ExpectedConditions.visibilityOfElementLocated(by));

                // 如果元素可见，则成功找到并返回
                if (element != null && element.isDisplayed()) {
                    System.out.println("成功找到元素: " + by + " (第 " + i + " 次尝试)");
                    return element;
                }
            } catch (TimeoutException e) {
                // 捕获超时异常，表示当前尝试未找到元素
                System.out.println("元素 " + by + " 在第 " + i + " 次尝试中超时未找到，进行重试...");
                // 可以选择在这里添加短暂的Thread.sleep(100)来模拟等待，但通常WebDriverWait已足够
            } catch (Exception e) {
                // 捕获其他可能的异常，例如NoSuchElementException（虽然ExpectedConditions通常会处理）
                System.out.println("元素 " + by + " 在第 " + i + " 次尝试中遇到其他异常: " + e.getMessage());
            }
        }
        // 所有重试都失败后，抛出最终的TimeoutException
        throw new TimeoutException(String.format("元素 %s 在 %s 次重试后仍未找到或不可见。", by, retryCount));
    }

    // 示例用法
    public static void main(String[] args) {
        // 假设 driver 已经被初始化，例如：
        // WebDriver driver = new ChromeDriver();
        // driver.get("http://your-test-url.com");

        // 模拟一个WebDriver实例 (实际应用中替换为真实的WebDriver)
        WebDriver driver = null; // 占位符，实际使用时需初始化

        try {
            // 尝试查找一个ID为'myElement'的元素，最多重试3次，每次等待10秒
            // WebElement myElement = findElementWithRetry(driver, By.id("myElement"), 3, 10);
            // System.out.println("最终找到元素: " + myElement.getTagName());

            // 模拟一个找不到的场景
            // 注意：在没有真实driver的情况下，此调用会抛出NullPointerException
            // 实际使用时，请确保driver已正确初始化并指向一个有效的页面
            System.out.println("--- 模拟查找一个不存在的元素 ---");
            WebElement nonExistentElement = findElementWithRetry(driver, By.id("nonExistentElement"), 3, 5);
            System.out.println("找到不存在的元素 (不应该发生): " + nonExistentElement);

        } catch (TimeoutException e) {
            System.err.println("错误：所有重试失败 - " + e.getMessage());
        } catch (Exception e) {
            System.err.println("发生其他错误: " + e.getMessage());
        } finally {
            // if (driver != null) {
            //     driver.quit();
            // }
        }
    }
}