流控制和失败：不允许数据库访问，使用“django_db”....错误

Question

我的芹菜任务之一遇到了一些奇怪的行为。

def run_single_test(test_name_or_decorator):
    # get dict of test names, test paths
    test_dict = get_single_test_names()

    # check to see if test is in the dict
    if test_name_or_decorator not in test_dict:
        return 'The requested test could not be found.'

    for test_name, test_path in test_dict.items():
        # if test name is valid run associated test
        if test_name == test_name_or_decorator:
            pytest.main(['-p', 'no:django','--json-report', test_path])
            report = return_test_result_json('.report.json')
            report_id = str(uuid.uuid4())
            test_run_data = TestResults.objects.create(name=report_id, data=report)

    return 'this is your test report: {}'.format(get_report(test_run_data.id))

此任务将执行 pytest.main() 命令并运行测试，但是当它使用 .create() 将 .report.json 插入 Db 时，我收到以下错误：

Failed: Database access not allowed, use the "django_db" mark, or the "db" or "transactional_db" fixtures to enable it.

现在，如果我将 for test_name.... 块中的所有功能简化并将其移至自己的功能中，一切正常：

def run_single_test_path():

    test_path = 'test_folder/test_file.py::TestClass::specific_test_name'
    pytest.main(['-p', 'no:django','--json-report', test_path])
    report = return_test_result_json('.report.json')
    report_id = str(uuid.uuid4())
    test_run_data = TestResults.objects.create(name=report_id, data=report)
    return 'this is your test report: {}'.format(get_report(test_run_data.id))

我得到了预期的回报：

"this is your test report: {'created': datetime.datetime(2018, 9, 27, 15, 51, 59, 297991, tzinfo=<UTC>), 'summary': {'total': 1, 'passed': 1}, 'exitcode': 0}"

这种行为不会发生在我使用相同 pytest.main() 和 .create() 变体的其他任务中。 另一个观察结果是，如果我首先运行此任务并引发数据库访问错误，则所有其他任务将失败并出现相同的错误。

我的工作理论是关于 for 循环或 if 的某些东西导致 .create() 吓坏了，但我完全不知道为什么。

Answer 1

一般来说，pytest.main 不关心测试运行后留下的全局状态，因为它假定进程在测试运行完成后以返回的退出代码终止。因此，如果您在pytest.main 完成后在同一进程中运行代码，则必须小心范围内的修改。其中之一是pytest-django在django测试配置上激活的数据库拦截器；测试执行完成后它不会被停用。要停用，请在代码中明确执行：

import pytest_django

def run_single_test(test_name_or_decorator):
    ...
    pytest.main(['-p', 'no:django','--json-report', test_path])
    pytest_django.plugin._blocking_manager.unblock()

    # database access unblocked, you can run the query now
    test_run_data = TestResults.objects.create(name=report_id, data=report)

或添加一个隐式执行的运行后挂钩，例如：

# conftest.py

import pytest_django

def pytest_unconfigure(config):
    pytest_django.plugin._blocking_manager.unblock()