pytest使用@patch如何直接覆盖FastAPI路由内的局部变量

你想直接给路由函数里的局部变量var_i_need_to_mock赋值的思路从根上就行不通：Python的作用域是隔离的，unittest.mock的原理是替换模块/对象层面的可引用属性，根本碰不到函数运行时栈里的局部变量。你现在能成功patch super_object.very_long_complicated_logic的写法，本来就是最标准的实现，没有更"直接"的正规方案。

• 你写的@patch('app.router.submit')是想替换整个路由函数，但FastAPI在应用启动的时候就已经把原始的submit函数注册到路由表里了，你后续patch模块属性不会影响已经注册好的路由处理逻辑，请求进来还是跑原来的代码。
• 测试函数里写的var_i_need_to_mock = True是测试函数自己的局部变量，和路由函数里的同名变量没有任何关联，两个变量在完全隔离的作用域里，不可能互相覆盖。
• 另外你路由定义的路径是/{item_id}/submit，带路径参数，测试里直接请求/submit会直接报404，路径本身就写错了。

方案1：Mock依赖调用（唯一推荐的稳定方案） #

就是你已经跑通的思路，直接把复杂逻辑的返回值固定成你要的结果，不需要碰局部变量，这也是FastAPI官方推荐的测试写法：

from unittest.mock import patch

# 注意patch路径要写super_object在路由文件中的导入路径，不要写super_object本身的定义路径
@patch("app.router.super_object.very_long_complicated_logic", return_value=True)
def test_submit(mock_logic, test_client):
    # 路径要带上item_id参数，和路由定义匹配
    response = test_client.post("/test_item_123/submit")
    assert response.status_code == 200
    # 还可以额外校验复杂逻辑确实被调用了
    mock_logic.assert_called_once()

方案2：硬改路由执行流（非特殊场景不推荐） #

如果你就是想完全跳过路由内部的所有逻辑，直接返回指定结果，不要直接patch路由函数本身，要同步修改FastAPI已经注册的路由表，否则patch不会生效：

from unittest.mock import patch
from app.main import app

def test_submit_skip_all_logic(test_client):
    with patch("app.router.submit", return_value=True) as mock_submit:
        # 手动同步更新FastAPI路由表里注册的处理函数
        for route in app.routes:
            if route.path == "/{item_id}/submit" and "POST" in route.methods:
                route.endpoint = mock_submit
        response = test_client.post("/test_item_123/submit")
        assert response.status_code == 200

注意：这个方案属于hack写法，后续只要路由路径、请求方法改了，测试就会直接失效，维护成本很高，除非万不得已不要用。

patch路径永远填「被mock对象在被测试模块中的引用路径」，不要填对象本身的定义路径，这是80%的人写patch不生效的核心原因。

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