在Django Admin自定义动作中调用pytest.main报错问题排查

为什么pytest.main会触发数据库错误？ #

当你在Django Admin的请求处理函数里直接调用pytest.main时，当前进程已经处于Django的请求生命周期中——Django已经初始化了数据库连接、配置了ORM环境。而pytest-django的核心机制之一，是在测试运行时对数据库访问做安全限制：默认情况下，它会阻止未标记@pytest.mark.django_db或未使用数据库fixture的测试访问数据库。

但这里的问题是，即使你的测试只是assert True，pytest-django的数据库检查逻辑依然会触发。因为当前进程已经有活跃的Django数据库上下文，pytest-django无法区分这是测试环境还是生产请求环境，直接触发了Database access not allowed的保护机制——它误以为你在测试里未经允许访问了数据库。

而通过CLI或subprocess运行测试时，是启动了全新的独立进程，pytest-django会在这个进程里从头初始化Django的测试环境（包括创建测试数据库、配置隔离的连接），和原Django Admin的进程完全隔离，所以不会有冲突。

pytest.main的使用错误 #

1. 环境未隔离：直接在Django请求进程内运行pytest，导致pytest-django的测试环境和当前请求的Django环境冲突，触发数据库安全检查。
2. stdout重写方式不当：直接替换sys.stdout = StringIO()会干扰pytest自身的输出捕获逻辑，虽然能拿到结果，但可能引发其他潜在问题（比如某些pytest插件的输出异常）。

方案1：调整pytest运行参数，跳过数据库检查 #

如果你的测试确实不需要访问数据库，可以在pytest.main的参数里添加--no-db，强制pytest-django跳过数据库初始化和检查：

def run_selected_test_cases(self, request, queryset):
    from io import StringIO
    import sys
    import pytest

    # 临时保存原stdout
    original_stdout = sys.stdout
    sys.stdout = capture_output = StringIO()
    
    try:
        pytest.main([
            "tests/functional/test_user_cases.py",
            "-vv",
            "--no-db"  # 跳过数据库相关初始化和检查
        ])
        result = capture_output.getvalue()
    finally:
        # 恢复原stdout
        sys.stdout = original_stdout
    
    response = HttpResponse(result, content_type="text/plain")
    response["Content-Disposition"] = f"attachment; filename=pytest_result.txt"
    return response

方案2：在独立进程内运行pytest.main #

如果需要保留数据库测试能力，可以用multiprocessing启动一个独立进程运行pytest，避免和当前请求的Django环境冲突：

from multiprocessing import Process, Queue
import pytest

def run_pytest(queue):
    import sys
    from io import StringIO
    original_stdout = sys.stdout
    sys.stdout = capture_output = StringIO()
    try:
        pytest.main([
            "tests/functional/test_user_cases.py",
            "-vv"
        ])
        queue.put(capture_output.getvalue())
    finally:
        sys.stdout = original_stdout

def run_selected_test_cases(self, request, queryset):
    queue = Queue()
    p = Process(target=run_pytest, args=(queue,))
    p.start()
    p.join()
    result = queue.get()
    
    response = HttpResponse(result, content_type="text/plain")
    response["Content-Disposition"] = f"attachment; filename=pytest_result.txt"
    return response

这种方式模拟了subprocess的隔离效果，但比subprocess轻量一些。

方案3：保留subprocess并优化速度 #

如果subprocess是可行方案但速度慢，可以尝试以下优化：

• 使用--tb=short或--tb=none减少pytest的输出内容，降低IO耗时
• 缓存Python解释器的环境，避免每次启动都重新加载依赖
• 针对选中的测试用例单独运行（而不是整个文件），比如通过-k参数指定测试名称，减少测试执行量

内容的提问来源于stack exchange，提问作者matetam