unittest 모듈 사용하기 - Tirrilee/TechTalk GitHub Wiki

unittest 모듈의 사용 장점

• Python 표준 라이브러리
• 테스트 자동화
• 테스트 구조화
• 테스트 결과 보고

아래와 같이 unittest 라이브러리를 사용한 테스트 코드가 하는 일은 위에 스크립트형 코드와 같다.

function_tests.py

import unittest
from selenium import webdriver
import os

# 현재 디렉토리 경로를 불러온다
base_path = os.path.dirname(os.path.abspath(__file__))
class NewVisitorTest(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome(base_path+'/chromedriver')

    def tearDown(self):
        self.browser.quit()

    def test_can_start_a_list_and_retrieve_it_later(self):
        # 영애씨는 온라인 일정관리 앱을 알게 되어 홈페이지에 방문한다.
        self.driver.get('http://localhost:8000')

        # 홈페이지에 방문해 보니 제목이 "일정관리"인 것을 보고 홈페이지에 올바르게 방문한 것을 확인한다.
        # assert '일정관리' in self.driver.title, "Browser title was " + self.driver.title
        self.assertIn('일정관리', self.driver.title)
        self.fail('테스트 종료')

        # 일정을 입력할 수 있는 페이지로 바로 이동한다.

        # 영애씨는 생일날 미역국을 끓이기 위해 텍스트박스에 "시장에서 미역 사기"를 입력한다.
        # 영애씨가 엔터를 입력하면 페이지를 새로고침해서 모든 일정 목록을 보여준다.
        # "1: 시장에서 미역 사기"가 첫 번째 할일로 일정 목록에서 보여진다.

        # 영애씨는 추가로 할일 텍스트박스에 입력할 수 있고
        # "미역을 물에 불리기"라고 입력한다.

        # 다시 페이지를 새로고침해서 입력한 일정 두 가지 모두 목록에 표시한다.

        # 영애씨는 일정 목록이 사이트에 올바로 저장되었는지 궁금해서
        # 고유 URL 생성을 확인한다.

        # 영애씨는 URL을 방문하고 일정 목록이 올바르게 있음을 확인한다.

        # 영애씨는 이제 만족하고 잠을 자러간다.


if __name__ == '__main__':
    unittest.main(warnings='ignore')

모든 테스트는 unittest.TestCase 클래스를 상속하고 test_can_start_a_list_and_retrieve_it_later 같은 메소드를 추가하여 테스트를 실행한다. 이때 메소드의 이름이 test_문자열로 시작하는 것들은 runner가 테스트로 실행한다.

• setUp과 teatDown은 브라우저를 열고 닫는 부분을 처리한다.
• assert 대신에 self.assertIn으로 헬퍼 함수를 활용할 수 있다.

unittest 모듈을 assertEqual, assetTrue, assetFalse` 같은 여러 헬퍼 함수를 제공한다.

• self.fail 함수로 무조건적으로 테스트를 실패시킨다.
• warnings='ignore'은 파라미터로 너무 많은 경고 메세지가 출력되는 것을 방지한다.

실행 결과

$ python function_tests.py 
F
======================================================================
FAIL: test_can_start_a_list_and_retrieve_it_later (__main__.NewVisitorTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "function_tests.py", line 17, in test_can_start_a_list_and_retrieve_it_later
    assert '일정관리' in self.browser.title, "Browser title was " + self.browser.title
AssertionError: Browser title was Welcome to Django

----------------------------------------------------------------------
Ran 1 test in 3.960s

FAILED (failures=1)