Selenium WebDriver多窗口操作与代理配置深度解析

import static org.junit.jupiter.api.Assertions.assertNotEquals;

import java.time.Duration;
import java.util.Set; // 导入Set以处理多个窗口句柄

import org.junit.jupiter.api.Test;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WindowType;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

public class MultiWindowTabExample {

    @Test
    void switchTabsInSameSessionTest() {
        // 设置ChromeDriver路径
        System.setProperty("webdriver.chrome.driver", "F:/drivers/chromedriver.exe"); // 请替换为您的chromedriver路径
        WebDriver driver = new ChromeDriver();
        driver.manage().timeouts().implicitlyWait(Duration.ofSeconds(5)); // 设置隐式等待

        try {
            // 1. 打开第一个URL
            driver.get("https://www.google.com");
            driver.manage().window().maximize();
            String firstTabHandle = driver.getWindowHandle(); // 保存第一个标签页的句柄
            System.out.println("第一个标签页句柄: " + firstTabHandle);

            // 2. 在同一浏览器会话中打开一个新的标签页
            // newTabDriver 实际上就是 driver，只是焦点已切换
            WebDriver newTabDriver = driver.switchTo().newWindow(WindowType.TAB);
            newTabDriver.get("https://www.msn.com/");
            String secondTabHandle = newTabDriver.getWindowHandle(); // 保存新标签页的句柄
            System.out.println("第二个标签页句柄: " + secondTabHandle);

            // 验证两个句柄是否不同
            assertNotEquals(firstTabHandle, secondTabHandle, "两个标签页的句柄应该不同");

            // 3. 在新标签页中进行操作（例如，等待页面加载）
            waitForBodyLoad(newTabDriver);

            // 4. 切换回第一个标签页
            driver.switchTo().window(firstTabHandle);
            System.out.println("已切换回第一个标签页");
            waitForBodyLoad(driver); // 等待页面加载

            // 5. 切换回第二个标签页
            driver.switchTo().window(secondTabHandle);
            System.out.println("已切换回第二个标签页");
            waitForBodyLoad(driver); // 等待页面加载

            // 6. 再次演示在不同窗口操作
            // 在第二个标签页（msn.com）搜索
            WebElement msnSearch = new WebDriverWait(driver, Duration.ofSeconds(10))
                .until(ExpectedConditions.elementToBeClickable(By.id("search"))); // 假设msn搜索框id为search
            if (msnSearch != null) {
                msnSearch.sendKeys("Selenium WebDriver");
                System.out.println("在MSN搜索框输入: Selenium WebDriver");
            }

            // 切换回第一个标签页（google.com）搜索
            driver.switchTo().window(firstTabHandle);
            WebElement googleSearch = new WebDriverWait(driver, Duration.ofSeconds(10))
                .until(ExpectedConditions.elementToBeClickable(By.name("q"))); // Google搜索框name为q
            if (googleSearch != null) {
                googleSearch.sendKeys("Java Automation");
                System.out.println("在Google搜索框输入: Java Automation");
            }

        } finally {
            // 确保浏览器会话最终被关闭
            if (driver != null) {
                driver.quit();
                System.out.println("浏览器会话已关闭");
            }
        }
    }

    // 辅助方法：等待页面body元素加载，表示页面已准备好
    private void waitForBodyLoad(WebDriver driver) {
        new WebDriverWait(driver, Duration.ofSeconds(5))
            .until(ExpectedConditions.visibilityOfElementLocated(By.xpath("//body")));
    }
}

// 第一个驱动实例及其代理配置
Proxy proxy1 = new Proxy();
proxy1.setHttpProxy("http://" + "proxy1.example.com:8080");
proxy1.setSslProxy("http://" + "proxy1.example.com:8080");
ChromeOptions options1 = new ChromeOptions();
options1.addArguments("start-maximized");
options1.setCapability("proxy", proxy1);
WebDriver driver1 = new ChromeDriver(options1);
driver1.get("https://www.siteA.com"); // driver1控制的浏览器会话使用proxy1

// 第二个驱动实例及其代理配置
Proxy proxy2 = new Proxy();
proxy2.setHttpProxy("http://" + "proxy2.example.com:8080");
proxy2.setSslProxy("http://" + "proxy2.example.com:8080");
ChromeOptions options2 = new ChromeOptions();
options2.addArguments("start-maximized");
options2.setCapability("proxy", proxy2);
WebDriver driver2 = new ChromeDriver(options2);
driver2.get("https://www.siteB.com"); // driver2控制的浏览器会话使用proxy2