本文介绍一种简洁、可读性强的方案，利用 contextlib.nullcontext 统一处理 aiohttp.ClientSession 等混合 sync/async 上下文管理器的链式 Mock，避免冗长中间变量，直接控制最终 await resp.json() 的返回值。
本文介绍一种简洁、可读性强的方案，利用 `contextlib.nullcontext` 统一处理 `aiohttp.clientsession` 等混合 sync/async 上下文管理器的链式 mock，避免冗长中间变量，直接控制最终 `await resp.json()` 的返回值。
在使用 pytest 测试异步 HTTP 客户端（如 aiohttp）时，常见的嵌套异步上下文结构：
async with aiohttp.ClientSession() as session:
async with session.post("https://example.com") as resp:
data = await resp.json()
其调用链涉及 同步对象创建（ClientSession() 返回 sync 对象）、异步进入上下文（__aenter__）、同步方法调用（session.post() 返回 ClientResponse 对象）、再次异步进入（resp.__aenter__），最后才是 异步方法调用（resp.json()）。传统 Mock 方式需逐层构造 Mock / AsyncMock 并手动设置 __aenter__ 返回值，代码臃肿且语义模糊：
# ❌ 冗余、易错、意图不清晰
mock_session = Mock()
mock_resp_cm = AsyncMock()
mock_resp = Mock()
mock_resp.json = AsyncMock(return_value={"ok": True})
mock_resp_cm.__aenter__.return_value = mock_resp
mock_session.post.return_value = mock_resp_cm
MockClientSession.return_value.__aenter__.return_value = mock_session
✅ 推荐解法：用 contextlib.nullcontext 将 AsyncMock “包装”为支持 __aenter__/__aenter__ 的上下文管理器，从而实现链式 .return_value 风格的简洁 Mock。
nullcontext(obj) 是 Python 3.7+ 引入的轻量级上下文管理器，它在 __enter__ 中直接返回 obj，在 __exit__ 中不做任何操作 —— 正好适配 AsyncMock 实例作为 async with 目标的场景。
以下是完整、可运行的测试示例：
import contextlib
import pytest
from unittest import mock
import aiohttp
@pytest.mark.asyncio
async def test_aiohttp_json_mock():
# ✅ 定义期望响应数据
expected_data = {"status": "success", "code": 200}
# ✅ 构建带规范 spec 的 AsyncMock 响应对象，并预设 json() 返回值
mock_response = mock.AsyncMock(
spec=aiohttp.ClientResponse,
**{"json.return_value": expected_data}
)
# ✅ 将 mock_response 包装为 async context manager（支持 async with）
# 同时构建 session.post() 的返回值（即该包装后的响应）
mock_session = mock.Mock(
spec=aiohttp.ClientSession,
**{
"post.return_value": contextlib.nullcontext(mock_response)
}
)
# ✅ Mock ClientSession 构造函数，使其返回一个支持 async with 的 session
with mock.patch("aiohttp.ClientSession", return_value=contextlib.nullcontext(mock_session)):
async with aiohttp.ClientSession() as session:
async with session.post("https://example.com") as resp:
result = await resp.json()
assert result == expected_data
? 关键要点说明：
-
contextlib.nullcontext(mock.AsyncMock(...))让AsyncMock兼容async with语法，无需手动设置__aenter__； -
mock.Mock(spec=..., **{...})支持字典式属性赋值，使post.return_value链式配置更紧凑； - 所有
spec=参数确保类型安全与 IDE 友好（如自动补全、类型检查）； - 无需显式 patch 模块路径（如
"__main__.aiohttp.ClientSession"），推荐使用"aiohttp.ClientSession"（注意：实际使用时需根据被测代码中import aiohttp的位置调整 patch 路径，通常为模块内引用路径）。
? 进阶提示：若需模拟不同 URL 返回不同响应，可将 mock_session.post 替换为 side_effect 函数，动态返回对应 nullcontext(AsyncMock(...))；若需校验请求参数，可断言 mock_session.post.call_args。
此方法显著提升异步上下文链式 Mock 的可维护性与表达力，让测试聚焦于「行为预期」而非「Mock 结构」。
