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

Question

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

部分代码

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();

Answer 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);

这对我有用......

Answer 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)