Pytest-Django에서 관리되지 않는 모델 처리

관리되지 않는 모델 테스트의 과제

Django 프로젝트에서는 관리되지 않는 모델, 즉 메타 옵션에 Managed = True가 설정되지 않은 모델이 가끔 있습니다. 특히 테스트 설정에 관리형 모델과 비관리형 모델이 혼합되어 있거나 여러 데이터베이스(예: 관리형 모델이 있는 모델과 비관리형 모델이 있는 데이터베이스)가 포함된 경우 이러한 모델은 테스트를 까다롭게 만들 수 있습니다.

이 블로그 게시물에서는 pytest-django를 사용하여 관리되지 않는 모델을 테스트하는 접근 방식을 탐색하고 이러한 시나리오를 효과적으로 관리하는 데 도움이 되는 장단점 및 해결 방법을 강조합니다.

접근 방식 1: 모든 모델을 관리됨으로 표시

테스트 중에 관리되지 않는 모델을 처리하는 간단한 방법 중 하나는 일시적으로 관리됨으로 표시하는 것입니다. 방법은 다음과 같습니다.

# Add this to conftest.py
@pytest.hookimpl(tryfirst=True)
def pytest_runtestloop():
    from django.apps import apps
    unmanaged_models = []
    for app in apps.get_app_configs():
        unmanaged_models += [m for m in app.get_models()
                             if not m._meta.managed]
    for m in unmanaged_models:
        m._meta.managed = True

참고: 이 접근 방식이 작동하려면 pytest 설정(또는 pytest.ini)에 --no-migrations 옵션을 추가해야 합니다

참고: 스택 오버플로

장점:

• 구현이 간단합니다.

단점:

• 여러 개발자가 동일한 프로젝트에서 작업할 때 문제를 일으킬 수 있는 마이그레이션 테스트를 건너뜁니다.

접근 방식 2: 관리되지 않는 모델을 수동으로 생성

또는 테스트 설정 중에 관리되지 않는 모델을 수동으로 생성할 수도 있습니다. 이 접근 방식을 사용하면 마이그레이션을 테스트할 수 있습니다.

@pytest.fixture(scope="session", autouse=True)
def django_db_setup(django_db_blocker, django_db_setup):
    with django_db_blocker.unblock():
        for _connection in connections.all():
            with _connection.schema_editor() as schema_editor:
                setup_unmanaged_models(_connection, schema_editor)
        yield

def setup_unmanaged_models(connection, schema_editor):
    from django.apps import apps

    unmanaged_models = [
        model for model in apps.get_models() if model._meta.managed is False
    ]
    for model in unmanaged_models:
        if model._meta.db_table in connection.introspection.table_names():
            schema_editor.delete_model(model)
        schema_editor.create_model(model)

장점:

• 테스트 사례의 일부로 마이그레이션을 테스트합니다.

단점:

• 약간 더 복잡합니다.
• transaction=True는 이 접근 방식에서 작동하지 않습니다(다음 섹션에서 설명).

트랜잭션 테스트 이해

Pytest-django는 데이터베이스 고정 장치인 django_db 및 django_db(transaction=True)를 제공합니다. 차이점은 다음과 같습니다.

django_db: 테스트 사례가 끝나면 변경 사항을 롤백합니다. 즉, 데이터베이스에 대한 실제 커밋이 이루어지지 않습니다.

django_db(transaction=True): 변경 사항을 커밋하고 각 테스트 사례 후에 데이터베이스 테이블을 자릅니다. 모든 테스트 후에는 관리되는 모델만 잘리므로 트랜잭션 테스트 중에 관리되지 않는 모델을 특별하게 처리해야 하는 이유가 바로 여기에 있습니다.

테스트 사례 예시

@pytest.mark.django_db
def test_example():
    # Test case logic here
    pass

@pytest.mark.django_db(transaction=True)
def test_transactional_example():
    # Test case logic here
    pass

관리되지 않는 모델에서 트랜잭션 테스트가 작동하도록 만들기

트랜잭션 테스트는 관리되는 모델만 자르기 때문에 테스트 실행 중에 관리되지 않는 모델을 관리하도록 수정할 수 있습니다. 이렇게 하면 잘림에 포함됩니다.

# Add this to conftest.py
@pytest.hookimpl(tryfirst=True)
def pytest_runtestloop():
    from django.apps import apps
    unmanaged_models = []
    for app in apps.get_app_configs():
        unmanaged_models += [m for m in app.get_models()
                             if not m._meta.managed]
    for m in unmanaged_models:
        m._meta.managed = True

on_commit 후크를 사용하여 transaction=True 방지(가능한 경우)

on_commit 후크와 관련된 시나리오에서는 pytest-django(>= v.4.4)의 고정 장치 django_capture_on_commit_callbacks를 사용하여 on_commit 콜백을 직접 캡처하고 실행함으로써 트랜잭션 테스트 사용을 피할 수 있습니다.

@pytest.fixture(scope="session", autouse=True)
def django_db_setup(django_db_blocker, django_db_setup):
    with django_db_blocker.unblock():
        for _connection in connections.all():
            with _connection.schema_editor() as schema_editor:
                setup_unmanaged_models(_connection, schema_editor)
        yield

def setup_unmanaged_models(connection, schema_editor):
    from django.apps import apps

    unmanaged_models = [
        model for model in apps.get_models() if model._meta.managed is False
    ]
    for model in unmanaged_models:
        if model._meta.db_table in connection.introspection.table_names():
            schema_editor.delete_model(model)
        schema_editor.create_model(model)

참고자료

• pytest-django 문서
• 스택 오버플로: 관리되지 않는 모델 테스트

관리되지 않는 모델을 처리하기 위한 다른 접근 방식이나 팁이 있나요? 아래 댓글로 공유해주세요!

위 내용은 Pytest-Django에서 관리되지 않는 모델 처리의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!