Pytest-django: 여러 데이터베이스로 테스트 간 정리 지원

두 개의 데이터베이스를 사용하는 패치 버전을 사용하여 예제 테스트를 제공할 수 있습니까? 여러 데이터베이스가 사용되는 방식과 multi_db를 true로 설정한 다음 다른 데이터베이스에 대한 제어가 필요한 사항을 이해하는 데 도움이 될 것이라고 생각합니다.

DATABASES = {
    'default': {
        'ENGINE'   : 'django.db.backends.sqlite3',
        'NAME'     : 'default.db',
    },
    'operations': {
        'ENGINE'   : 'django.db.backends.sqlite3',
        'NAME'     : 'operations.db',
    },
}

대부분의 / 모든 모델은 두 번째 데이터베이스 "작업"과 연결되어 있습니다. Django의 TestCase는 기본 db에서 트랜잭션을 올바르게 플러시하지만 multi_db=True 아닌 한 다른 db에서는 플러시하지 않을 것이라고 생각합니다.

다음은 몇 가지 샘플 테스트입니다.

from collection.models import Foo

def test_one(db):
    Foo.objects.create()
    assert Foo.objects.count() == 1

def test_two(db):
    assert Foo.objects.count() == 0

다음은 위에서 설명한 패치된 db 고정 장치의 출력입니다.

$ py.test multi_db_test/ -vv
======================================= test session starts =======================================
platform linux2 -- Python 2.7.2 -- py-1.4.20 -- pytest-2.5.2 -- bin/python
plugins: django
collected 2 items

multi_db_test/test_foo.py:3: test_one PASSED
multi_db_test/test_foo.py:7: test_two PASSED

==================================== 2 passed in 5.75 seconds =====================================

다음은 pytest_django 자체 db 고정 장치의 출력입니다.

$ py.test multi_db_test/ -vv
======================================= test session starts =======================================
platform linux2 -- Python 2.7.2 -- py-1.4.20 -- pytest-2.5.2 -- bin/python
plugins: django
collected 2 items

multi_db_test/test_foo.py:3: test_one PASSED
multi_db_test/test_foo.py:7: test_two FAILED

============================================ FAILURES =============================================
____________________________________________ test_two _____________________________________________

db = None

    def test_two(db):
>       assert Foo.objects.count() == 0
E       assert 1 == 0
E        +  where 1 = <bound method Manager.count of <django.db.models.manager.Manager object at 0x3cd0890>>()
E        +    where <bound method Manager.count of <django.db.models.manager.Manager object at 0x3cd0890>> = <django.db.models.manager.Manager object at 0x3cd0890>.count
E        +      where <django.db.models.manager.Manager object at 0x3cd0890> = Foo.objects

multi_db_test/test_foo.py:8: AssertionError
=============================== 1 failed, 1 passed in 5.85 seconds ===============================

따라서 간단히 보면 multi_db 고정 장치와 pytest.mark.django_db(multi=True) 가 API로 작동할 수 있는 것처럼 보입니다. 더 나은 아이디어가 있는 사람이 있습니까?

합리적으로 들립니다. 더 좋든 그렇지 않든 나는 확실히 다른 아이디어가 없습니다.

여보세요. 이 문제 외에 multi_db 지원에 대한 논의가 있습니까? 문서 에는 연락해야 한다고 나와 있지만 현재 상태는 어떻습니까?

@slafs 이 외에 다중 데이터베이스 지원에 대한 논의/요청이 많지 않습니다.

multi_db=True 를 전달하는 것은 매우 간단하지만 이에 대한 일종의 테스트도 필요하므로 보조 데이터베이스 설정도 필요합니다. 누구든지 이 작업을 수행할 의향이 있다면 훌륭할 것입니다. :)

시작 게시물에서 @ftobia가 언급한 해결 방법을 사용했지만 1.7에서 Django 1.8.4로 업그레이드한 후 작동이 중지되었습니다.

setup/tearDown을 처리하는 코드가 인스턴스 속성 대신 multi_db 클래스 속성을 찾도록 Django가 변경 되었습니다.

@ftobia 의 해결 방법을 다음으로 변경하면

case = TestCase(methodName='__init__')
case.__class__.multi_db = True
case._pre_setup()
request.addfinalizer(case._post_teardown)
request.addfinalizer(_django_cursor_wrapper.disable)

+1:+1:

이것은 Django 테스트 러너에서 pytest로 전환하는 것을 막고 있습니다.

기본적으로 https://github.com/pytest-dev/pytest-django/blob/master/pytest_django/fixtures.py#L107에서 django_case.multi_db = True 사용하지 않는 이유는 무엇입니까?

+1

+1

django_db 마커에 마커가 있으면 좋은 API가 될 것이라고 생각합니다.

@pytest.mark.django_db(transactional=True, multi_db=True)
def test_a():
    pass

구현은 reset_sequences (PR #308) 또는 serialized_rollback (PR #353)와 비교적 유사해야 합니다.

"어려운" 부분은 여러 데이터베이스를 포함하도록 pytest-django의 내부 테스트 제품군을 개선하는 것입니다.

누군가이 작업을 원하면 확실히 검토하고 병합하는 데 도움이 될 것입니다. 이것이 당신에게 흥미로울 것이라면 PR 작업을 시작하십시오!

https://github.com/pytest-dev/pytest-django/pull/397#issuecomment -261987751에 몇 가지 의견을 남기고 일반적인 방식으로 변경할 수 있는 PR을 만들었습니다: https://github. com/pytest-dev/pytest-django/pull/431.com/pytest-dev/pytest-django/pull/431.com/pytest-dev/pytest-django/pull/431

임시 솔루션을 찾는 사람들을 위해 multi_db를 활성화하는 또 다른 옵션입니다. 이것을 conftest.py에 넣으십시오.

def pytest_sessionstart(session):
    from django.test import TransactionTestCase
    TransactionTestCase.multi_db = True

해결 방법을 찾는 다른 사람을 위해 multi_db 는 Django 2.2에서 더 이상 사용되지 않습니다. 나를 위해 일한 교체는 다음과 같습니다.

TransactionTestCase.databases = set(settings.DATABASES.keys())

@jcushman 사용한 전체 고정 장치나 코드 조각을 공유해도 될까요?

이유가 무엇이든 우리 기지 conftest.py pytest_sessionstart() conftest.py 은(는) 아무 일도 하지 않았습니다. 표준 pytest monkeypatch 과 함께 #342에서 페이지를 가져 와서 다중 데이터베이스 사용을 위해 다음을 방문했습니다.

# OPTION 1: Function fixture; must be included with tests
@pytest.fixture
def django_db_multiple(monkeypatch, request, settings):
    """
    Ensure all test functions using Django test cases have multiple database
    support. This is mostly/only so that Django will wrap ALL database use with
    atomic blocks like it does for DEFAULT_DB_ALIAS.

    https://github.com/django/django/blob/master/django/test/testcases.py#L903
    https://github.com/pytest-dev/pytest-django/issues/76
    """
    from django.test import TestCase
    from django.test import TransactionTestCase

    db_keys = set(settings.DATABASES.keys())

    monkeypatch.setattr(TestCase, 'databases', db_keys)
    monkeypatch.setattr(TransactionTestCase, 'databases', db_keys)

@pytest.mark.django_db
def test_some_test(django_db_multiple):
    pass

# OPTION 2: Session fixture
@pytest.fixture(autouse=True, scope='session')
def django_db_multiple():
    """
    Ensure all test functions using Django test cases have multiple database
    support. This is mostly/only so that Django will wrap ALL database use with
    atomic blocks like it does for DEFAULT_DB_ALIAS.

    https://github.com/django/django/blob/master/django/test/testcases.py#L903
    https://github.com/pytest-dev/pytest-django/issues/76
    https://github.com/pytest-dev/pytest/issues/1872
    """
    from _pytest.monkeypatch import MonkeyPatch
    from django.test import TestCase
    from django.test import TransactionTestCase
    from django.conf import settings

    db_keys = set(settings.DATABASES.keys())

    monkeypatch = MonkeyPatch()
    monkeypatch.setattr(TestCase, 'databases', db_keys)
    monkeypatch.setattr(TransactionTestCase, 'databases', db_keys)

    yield monkeypatch

    monkeypatch.undo()

_편집: 세션 고정 장치로 이동했습니다._

unittest Django 테스트 케이스로 전환하여 pytest 초능력을 잃지 않기 위해 pytest-django 내부를 이 못생긴(아직 작동하는!) 해킹에 복사하고 패치했습니다.

from typing import Optional, Type, TypeVar

import pytest
from django.test import TransactionTestCase
from pytest_django.django_compat import is_django_unittest

TestCase = TypeVar('TestCase', bound=TransactionTestCase)


def _django_db_fixture_helper(
    request, django_db_blocker, transactional: bool = False, reset_sequences: bool = False,
) -> Optional[Type[TestCase]]:
    if is_django_unittest(request):
        return None

    if not transactional and 'live_server' in request.fixturenames:
        # Do nothing, we get called with transactional=True, too.
        return None

    django_db_blocker.unblock()
    request.addfinalizer(django_db_blocker.restore)

    if transactional:
        from django.test import TransactionTestCase as DjangoTestCase  # noqa: WPS433

        if reset_sequences:

            class ResetSequenceTestCase(DjangoTestCase):  # noqa: WPS431
                reset_sequences = True

            DjangoTestCase = ResetSequenceTestCase  # type: ignore[misc] # noqa: N806
    else:
        from django.test import TestCase as DjangoTestCase  # type: ignore[no-redef] # noqa: WPS433

    return DjangoTestCase  # type: ignore[return-value]


@pytest.fixture()
def db_case(request, django_db_setup, django_db_blocker):
    """Require a django test database.

    This database will be setup with the default fixtures and will have
    the transaction management disabled. At the end of the test the outer
    transaction that wraps the test itself will be rolled back to undo any
    changes to the database (in case the backend supports transactions).
    This is more limited than the ``transactional_db`` resource but
    faster.

    If multiple database fixtures are requested, they take precedence
    over each other in the following order (the last one wins): ``db``,
    ``transactional_db``, ``django_db_reset_sequences``.
    """
    if 'django_db_reset_sequences' in request.fixturenames:
        request.getfixturevalue('django_db_reset_sequences')
    if (
        'transactional_db' in request.fixturenames
        or 'live_server' in request.fixturenames
    ):
        request.getfixturevalue('transactional_db')
    else:
        django_case: Optional[Type[TransactionTestCase]] = _django_db_fixture_helper(
            request, django_db_blocker, transactional=False,
        )

        def factory(dbs=None):  # noqa: WPS430
            if django_case is None:
                return
            CaseType: Type[TransactionTestCase] = django_case  # noqa: N806
            if dbs is not None:
                class DatabasesSetTestCase(  # noqa: WPS431
                    CaseType,  # type: ignore[valid-type, misc]
                ):
                    databases = dbs
                CaseType = DatabasesSetTestCase  # noqa: N806
            test_case: TransactionTestCase = CaseType(methodName='__init__')
            test_case._pre_setup()  # type: ignore[attr-defined] # noqa: WPS437
            request.addfinalizer(
                test_case._post_teardown,  # type: ignore[attr-defined] # noqa: WPS437
            )

        return factory

그것으로 다음과 같은 것을 사용할 수 있습니다.

class TestCase(object):

    def test_ok(
        self,
        db_case,
        ...
    ):
        db_case(dbs=('non_default_db_alias',))

이것은 db 고정장치나 pytest.mark.django_db 마크 대신에 사용되어야 합니다.

@jcushman 사용한 전체 고정 장치나 코드 조각을 공유해도 될까요?

다른 사람이 이에 부딪칠 경우를 대비하여 답장을 추가합니다.

conftest.py 아래 코드를 추가해야 합니다. 이 고정 장치는 자동으로 선택되므로 테스트에 추가할 필요가 없습니다.

```파이
def pytest_sessionstart(세션):
django.test에서 TransactionTestCase 가져오기

TransactionTestCase.databases = set(settings.DATABASES.keys())

````

따라서 이 문제는 적절한 다중 DB 지원으로 귀결된다고 생각합니다. 나는 이 문제의 중복으로 이 문제를 닫을 것입니다.