怎么在Python中用pytest测试FastAPI接口-Python教程-PHP中文网

FastAPI官方推荐用TestClient模拟HTTP请求，需传入已配置的app实例，正确处理路径/查询参数、依赖覆盖及异步依赖，避免污染测试环境。

怎么在python中用pytest测试fastapi接口

用 `TestClient` 模拟真实 HTTP 请求最直接

FastAPI 官方推荐用 fastapi.testclient.TestClient，它底层基于 httpx，能完整走通路由、依赖注入、中间件和异常处理器，比手写 httpx.AsyncClient 或 aiohttp 更省心，也比 mock 视图函数更贴近真实行为。

关键点：必须传入 app 实例（不是模块名或字符串），且不能在测试中重复调用 app = FastAPI() —— 否则依赖、中间件、路由注册全丢失。

正确写法：client = TestClient(app)，其中 app 是你主模块里已配置好的 FastAPI 实例
常见错误：在测试文件里重新 app = FastAPI()，再挂路由，导致测试的不是生产代码
如果 app 依赖环境变量或数据库连接，需在测试前用 monkeypatch.setenv() 或临时替换 Depends() 的返回值

测试带路径参数、查询参数和请求体的接口要对齐 Pydantic 验证

FastAPI 的参数解析和校验发生在 TestClient 的 .get()/.post() 调用过程中，所以传参方式必须匹配声明方式 —— 否则会触发 422 错误，而不是你预期的业务逻辑分支。

示例：假设接口是 @app.post("/items/{item_id}")，接收 item_id: int 和 JSON body Item：

def test_create_item():
    response = client.post(
        "/items/123",  # ✅ 路径参数必须拼进 URL 字符串
        json={"name": "foo", "price": 9.99}  # ✅ body 用 json=，自动序列化 + 设 Content-Type
    )
    assert response.status_code == 200

路径参数不能塞进 params=，否则 404；查询参数才用 params=
表单数据用 data=，上传文件用 files=，别混用 json=
如果接口用了 Body(..., embed=True)，JSON body 必须是嵌套对象，比如 {"item": {...}}

测试依赖覆盖要用 `app.dependency_overrides`，别改源码或全局 patch

真实接口常依赖数据库、缓存、认证等外部服务。测试时必须切断这些依赖，但又不能破坏 FastAPI 的依赖解析链 —— 直接修改模块级变量或 monkeypatch 函数容易漏掉异步依赖或被中间件绕过。

标准做法是在每个测试前后用 app.dependency_overrides 注入可控的替代实现：

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

def test_read_user_with_mock_db():
    async def override_get_db():
        return MockDBSession()  # 返回假 session，不连真实 DB
<pre class="brush:php;toolbar:false;">app.dependency_overrides[get_db] = override_get_db
try:
    response = client.get("/users/1")
    assert response.json()["id"] == 1
finally:
    app.dependency_overrides.clear()  # ⚠️ 必须清理，否则污染其他测试

务必在 try/finally 或 pytest fixture 的 yield 后清理 dependency_overrides
不要用 unittest.mock.patch 去 patch get_db，FastAPI 不会识别被 patch 的函数对象
如果依赖是类实例（如 CacheService），可直接 app.dependency_overrides[CacheService] = lambda: MockCache()

异步依赖或 `BackgroundTasks` 需手动 await 或 patch

FastAPI 允许路由函数返回 coroutine，或接收 BackgroundTasks。但 TestClient 是同步的，它会自动 await 路由返回的协程 —— 所以你写的测试函数本身可以是普通函数，不用 async def。

但有两点必须注意：

如果依赖里有 await（比如 async def get_current_user()），确保你覆盖的 mock 也返回 awaitable，或直接返回结果（TestClient 会兼容）
BackgroundTasks.add_task() 在测试中默认不会执行，需用 monkeypatch.setattr 拦截并验证是否被调用，或改用同步模拟版本
若真要测任务执行逻辑，得换 httpx.AsyncClient + pytest-asyncio，但代价高，通常 mock 更合理