如何使用Selenium WebDriver和Java关闭特定窗口?

我使用Selenium WebDriver。我打开第一页,然后打开第二页 - 执行一些操作并返回第一页。在我想关闭第二页之前,我使用命令,但它关闭了第一页而不是第二页。如何使硒关闭特定窗口?driver.close();

部分代码

String HandleBefore = driver.getWindowHandle();

 driver.findElement(By.xpath("...")).click();
 for (String twohandle : driver.getWindowHandles()) {
        driver.switchTo().window(twohandle);
    }       
 driver.findElement(By.linkText("001")).click();
 driver.close();

答案 1
    String base = driver.getWindowHandle();

    Set <String> set = driver.getWindowHandles();

    set.remove(base);
    assert set.size()==1;

    driver.switchTo().window(set.toArray(new String[0]));

    driver.close();
    driver.switchTo().window(base);

这对我有用...


答案 2

在 Python 中

default_handle = driver.current_window_handle
handles = list(driver.window_handles)
assert len(handles) > 1

handles.remove(default_handle)
assert len(handles) > 0

driver.switch_to_window(handles[0])
# do your stuffs
driver.close()
driver.switch_to_window(default_handle)