[python] 테스트 환경 복원을 위한 setUp()과 tearDown() 사용법

15 Dec 2023

python

안녕하세요! 이번 글에서는 파이썬의 unittest 모듈에서 테스트 환경을 구성하고 청소하는 데 사용되는 setUp()과 tearDown() 메서드에 대해 알아보겠습니다.

1. setUp() 메서드

setUp() 메서드는 테스트를 수행하기 전에 필요한 환경을 설정하는 데 사용됩니다. 예를 들어, 데이터베이스 연결을 설정하거나 파일을 읽어들이는 등의 초기화 작업을 수행할 수 있습니다.

import unittest

class MyTest(unittest.TestCase):
    def setUp(self):
        # 데이터베이스 연결 설정
        self.db = create_db_connection()
        
        # 기타 초기화 작업
        # ...

2. tearDown() 메서드

tearDown() 메서드는 테스트 이후에 사용한 자원을 정리하는 데 사용됩니다. 예를 들어, 데이터베이스 연결을 닫거나 파일을 삭제하는 등의 작업을 수행할 수 있습니다.

import unittest

class MyTest(unittest.TestCase):
    def tearDown(self):
        # 데이터베이스 연결 닫기
        self.db.close()
        
        # 기타 정리 작업
        # ...

3. 사용 예시

다음은 setUp()과 tearDown()을 사용하여 테스트 환경을 설정하고 청소하는 예시입니다.

import unittest

class MyTest(unittest.TestCase):
    def setUp(self):
        # 데이터베이스 연결 설정
        self.db = create_db_connection()
        
    def test_something(self):
        # 테스트 코드 작성
        # ...

    def tearDown(self):
        # 데이터베이스 연결 닫기
        self.db.close()

setUp()과 tearDown() 메서드를 사용하면, 테스트를 실행할 때마다 반복적으로 환경을 설정하고 정리하는 번거로움을 줄일 수 있습니다.

이상으로, setUp()과 tearDown() 메서드의 사용법에 대해 알아보았습니다. 효율적인 테스트 환경 구성을 위해 이들 메서드를 적절히 활용해보세요!