[PYTHON] 통합 테스트(Integration Test) 시 데이터베이스 상태 관리 3가지 해결 방법과 차이점

통합 테스트 (Integration Test)

 

파이썬 애플리케이션 개발에서 통합 테스트(Integration Test)는 단위 테스트만큼이나 중요합니다. 특히 데이터베이스(DB)와 상호작용하는 로직을 검증할 때, 가장 큰 난관은 바로 '데이터의 일관성(State Management)'을 어떻게 유지하느냐입니다. 테스트가 실행될 때마다 DB 상태가 변하면 테스트 결과가 달라지는 '플래키 테스트(Flaky Test)'가 발생하기 때문입니다. 본 포스팅에서는 전문 백엔드 엔지니어의 시각에서 파이썬 통합 테스트 시 DB 상태를 깨끗하게 유지하고 관리하는 3가지 핵심 전략과 구체적인 해결 방법을 심층적으로 분석합니다.

---

1. 통합 테스트에서 DB 상태 관리가 필요한 이유

단위 테스트와 달리 통합 테스트는 실제 DB(또는 테스트용 DB)와 연결되어 쿼리를 수행합니다. 이때 상태 관리가 제대로 되지 않으면 다음과 같은 02가지 치명적인 문제가 발생합니다.

• 테스트 간 오염: A 테스트에서 생성한 데이터가 B 테스트의 조회 결과에 영향을 주어 오류를 유발합니다.
• 멱등성 상실: 테스트를 두 번 실행했을 때, 중복 키 오류 등으로 인해 결과가 달라집니다.

---

2. DB 상태 관리 전략 3가지 비교 및 차이

테스트 환경에 따라 적합한 전략이 다릅니다. 각 방식의 메커니즘과 장단점을 표로 비교해 보겠습니다.

구분	트랜잭션 롤백 (Transaction Rollback)	테이블 초기화 (Truncate/Cleanup)	도커 컨테이너 (Docker-based)
핵심 원리	테스트 종료 후 Rollback 수행	테스트 전/후 모든 데이터 삭제	매 테스트마다 새 컨테이너 실행
수행 속도	매우 빠름	보통	느림 (Cold Start 발생)
격리 수준	높음 (논리적 격리)	중간 (물리적 잔재 가능성)	매우 높음 (완벽한 물리 격리)
주요 단점	Nested Transaction 처리가 복잡함	외래 키 제약 조건 관리가 번거로움	CI/CD 리소스 소모가 큼

---

3. 구체적인 해결 방법과 Sample Example

해결 01: SQLAlchemy의 Session을 활용한 트랜잭션 롤백

파이썬 진영에서 가장 선호되는 방식입니다. `pytest`의 `fixture`를 사용하여 테스트 시작 시 트랜잭션을 열고, 테스트가 끝나면 실제 DB에 반영하지 않고 무조건 롤백합니다.

import pytest
from sqlalchemy import create_mock_engine
from my_app.database import SessionLocal, Base, engine

@pytest.fixture(scope="function")
def db_session():
    # 1. 테스트 전용 트랜잭션 시작
    connection = engine.connect()
    transaction = connection.begin()
    session = SessionLocal(bind=connection)

    yield session  # 테스트 수행

    # 2. 테스트 종료 후 롤백 (상태 복구)
    session.close()
    transaction.rollback()
    connection.close()

def test_create_user(db_session):
    # 실제 DB에 기록되는 것 같지만, 종료 후 사라짐
    new_user = User(name="Chaewon")
    db_session.add(new_user)
    db_session.commit() # 실제로는 내부 트랜잭션만 커밋됨
    
    assert db_session.query(User).count() == 1

해결 02: Database-Cleaner 패턴 (Truncate 방식)

비동기 드라이버를 사용하거나 트랜잭션 제어가 까다로운 경우, 매 테스트 시작 전 모든 테이블의 데이터를 비우는 방식입니다. `factory-boy`와 같은 도구와 병행하면 매우 강력합니다.

# 팁: PostgreSQL 사용 시 모든 테이블을 한 번에 비우는 SQL 예시
# TRUNCATE TABLE users, orders, products RESTART IDENTITY CASCADE;

해결 03: Testcontainers를 이용한 환경 격리

로컬 DB가 아닌, 테스트 실행 시점에만 살아있는 임시 DB 컨테이너를 띄우는 방법입니다. 가장 확실한 해결 방법이지만 리소스를 많이 사용합니다.

---

4. 전문 지식: 성능 최적화를 위한 04가지 팁

• In-Memory SQLite 활용: 복잡한 DB 기능(Window Function 등)을 쓰지 않는다면 `sqlite:///:memory:`가 가장 빠릅니다.
• Fixture Scope 조정: 테이블 생성(`create_all`)은 `session` 레벨이 아닌 `module` 레벨에서 한 번만 수행하세요.
• Parallel Testing 주의: `pytest-xdist`로 병렬 테스트 시, 동일 DB를 바라보면 상태가 꼬입니다. 프로세스별로 다른 DB 스키마를 할당해야 합니다.
• Migration 검증: 테스트 DB 구축 시 단순 `create_all` 대신 `Alembic` 등의 마이그레이션 도구를 직접 실행하여 운영 환경과 동일한 구조를 유지하세요.

---

5. 내용의 출처 및 참고 문헌

• SQLAlchemy 2.0 Documentation: "Using Transactions and Connection Pools".
• Pytest Documentation: "Fixtures as Function Arguments".
• "Test-Driven Development with Python" by Harry Percival (O'Reilly).
• Testcontainers for Python project (github.com/testcontainers/testcontainers-python).