Сначала вам нужно создать объект драйвера, а после вы можете переключить windows по идентификатору, имени и WebElement. Затем у объекта драйвера есть функции для возврата к окну по умолчанию. Как в этом примере.
// create driver object
WebDriver driver = DriverManager.getDriver();
// change window using iframe id or iframe name
driver.switchTo().frame("frame id or frame name");
// change window using WebElement object
driver.switchTo().frame(webElement);
Переключить назад окно
// switch back to main frame
driver.switchTo().parentFrame();
// switch back one frame
driver.switchTo().defaultContent();
Копия содержимого из (прочтите эту статью) статья об изменении окна селена . Важная информация здесь.