本文介绍在 Python 单元测试中精准 Mock asyncio.sleep 的最佳实践，通过保存原始函数并注入加速逻辑（如除以 10），既避免污染测试环境，又能保留异步行为语义，有效提升测试速度并暴露潜在竞态问题。
本文介绍在 python 单元测试中精准 mock `asyncio.sleep` 的最佳实践，通过保存原始函数并注入加速逻辑（如除以 10），既避免污染测试环境，又能保留异步行为语义，有效提升测试速度并暴露潜在竞态问题。
在异步单元测试中，频繁调用 asyncio.sleep(5) 或更长延迟会显著拖慢测试执行——但盲目全局 patch asyncio.sleep 又可能导致测试自身逻辑（如 await asyncio.sleep()）意外失效，甚至引发 AttributeError 或难以调试的时序异常。
关键在于：不覆盖原始 asyncio.sleep，而是“借用”它实现可控加速。正确做法是：在 patch 作用域内，先缓存真实的 asyncio.sleep，再定义一个 fast_sleep 辅助函数，内部仍调用原始函数但缩放时间参数：
import asyncio
import pytest
from unittest.mock import patch
import sut # 假设被测模块为 sut.py
@pytest.mark.asyncio
async def test_sut_with_fast_sleep():
original_sleep = asyncio.sleep # ✅ 安全捕获原始函数
async def fast_sleep(duration: float):
# 将等待时间缩短为原来的 1/10，同时保持真实事件循环调度
await original_sleep(max(0.001, duration / 10)) # 防止传入 0 导致跳过
with patch("sut.asyncio.sleep") as mock_sleep:
mock_sleep.side_effect = fast_sleep
await sut.some_async_function() # 此处调用的 sleep 已被加速
⚠️ 注意事项：
-
必须 patch
"sut.asyncio.sleep"（目标模块路径），而非"asyncio.sleep"：前者仅影响sut.py中对sleep的引用；后者会劫持所有导入asyncio模块的位置（包括test.py自身），导致await asyncio.sleep(...)在测试体内也失效。 -
避免在
@patch装饰器中直接使用async def作为side_effect：装饰器在测试函数执行前即完成 patch，此时asyncio.sleep已被替换，无法在定义阶段调用原函数。 - 建议封装为 fixture 提升复用性与可维护性：
@pytest.fixture
def fast_sleep():
original = asyncio.sleep
async def _fast_sleep(t):
await original(max(0.001, t / 10))
with patch("sut.asyncio.sleep", side_effect=_fast_sleep):
yield
@pytest.mark.asyncio
async def test_user_timeout_handling(fast_sleep):
result = await sut.process_with_timeout()
assert result == "success"
该方案兼顾了隔离性（仅影响被测模块）、真实性（仍经由真实事件循环调度）和可观测性（若因加速暴露出竞态条件，说明生产环境同样脆弱）。最终，你的 5 秒等待在测试中变为 500ms，而代码逻辑与并发边界验证毫发无损。
