使用pytest测试Django配置：环境变量加载失效问题

问题根源 #

Django的settings模块在pytest会话初始化阶段就会被导入执行，模块级的CONFIG变量此时已完成初始化。你定义的set_env fixture是在测试函数运行时才执行，这时环境变量的修改无法触发CONFIG重新加载，导致断言失败。

可行解决方案 #

方案1：测试用例内重新加载settings模块

在设置环境变量后，手动重新加载settings模块，触发CONFIG的重新初始化：

# some_app.tests.test_settings.py
import os
import json
import importlib
from tempfile import NamedTemporaryFile
import pytest
from proj import settings

@pytest.fixture
def data_raw():
    return dict(foo="bar")

@pytest.fixture
def data_file(data_raw):
    # 设置delete=False，避免with块结束后文件被自动删除
    with NamedTemporaryFile(mode="w+", delete=False) as f:
        json.dump(data_raw, f)
    yield f.name
    # 测试结束后手动清理临时文件
    os.unlink(f.name)

@pytest.fixture
def set_env(monkeypatch, data_file):
    # 使用monkeypatch安全设置环境变量，避免污染全局环境
    monkeypatch.setenv('FILE_PATH', data_file)

def test_it_loads_data(set_env, data_raw):
    # 重新加载settings模块，触发CONFIG的重新初始化
    importlib.reload(settings)
    assert settings.CONFIG == data_raw

方案2：会话级钩子提前设置环境变量

通过pytest的pytest_configure和pytest_unconfigure钩子，在整个测试会话启动前设置环境变量，确保settings加载时能读取到：

# 项目根目录下的conftest.py
import os
import json
import tempfile
import pytest

def pytest_configure(config):
    # 创建临时配置文件
    data_raw = dict(foo="bar")
    with tempfile.NamedTemporaryFile(mode="w+", delete=False) as f:
        json.dump(data_raw, f)
    # 设置环境变量
    os.environ['FILE_PATH'] = f.name
    # 保存临时文件路径用于后续清理
    config._temp_config_file = f.name

def pytest_unconfigure(config):
    # 清理临时文件并移除环境变量
    if hasattr(config, '_temp_config_file'):
        os.unlink(config._temp_config_file)
        del os.environ['FILE_PATH']

对应的测试代码可以简化：

# some_app.tests.test_settings.py
from proj import settings

def test_it_loads_data(data_raw):
    assert settings.CONFIG == data_raw

@pytest.fixture
def data_raw():
    return dict(foo="bar")

方案3：模块级自动生效fixture

如果只需在当前测试模块内生效，使用模块级autouse fixture，在模块加载后自动设置环境变量并重新加载settings：

# some_app.tests.test_settings.py
import importlib
import json
import os
from tempfile import NamedTemporaryFile
import pytest
from proj import settings

@pytest.fixture(scope='module', autouse=True)
def setup_module_env(data_raw):
    with NamedTemporaryFile(mode="w+", delete=False) as f:
        json.dump(data_raw, f)
    os.environ['FILE_PATH'] = f.name
    # 重新加载settings模块
    importlib.reload(settings)
    yield
    # 清理资源，恢复环境
    os.unlink(f.name)
    del os.environ['FILE_PATH']
    settings.CONFIG = None

@pytest.fixture
def data_raw():
    return dict(foo="bar")

def test_it_loads_data(data_raw):
    assert settings.CONFIG == data_raw

关键注意点 #

• 使用NamedTemporaryFile时需添加delete=False，否则文件会在with块结束后自动删除，导致后续读取失败，测试结束后需手动清理文件。
• 优先使用monkeypatch设置环境变量，避免直接修改os.environ污染全局测试环境。

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