Python测试API请求，pytest-mock拦截网络调用

本文深入解析了在Python测试中使用pytest-mock模拟API请求的常见误区与正确实践，重点揭示了为何直接patch requests.get往往失效——根本原因在于mock必须精准作用于被测代码中实际导入和调用的对象路径，而非requests库本身；文章通过import requests、from requests import get、requests.Session三种典型场景，手把手演示了patch路径的精确写法，并强调了mock响应需显式配置json()方法、status_code等关键属性，避免AttributeError或逻辑判断失败；同时延伸至aiohttp异步测试，指出解耦依赖（如将session作为参数传入）比强行patch动态对象更可靠、可维护。这不仅是一份实用指南，更是对测试稳定性和重构安全性的深刻提醒：mock是否生效，永远取决于你和被测代码的“导入契约”是否依然严丝合缝。

pytest-mock 为什么不能直接拦截 requests.get？

因为 pytest-mock 本身不提供网络层拦截能力，它只是 unittest.mock 的轻量封装。你 mock 的必须是「被测代码里实际 import 并调用的对象」，而不是底层 socket 或 requests 库的内部函数。

常见错误现象：requests.get 调用依然发出真实 HTTP 请求，mock 像没生效一样；或者 mock 成功但测试结果和预期不符，其实是 patch 错了位置。

• 确认被测函数里怎么写的——比如是 import requests 后调用 requests.get(...)，那就得 patch requests.get
• 如果被测模块是 from requests import get，就得 patch your_module.get，不是 requests.get
• patch 必须在被测函数所在的模块层级生效，路径写错一个点（比如少写包名）就完全不生效

正确 patch requests.get 的三种典型写法

取决于你的导入方式和模块结构。下面三个例子都基于被测函数在 api_client.py 中：

场景1： api_client.py 里写的是 import requests → requests.get(...)

def test_fetch_user(mocker):
    mock_resp = mocker.Mock()
    mock_resp.json.return_value = {"id": 1, "name": "alice"}
    mocker.patch("api_client.requests.get", return_value=mock_resp)
    result = fetch_user(1)
    assert result["name"] == "alice"

场景2： api_client.py 里是 from requests import get → get(...)

• 必须 patch api_client.get，不是 requests.get
• 否则 mock 注入不到局部作用域里的 get 名字上

场景3： 使用 requests.Session() 实例发请求

• 要 mock Session.get 方法，或直接 mock 整个 session 实例
• 注意：mocker.patch("api_client.session.get") 前提是你在模块里定义了全局 session = requests.Session()

返回值设错导致 JSON 解析失败

mock 对象默认不带 .json() 方法，直接调用会报 AttributeError: Mock object has no attribute 'json'。这不是 requests 问题，是 mock 没配好。

• 最简方案：用 mocker.Mock(return_value={"id": 1}) 不够，得显式加方法：mocker.Mock(json=mocker.Mock(return_value={"id": 1}))
• 更稳写法：用 Response 类模拟（需 from requests.models import Response），设置 _content 和 status_code
• 别只 mock 返回值，漏掉 status_code —— 很多业务逻辑会检查 r.status_code == 200

测试异步 API 调用（aiohttp）怎么办？

pytest-mock 对 aiohttp.ClientSession 同样无效，原因一致：你要 patch 的是被测代码里实际 await 的那个对象。

• 如果被测函数里是 async with aiohttp.ClientSession() as session: → await session.get(...)，那得 patch session.get，但 session 是运行时创建的，得用 mocker.patch.object 或提前构造 mock session
• 推荐做法：把 session 作为参数传入被测函数，测试时直接传 mock，避免 patch 动态对象
• 别试图 mock aiohttp.request 全局函数——它不是被调用路径上的目标

真实项目里最容易被忽略的，是 patch 路径和被测代码中 import 方式的耦合关系。写对一次不难，但重构时改了 import 方式，mock 就悄无声息地失效了。每次 CI 报网络超时失败，先查 patch 路径是不是还对得上。

本篇关于《Python测试API请求，pytest-mock拦截网络调用》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于文章的相关知识，请关注golang学习网公众号！