Selenium - как восстановить файлы cookie в новом окне браузера

Предположим, нам нужно проверить следующий сценарий:

1. Перейдите на страницу входа и войдите в приложение.
2. Закройте браузер.
3. Откройте браузер и перейдите на страницу входа в систему - пользователь не должен видеть форму входа и должен быть уже авторизован.

При первом входе в систему в браузере сохраняются файлы cookie. В WebDriver, когда окно браузера закрывается, все данные сеанса и файлы cookie удаляются, поэтому тестирование описанного выше сценария становится невозможным.


К счастью, у WebDriver есть функция чтения файлов cookie из браузера перед его закрытием и последующего восстановления файлов cookie в новом окне браузера.

import org.openqa.selenium.By; import org.openqa.selenium.Cookie; import org.openqa.selenium.WebDriver; import org.openqa.selenium.firefox.FirefoxDriver; import org.testng.Assert; import java.util.Set; public class CookieTest {
WebDriver driver;
@Test
public void login_state_should_be_restored() {
driver = new FirefoxDriver();

driver.get('http://www.example.com/login');
driver.findElement(By.id('username')).sendKeys('admin');
driver.findElement(By.id('password')).sendKeys('12345');
driver.findElement(By.id('login')).click();

Assert.assertTrue(


driver.findElement(By.id('welcome')).isDisplayed());

//Before closing the browser, read the cookies
Set allCookies = driver.manage().getCookies();

driver.close();

//open a new browser window
driver = new FirefoxDriver();

//restore all cookies from previous session
for(Cookie cookie : allCookies) {

driver.manage().addCookie(cookie);
}

driver.get('http://www.example.com/login'); //Login page should not be disaplyed
Assert.assertTrue(


driver.findElement(By.id('welcome')).isDisplayed());

driver.close();
} }