[파이썬] Selenium 테스트 스위트 작성

06 Sep 2023

Selenium

Selenium은 웹 애플리케이션을 테스트하기 위한 강력하고 유연한 프레임워크입니다. 테스트 스위트(Test Suite)는 Selenium을 사용하여 여러 테스트 케이스를 번들로 그룹화한 것을 말합니다. 이제 Selenium을 사용하여 테스트 스위트를 작성하는 방법을 알아봅시다.

Selenium 설치

Selenium을 사용하기 위해서는 Python에서 Selenium 패키지를 설치해야 합니다. 다음 명령어를 사용하여 설치할 수 있습니다:

pip install selenium

테스트 스위트 구성

테스트 스위트는 여러 테스트 케이스를 그룹화하여 한 번에 실행할 수 있는 구조입니다. 테스트 스위트를 작성하려면 다음 단계를 따르세요:

필요한 라이브러리와 모듈을 가져옵니다.
테스트 케이스 클래스를 작성합니다.
테스트 케이스 클래스를 테스트 스위트에 추가합니다.

아래 예제는 Google 검색 기능을 테스트하는 테스트 스위트를 작성하는 방법을 보여줍니다.

# 1. 필요한 라이브러리와 모듈 가져오기
import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys

# 2. 테스트 케이스 클래스 작성하기
class GoogleSearchTest(unittest.TestCase):
    def setUp(self):
        # 웹 드라이버 초기화
        self.driver = webdriver.Chrome()

    def test_google_search(self):
        # Google 검색 페이지로 이동
        self.driver.get("https://www.google.com")

        # 검색어 입력
        search_box = self.driver.find_element_by_name("q")
        search_box.send_keys("Selenium 테스트 스위트 작성")

        # 엔터 키 입력
        search_box.send_keys(Keys.RETURN)

        # 검색 결과 확인
        self.assertIn("Selenium 테스트 스위트 작성", self.driver.title)

    def tearDown(self):
        # 웹 드라이버 종료
        self.driver.quit()

# 3. 테스트 스위트에 테스트 케이스 클래스 추가하기
suite = unittest.TestLoader().loadTestsFromTestCase(GoogleSearchTest)

# 테스트 실행
unittest.TextTestRunner(verbosity=2).run(suite)

위 예제에서는 unittest 모듈을 사용하여 테스트 스위트를 작성하고 실행합니다. setUp 메서드에서 웹 드라이버를 초기화하고 tearDown 메서드에서 종료하는 것에 주의하세요. 테스트 케이스는 test_로 시작하는 메서드로 작성되며, 검색어를 입력하고 검색 결과를 확인합니다.

테스트 스위트는 unittest.TestLoader().loadTestsFromTestCase 메서드를 사용하여 GoogleSearchTest 클래스를 가져와 작성합니다. 마지막으로 unittest.TextTestRunner(verbosity=2).run 메서드를 사용하여 테스트를 실행합니다.

Selenium을 사용하여 테스트 스위트를 작성하면 여러 테스트 케이스를 한 번에 실행하고, 웹 애플리케이션의 동작을 자동으로 확인할 수 있습니다.