[파이썬] Django를 이용한 테스트 코드 작성

01 Sep 2023

python

Django는 파이썬 기반의 웹 애플리케이션 프레임워크로, 테스트 주도 개발(Test-Driven Development, TDD)을 지원합니다. 테스트 코드를 작성하여 애플리케이션의 기능을 확실히 검증하고 버그를 방지하는 것은 매우 중요합니다. 이번 포스트에서는 Django에서의 테스트 코드 작성에 대해 알아보겠습니다.

테스트 환경 설정

Django는 django.test 모듈을 통해 테스트 기능을 제공합니다. 테스트를 작성하려면, 테스트용 데이터베이스를 설정하고, django.test.TestCase를 상속받는 테스트 케이스 클래스를 생성해야 합니다. 테스트 시작 전에 필요한 데이터를 미리 생성하는 등의 작업을 할 수 있습니다.

from django.test import TestCase

class MyTestCase(TestCase):
    def setUp(self):
        # 테스트 환경 설정 작업
        pass

    def test_something(self):
        # 테스트 코드 작성
        pass

    def tearDown(self):
        # 테스트 환경 정리 작업
        pass

테스트 메소드 작성

TestCase 클래스 내에서는 테스트 메소드를 작성할 수 있습니다. 각 테스트 메소드는 test_로 시작해야 합니다. 테스트 메소드는 테스트할 기능을 호출하고, 예상 결과와 실제 결과를 비교하여 테스트를 수행합니다.

def test_addition(self):
    result = add(2, 3)
    self.assertEqual(result, 5)

def test_subtraction(self):
    result = subtract(5, 3)
    self.assertEqual(result, 2)

Assertion 메소드 사용

TestCase 클래스는 다양한 Assertion 메소드를 제공하여 예상 결과와 실제 결과를 비교합니다. 몇 가지 자주 사용되는 Assertion 메소드를 살펴보겠습니다.

assertEqual(a, b): a와 b가 동일한지 확인합니다.
assertTrue(x): x가 True인지 확인합니다.
assertFalse(x): x가 False인지 확인합니다.
assertIs(a, b): a와 b가 동일한 객체인지 확인합니다.
assertIn(a, b): a가 b에 속하는지 확인합니다.

테스트 실행

python manage.py test 명령어로 Django 프로젝트 내에서 테스트를 실행할 수 있습니다. 이때, Django는 tests.py 파일을 찾아 테스트 코드를 실행합니다. 자세한 테스트 실행 결과와 각 테스트의 성공여부를 확인할 수 있습니다.

$ python manage.py test

마무리

Django에서 테스트 코드를 작성함으로써 애플리케이션의 기능을 검증하는데 있어 많은 이점을 얻을 수 있습니다. 테스트 주도 개발은 안정적이고 견고한 애플리케이션을 만들기 위한 필수적인 접근 방법입니다. Django에서 제공하는 테스트 기능을 적절히 활용하여 애플리케이션의 품질을 높이도록 노력해보세요.