如何在pytest函数式fixture中用MonkeyPatch上下文管理器设置环境变量？

问题根源 #

你用MonkeyPatch.context()写fixture无效，核心原因是上下文管理器的生命周期没覆盖到测试函数执行阶段：如果在fixture里仅用with块做修改但不使用yield，with块会在fixture执行完成后立即退出，所有环境变量修改被自动还原，测试函数运行时根本看不到修改。

而依赖monkeypatch fixture的写法有效，是因为monkeypatch内部已经帮你处理了修改的生命周期——测试前应用修改，测试后自动还原。

正确封装方式 #

方式1：结合yield延长上下文生命周期 #

用yield让MonkeyPatch.context()的作用范围覆盖测试函数的全程，测试函数执行时上下文保持，执行完后自动清理：

import os
from typing import Generator
import pytest
from _pytest.monkeypatch import MonkeyPatch

@pytest.fixture
def setup_test_env() -> Generator[None, None, None]:
    with MonkeyPatch.context() as mp:
        mp.setenv("TEST_VAR", "patched_value")
        yield  # 暂停，执行测试函数，结束后回到这里退出上下文

def test_env_var(setup_test_env):
    assert os.getenv("TEST_VAR") == "patched_value"

方式2：手动实例化MonkeyPatch并注册清理函数 #

自己创建MonkeyPatch实例，通过request.addfinalizer注册清理逻辑，同样能实现无依赖的封装：

import os
import pytest
from _pytest.monkeypatch import MonkeyPatch

@pytest.fixture
def setup_test_env(request):
    mp = MonkeyPatch()
    mp.setenv("TEST_VAR", "patched_value")
    # 测试结束后自动执行undo还原修改
    request.addfinalizer(mp.undo)

def test_env_var(setup_test_env):
    assert os.getenv("TEST_VAR") == "patched_value"

类型提示优化 #

给fixture添加Generator类型标注（方式1示例中已体现），可以让类型检查工具识别fixture的返回类型，提升类型友好性，同时避免传递多余的monkeypatch fixture，减少测试函数的参数数量。

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