Selenium: 팝업 창 컨트롤하기

사이트에 따라, 로그인을 하기 위해 팝업 창에서 로그인 정보를 입력하는 케이스들이 있다. 따라서 현재 드라이버에서 팝업창으로 접근하기 위한 방법과 이를 적용하면, 쉽게 컨트롤할 수 있다.

 

구현 하기

from selenium import webdriver


class DefaultTest:
    def __init__(self, address: str):
        self.options = webdriver.ChromeOptions()
        # skip ssl
        self.options.add_argument("--ignore-certificate-errors")
        self.driver = webdriver.Chrome(chrome_options=self.options)
        self.address = address

    def teardown(self):
        self.driver.quit()

    def change_window(self, target: str):
        if target == 'parent':
            # child window close
            self.driver.close()
            self.driver.switch_to.window(self.driver.window_handles[0])
        elif target == 'child':
            self.driver.switch_to.window(self.driver.window_handles[1])
        else:
            print("Wrong target!")

    def auto_test(self):
        self.driver.get(self.address)
        self.driver.set_window_size(974, 1040)


if __name__ == '__main__':
    test_address = "192.168.0.1"
    test = DefaultTest(test_address)
    test.auto_test()

 위와 같이 `chage_window`라는 함수를 작성하면 팝업 창을 쉽게 컨트롤 할 수 있다. 예를 들어 테스트하는 페이지에 접근 후 팝업 창이 발생한다면, `change_window("child")`를 호출하게 되면 현재 driver가 탐색하는 윈도우는 팝업 창이 된다. 따라서, 팝업 창이 실행된다면 `window_handles[1]`과 같이 리스트로 driver가 탐색할 수 있는 페이지가 추가된다.

 

 팝업 창에 대한 처리가 끝난다면, driver가 탐색하는 윈도우를 메인 윈도우로 변경하여야 한다. 이는 처리하고자 하는 로직에 따라 상이할 수 있다.  즉, 팝업 창에 대한 처리는 `window_handles`와 `switch_to.window`를 적절히 활용하면 쉽게 컨트롤할 수 있다.

 

간단한 테스트 하기

정상적으로 실행 될 경우 웹페이지 화면

from selenium import webdriver
from selenium.webdriver.common.by import By
from time import sleep

class DefaultTest:
    def __init__(self, address: str):
        self.driver = webdriver.Chrome()
        self.address = address

    def teardown(self):
        self.driver.quit()

    def change_window(self, target: str):
        if target == 'parent':
            # child window close
            self.driver.close()
            self.driver.switch_to.window(self.driver.window_handles[0])
        elif target == 'child':
            self.driver.switch_to.window(self.driver.window_handles[1])
        else:
            print("Wrong target!")

    def auto_test(self):
        self.driver.get(self.address)
        self.driver.set_window_size(974, 1040)
        self.driver.find_element(By.XPATH, "/html/body/p/a").click()
        self.change_window("child")
        self.driver.find_element(By.XPATH, '/html/body/form/table/tbody/tr[6]/td[2]/input').click()
        # 결과를 보기 위해 5초간 sleep
        sleep(5)
        self.change_window("parent")


if __name__ == '__main__':
    test_address = "http://demo.guru99.com/popup.php"
    test = DefaultTest(test_address)
    test.auto_test()
    test.teardown()

 `http://demo.guru99.com/popup.php`이라는 샘플 페이지가 있길래, 이 페이지를 통해 팝업 테스트를 진행해 보았다. 정상적으로 실행이 된다면, 팝업 창(새 창)이 실행된 후에 위와 같이 `Submit`을 클릭 후 실행한 모든 창을 종료한다.