Selenium 쿠키 핸들링, cookie 사용하여 로그인 자동화 java

해당 포스트는 java 언어로 작성되었다. 오늘은 웹 드라이버의 기능 중에 쿠키를 다루는 방법을 알아본다.

 

쿠키 핸들링 방법

 

쿠키 다루기

예를 들어, 로그인을 해야 하는 시나리오를 작성한다고 해보자.

로그인해서 장바구니 진입, 로그인해서 최근 구매 목록 진입, 로그인해서....

 

매번 시나리오마다 로그인을 따로 한다면 테스트 실행 시간은 그만큼 늘어날 것이다. 로그인 과정을 생략하기 위해, 최초 로그인할 때 쿠키 정보를 파일에 보관한 후, 다음번 로그인에는 쿠키를 가져와 드라이버에 추가하는 식으로 처리할 수 있다.

 

 

 

쿠키 저장하기

웹 드라이버는 아래 메서드로 현재 로드한 웹 페이지의 모든 쿠키 정보를 가져올 수 있다.

driver.manage().getCookies()

쿠키는 이름, 값, 도메인, 경로, 만료일, 암호화 관련 정보를 가진다. 쿠키 정보를 파일로 저장하고, TC 실행 시 파일을 읽어 필요한 정보를 드라이버에 전달해 보자. 네이버 서비스로 예제를 작성해 보았다.

 

네이버 로그인 쿠키 정보를 browser.data 파일에 저장하자.

@Test
public void login() {
	  
	  // 네이버 아이디 비밀번호 입력
	  driver.findElement(By.id("id")).sendKeys("******");
	  driver.findElement(By.id("pw")).sendKeys("******");
	  
	  // 로그인 버튼 클릭
	  driver.findElement(By.id("log.login")).click();
	  
	  // 쿠키 정보를 파일로 저장
	  File dataFile = new File("./target/browser.data");
	  try {
		  dataFile.delete();
		  dataFile.createNewFile();
		  FileWriter fos = new FileWriter(dataFile);
		  BufferedWriter bos = new BufferedWriter(fos);
		  
		  for (Cookie ck : driver.manage().getCookies()) {
			  bos.write((ck.getName() + ";" + ck.getValue() + ";" + ck.getDomain()
					  + ";" + ck.getPath() + ";" + ck.getExpiry() + ";" + ck.isSecure() ) );
			  bos.newLine();
		  }
		  bos.flush();
		  bos.close();
		  fos.close();
	  } catch (Exception ex) {
		  ex.printStackTrace();
	  }
	  
}

@BeforeMethod
public void beforeMethod() {
	  System.setProperty("webdriver.chrome.driver", "./src/test/resources/drivers/chromedriver");
	  
	  driver = new ChromeDriver();
	  
	  // 네이버 로그인 페이지 진입
	  driver.get("https://nid.naver.com/nidlogin.login?mode=form&url=https%3A%2F%2Fwww.naver.com");
	  
}

@AfterMethod
public void afterMethod() {
	  driver.quit();
}

 

 

위 코드에서 중요한 부분은 쿠키를 가져와서 저장하는 부분이다. 아래와 같은 방식으로 현재 세션의 쿠키를 가져와서 저장한다.

 

for (Cookie ck : driver.manage().getCookies()) {
			  bos.write((ck.getName() + ";" + ck.getValue() + ";" + ck.getDomain()
					  + ";" + ck.getPath() + ";" + ck.getExpiry() + ";" + ck.isSecure() ) );
			  bos.newLine();

 

코드를 실행하면, target 폴더에 browser.data 파일이 저장된다.

 

저장된 파일 확인

 

 

 

 

 

 

쿠키 로드하기

쿠키를 저장하는 과정을 성공했다면, 이제 저장한 쿠키를 가져와서 자동 로그인에 활용해야 한다. browser.data 파일에서 쿠키 정보를 로드해서 드라이버에 추가하자.

 

@Test
public void loadCookie() {
	  try {
			// 쿠키 파일을 로드
		  File dataFile = new File("./target/browser.data");
		  FileReader fr = new FileReader(dataFile);
		  BufferedReader br = new BufferedReader(fr);
		  String line;
		  
		  while ( (line = br.readLine()) != null) {
			  StringTokenizer str = new StringTokenizer(line, ";");
			  while (str.hasMoreTokens()) {
				  String name = str.nextToken();
				  String value = str.nextToken();
				  String domain = str.nextToken();
				  String path = str.nextToken();
				  Date expiry = null;
				  String dt;
				  if (!(dt = str.nextToken()).equals("null")) {
					  SimpleDateFormat formatter = 
							  new SimpleDateFormat("E MMM d HH:mm:ss z yyyy");
					  expiry = formatter.parse(dt);
				  }
				  
				  boolean isSecure = new Boolean(str.nextToken()).booleanValue();
				  
					Cookie ck = new Cookie(name, value, domain, path, expiry, isSecure);
				  
					// 드라이버에 쿠키 전달
					driver.manage().addCookie(ck);
			  }
		  }
		  
			// 네이버 프로필 수정 페이지로 이동 (로그인해야 정상 이동됨)
		  driver.get("https://nid.naver.com/user2/help/naverProfile");

			// 프로필 수정 페이지 우측 상단의 닉네임 비교해서 맞으면 pass
		  assertThat(driver.findElement(By.className("gnb_name"))
				  .getText()).isEqualTo("****");
	  } catch (Exception ex) {
		  ex.printStackTrace();
	  }
	  
}
@BeforeMethod
public void beforeMethod() {
	  System.setProperty("webdriver.chrome.driver", "./src/test/resources/drivers/chromedriver");
	  
	  ChromeOptions chromeOptions = new ChromeOptions();
	
	  driver = new ChromeDriver();
	  driver.get("http://www.naver.com/");
	  
}

@AfterMethod
public void afterMethod() {
	  driver.quit();
}

 

위 코드에서 중요한 부분은 가져온 쿠키 내용을 드라이버에 전달하는 부분이다.

 

Cookie ck = new Cookie(name, value, domain, path, expiry, isSecure);
				  
// 드라이버에 쿠키 전달
driver.manage().addCookie(ck);

 

 

네이버 로그인 쿠키를 가져온 다음에 프로필 수정 페이지로 이동하면, 이동이 되어야 한다. (로그인하지 않은 상태에서는 프로필 수정 페이지로 이동해도 로그인 페이지로 이동된다.)

driver.get("https://nid.naver.com/user2/help/naverProfile");

 

프로필 수정 페이지는 아래와 같다.

 

네이버 프로필 페이지

 

우측 상단의 네이버 닉네임을 비교하여, 내가 저장한 쿠키의 계정 정보와 같으면 제대로 된 쿠키 정보를 가져온 것이다.

// 프로필 수정 페이지 우측 상단의 닉네임 비교해서 맞으면 pass
assertThat(driver.findElement(By.className("gnb_name")).getText()).isEqualTo("****");

 

대부분의 서비스가 로그인 기반으로 이루어지다 보니 쿠키를 다루는 부분은 필수적으로 알아둬야 할 것 같다. 테스트 스크립트를 작성할 때 잘 활용해 보길 바란다.

 

 

 

 

함께보면 좋은 글

 

Selenium 묵시적 대기와 명시적 대기, NoSuchElementException

 

Selenium 브라우저 내비게이션 제어 , 뒤로가기 앞으로가기 새로고침