如何修复pytest中Django Channels测试服务器使用生产数据库的问题？

核心问题是macOS默认采用spawn多进程上下文，子进程不会继承父进程内存中的Django配置，导致启动的服务器加载了生产配置而非测试配置。以下是两种可行的解决方式：

方式一：显式指定测试配置模块 #

在自定义的ChannelsLiveServerTestCase子类中添加settings_module属性，确保服务器子进程启动时加载正确的测试配置：

import pytest
from django.conf import settings


@pytest.fixture(scope="function")
def channels_live_server(request):
    from channels.testing import ChannelsLiveServerTestCase

    class MyChannelsLiveServerTestCase(ChannelsLiveServerTestCase):
        # 复用pytest已加载的测试配置，无需硬编码路径
        settings_module = settings.SETTINGS_MODULE
        
        @property
        def url(self):
            return self.live_server_url

    server = MyChannelsLiveServerTestCase()
    server._pre_setup()

    yield server

    server._post_teardown()

如果你的测试配置是独立文件（比如myproject.settings.test），也可以直接硬编码：

settings_module = "myproject.settings.test"

方式二：通过环境变量传递配置 #

设置DJANGO_SETTINGS_MODULE环境变量，确保子进程继承该变量并加载测试配置：

import os
import pytest
from django.conf import settings


@pytest.fixture(scope="function")
def channels_live_server(request):
    # 保存原环境变量，测试完成后恢复
    original_settings_env = os.environ.get("DJANGO_SETTINGS_MODULE")
    os.environ["DJANGO_SETTINGS_MODULE"] = settings.SETTINGS_MODULE

    from channels.testing import ChannelsLiveServerTestCase

    class MyChannelsLiveServerTestCase(ChannelsLiveServerTestCase):
        @property
        def url(self):
            return self.live_server_url

    server = MyChannelsLiveServerTestCase()
    server._pre_setup()

    yield server

    server._post_teardown()
    
    # 恢复原环境变量
    if original_settings_env:
        os.environ["DJANGO_SETTINGS_MODULE"] = original_settings_env
    else:
        del os.environ["DJANGO_SETTINGS_MODULE"]

额外注意事项 #

确保你的测试配置文件中正确配置了测试数据库，例如使用SQLite内存数据库或独立的测试文件数据库，避免污染生产数据：

# 示例测试配置中的数据库设置
DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.sqlite3',
        'NAME': ':memory:',  # 内存数据库，测试完自动销毁
    }
}

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