[파이썬] 웹 프레임워크를 이용한 유닛 테스트

04 Sep 2023

python

웹 개발을 하다 보면 테스트는 필수적인 작업입니다. 테스트를 통해 코드의 동작을 검증하고 버그를 예방할 수 있습니다. 이번 포스트에서는 파이썬 웹 프레임워크를 이용하여 유닛 테스트를 작성하는 방법에 대해 알아보겠습니다.

왜 유닛 테스트를 해야 할까요?

유닛 테스트는 일반적으로 작은 코드 한 단위를 테스트하는 것을 말합니다. 웹 개발에서는 주로 각각의 엔드포인트(endpoint)나 API 메서드를 유닛으로 테스트합니다. 유닛 테스트를 작성하는 이유는 다음과 같습니다:

버그 예방: 유닛 테스트를 통해 코드의 동작을 검증함으로써 버그가 발생할 가능성을 줄일 수 있습니다.
코드 품질 향상: 테스트 케이스를 작성함으로써 코드의 품질을 높일 수 있고, 리팩토링 작업의 안정성을 보장할 수 있습니다.
문서화: 테스트 케이스는 코드의 사용법과 예제를 담고 있기 때문에, 문서화된 형태로 사용될 수 있습니다.
재사용성: 테스트 케이스는 웹 애플리케이션을 수정하거나 확장할 때 유용하게 사용될 수 있습니다.

웹 프레임워크의 유닛 테스트 기능

웹 프레임워크는 종종 유닛 테스트를 작성할 수 있는 기능을 제공합니다. 대표적인 예로는 Django의 django.test, Flask의 flask.testing 등이 있습니다. 이러한 테스트 기능들은 테스트 서버를 생성하고 해당 서버에 요청을 보내는 등의 작업을 쉽게 수행할 수 있도록 도와줍니다.

Flask를 이용한 유닛 테스트 예제

이제 간단한 Flask 애플리케이션의 유닛 테스트를 작성해보겠습니다. 아래의 예제는 간단한 Hello, World! 메시지를 반환하는 엔드포인트를 가진 애플리케이션을 테스트하는 코드입니다.

import unittest
from flask import Flask
from flask_testing import TestCase

class HelloWorldTest(TestCase):

    def create_app(self):
        app = Flask(__name__)
        app.config['TESTING'] = True
        return app

    def test_hello_world(self):
        response = self.client.get('/')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data, b'Hello, World!')

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

이 예제에서는 unittest.TestCase를 상속받은 HelloWorldTest 클래스를 작성하였습니다. create_app 메서드를 오버라이딩하여 Flask 애플리케이션을 생성하고 app.config['TESTING'] 속성을 True로 설정하여 테스트 모드로 실행되도록 합니다.

test_hello_world 메서드에서는 self.client.get 메서드를 사용하여 테스트 서버에 / 엔드포인트에 GET 요청을 보냅니다. 그리고 response.status_code와 response.data를 검증하여 예상대로 동작하는지 확인합니다.

테스트를 실행하려면 스크립트 파일을 실행하거나 테스트 클래스를 지정하여 실행할 수 있습니다. 위 예제에서는 if __name__ == '__main__': unittest.main() 부분을 사용하여 테스트 클래스를 실행하도록 하였습니다.

마무리

웹 프레임워크를 이용하여 유닛 테스트를 작성하는 것은 코드의 품질 향상과 버그 예방에 큰 도움이 됩니다. 테스트 케이스를 작성하여 코드를 안정적이고 견고하게 만들어보세요!