如何实现在Python异步代码中进行单元测试_利用AsyncMock模拟异步调用

落枫同学_6704

落枫同学_6704

2026-05-23

934人浏览

原创

asyncmock仅支持python 3.8+,低版本需升级或用patch+协程模拟;patch必须作用于调用位置而非定义位置;return_value自动包装为协程但类型非真实,需side_effect或cast确保类型;嵌套await需显式配置子属性为asyncmock。

如何实现在python异步代码中进行单元测试_利用asyncmock模拟异步调用

AsyncMock 在 Python 3.8+ 中才能直接使用

Python 3.7 及更早版本没有 AsyncMock,强行导入会报 NameError: name 'AsyncMock' is not defined。如果你的项目还跑在 3.7 上,要么升级解释器,要么用 unittest.mock.patch 配合手动返回 coroutine 对象(比如用 asyncio.sleep(0) 或 mock.AsyncMock 的 backport 库 asynctest —— 但注意它已停止维护)。

实操建议:

  • 检查 Python 版本:python --version,确认 ≥ 3.8
  • 直接导入:from unittest.mock import AsyncMock,不用再装第三方包
  • 若用 pytest,确保 pytest ≥ 6.2(对 AsyncMock 的 await 行为修复更完整)

用 AsyncMock 替换 async def 函数时必须 patch 到调用位置

常见错误是 patch 了定义处(比如 @patch('mylib.fetch_data')),但测试代码实际 import 并调用的是 from mylib import fetch_data —— 这时 patch 失效,因为 mock 没覆盖到被引用的对象。

正确做法是 patch **测试模块中如何导入并使用该函数的位置**。例如:

# test_api.py
from myapp.api import call_service  # ← 这里用了 mylib.fetch_data
<p>@patch('myapp.api.fetch_data')  # ✅ patch 调用方所在模块里的名字
async def test_call_service(mock_fetch):
mock_fetch.return_value = {'ok': True}
result = await call_service()
assert result['ok']
</p>

关键点:

python 查询技能
python 查询技能

查询客流数据,输出JSON格式,可直接导入Bitable等可视化工具

下载
  • AsyncMock 和普通 Mock 一样,遵循“patch 导入点”原则,不是 patch 定义点
  • 如果函数是 await fetch_data(),return_value 设成普通值即可(AsyncMock 会自动包装成协程);若需控制返回协程的行为(如抛异常),用 side_effect 返回 Exception 或 awaitable

AsyncMock.return_value 不等于 awaitable,要小心类型断言失败

当你写 mock_fn.return_value = {'data': 1},await mock_fn() 确实能返回该字典——但它的返回类型其实是 Mock,不是 dict。如果被测函数或类型检查器(如 mypy)做了严格类型注解(例如声明返回 dict[str, Any]),运行时虽不报错,静态检查可能失败,或引发 isinstance 判断误判。

解决方式:

  • 需要真实类型时,用 side_effect 返回一个真正的协程:mock_fn.side_effect = lambda: asyncio.sleep(0) or {'data': 1}
  • 或者用 AsyncMock(return_value=...) 后,显式 cast:mock_fn.return_value = cast(dict, {'data': 1})(配合 typing.cast)
  • 避免在 mock 上做 isinstance(result, dict) 这类判断;改用 hasattr(result, 'keys') 或 duck typing

测试嵌套 await 时,AsyncMock 默认不递归 mock awaitable 属性

比如你 mock 了一个返回对象的异步方法:resp = await http_client.get(...),然后又写 await resp.json()。如果只 mock get,resp 是个 AsyncMock,但它的 json 属性默认还是普通 Mock,不能 await —— 此时会报 TypeError: object AsyncMock can't be used in 'await' expression(因为 resp.json 不是协程)。

正确做法是让嵌套属性也支持 await:

  • 设置 resp.json = AsyncMock(return_value={'id': 123})
  • 或初始化时用 AsyncMock 的 configure_mock:resp.configure_mock(json=AsyncMock(return_value={'id': 123}))
  • 更简洁:用 AsyncMock(spec=type(real_resp))(前提是 real_resp 类型有明确的异步方法签名)

这容易被忽略,尤其当 API 客户端封装多层 await 时,漏掉一层就会卡在 await 报错,而不是业务逻辑出错。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1711

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4324

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1689

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

25197

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3067

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3087

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1163

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2403

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程