基于python-dotenv，如何为单个pytest测试用例设自定义配置？

嘿，我刚好碰到过类似的场景，这个细粒度的测试配置需求完全可以通过pytest的fixture+重构配置加载逻辑来优雅解决，不用再写一堆繁琐的mock啦！

1. 先重构你的配置加载逻辑 #

首先要把原来的全局Config对象改成可注入、可重置的模式，让所有应用代码都通过统一入口获取配置，而不是直接引用全局对象。这样后续测试的时候只需要修改这个入口的返回值，不用到处mock。

举个例子，创建config.py文件：

from dotenv import load_dotenv
import os

# 用内部变量存储配置实例
_config_instance = None

class AppConfig:
    def __init__(self):
        # 加载.env文件
        load_dotenv()
        # 初始化配置项，这里可以加默认值
        self.foo = os.getenv("FOO", "default_foo")
        self.bar = os.getenv("BAR", 123)
        # 其他配置项...

def get_config():
    """统一的配置获取入口"""
    global _config_instance
    if not _config_instance:
        _config_instance = AppConfig()
    return _config_instance

之后你的应用代码里，所有需要用配置的地方都这样写：

from config import get_config

def run_code_that_uses_config():
    config = get_config()
    return f"Value of foo: {config.foo}"

2. 编写pytest专属的配置Fixture #

接下来在项目根目录的conftest.py里写一个fixture，负责每次测试前重置配置，并且返回一个可修改的配置实例，保证每个测试用例的配置都是独立的：

import pytest
from config import _config_instance, AppConfig

@pytest.fixture
def config():
    """每个测试用例专属的可修改配置实例"""
    # 重置全局配置实例
    global _config_instance
    _config_instance = AppConfig()
    # 返回配置实例给测试用例使用
    yield _config_instance
    # 测试结束后再次重置，避免影响其他测试
    _config_instance = None

3. 在测试用例里直接用（完全符合你的预期） #

现在你就可以像最开始想要的那样，在每个测试用例里自由修改配置了，完全不用写mock：

def test_foo_with_bar(config):
    # 给当前测试用例设置专属配置
    config.foo = 'bar'
    # 运行依赖配置的代码
    result = run_code_that_uses_config()
    # 断言结果
    assert result == "Value of foo: bar"

def test_foo_with_bleh(config):
    # 另一个测试用例的专属配置，完全不影响上一个测试
    config.foo = 'bleh'
    result = run_code_that_uses_config()
    assert result == "Value of foo: bleh"

4. 可选：给测试添加基础配置 #

如果想给所有测试用例设置统一的基础测试配置（比如从.env.test加载），可以修改AppConfig的初始化逻辑：

class AppConfig:
    def __init__(self, env="development"):
        # 先加载基础.env
        load_dotenv()
        # 如果是测试环境，加载测试专属的.env文件并覆盖基础配置
        if env == "test":
            load_dotenv(".env.test", override=True)
        self.foo = os.getenv("FOO", "default_foo")
        # ...其他配置

然后在fixture里指定测试环境：

@pytest.fixture
def config():
    global _config_instance
    # 加载测试环境的基础配置
    _config_instance = AppConfig(env="test")
    yield _config_instance
    _config_instance = None

这样每个测试用例会先有.env.test里的基础配置，你只需要修改个别项就可以了。

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