讲师中心 微信公众号
AI工具推荐 视频效率加速

Python 单元测试 mock 的分层原则

冬浩君_9871

冬浩君_9871

发布时间:2026-02-15 16:50:03

|

429人浏览过

|

来源于php中文网

原创

mock 应封装成 pytest fixture 而非在测试函数中重复使用 @patch;fixture 需用 yield + with 确保自动还原,命名应表达意图(如 fake_api_response),并注意 patch 路径必须是使用处而非定义处。

python 单元测试 mock 的分层原则

mock 应该放在哪一层:测试代码里直接写 patch 还是封装成 fixture?

mock 的位置直接决定可维护性和耦合度。在测试函数里反复写 @patch('xxx.yyy') 或 mock.patch(...),会导致 patch 路径散落、重复、难定位——尤其当被测模块路径重构时,一堆测试会悄无声息地失效(因为 patch 的路径根本没被调用,但 mock 依然“成功”了)。

更稳妥的做法是把 patch 封装进 pytest fixture,统一管理路径和返回值:

import pytest
from unittest.mock import MagicMock

@pytest.fixture
def mock_requests_get():
    with mock.patch('requests.get') as m:
        m.return_value.json.return_value = {'id': 1}
        yield m

这样既避免路径硬编码污染测试逻辑,又能让多个测试复用同一 mock 行为。注意:fixture 必须用 yield + with,否则退出时 patch 不自动还原,可能污染后续测试。

  • 别在 setup_method 或 setUp 里做 patch——pytest fixture 的作用域控制更清晰
  • fixture 名不要带 mock_ 前缀,比如叫 fake_api_response 比 mock_requests_get 更表达意图
  • 如果 patch 的是类的实例方法,确保 patch 的是「使用处」的路径,不是定义处(常见错误:@patch('module.Class.method') 错,应为 @patch('caller_module.Class.method'))

什么时候该 mock 外部依赖,什么时候不该?

mock 的核心目标是隔离不确定性,不是消灭所有真实调用。数据库、HTTP、文件系统、时间等不可控或慢速依赖必须 mock;但像 json.loads、datetime.now() 这类纯函数或轻量操作,mock 反而增加认知负担,且容易因过度隔离导致测试失真。

立即学习“Python免费学习笔记(深入)”;

Python 编程规范
Python 编程规范

Python 编程规范与最佳实践。强制 PEP 8 代码风格、语法校验、单元测试执行,适用于代码编写、审查和重构。触发词:Python、代码规范、py_compile、单元测试。

下载

一个简单判断标准:这个调用是否「可能失败」「耗时不可控」「结果不固定」或「触发副作用(如发邮件)」。如果不是,让它真实运行反而更可靠。

  • 别 mock os.path.exists 来测试路径逻辑——改用 tempfile.TemporaryDirectory 创建真实临时目录更直观
  • mock time.time() 是合理的,但 mock math.sqrt() 就毫无意义
  • 对第三方 SDK(如 boto3、redis-py),优先 mock 客户端实例,而不是底层 HTTP 请求——层级太高易断,太低难覆盖

side_effect 和 return_value 怎么选才不翻车?

return_value 适合返回固定数据,side_effect 才能模拟异常、多态响应或状态变化。但很多人误以为 side_effect=ValueError 就能抛异常——其实它只会把 ValueError 当作返回值返回,真正抛异常得写 side_effect=ValueError() 或 side_effect=lambda: raise ValueError()。

另一个高频坑:用 side_effect 返回迭代器时,耗尽后再次调用会直接抛 StopIteration,而不是静默返回 None。这会让本该只调用一次的函数,在测试中被多调用就崩掉。

  • 抛异常:用 side_effect=ValueError('boom'),不是 side_effect=ValueError
  • 模拟多次不同返回:用 side_effect=[{'a':1}, {'a':2}, ValueError]
  • 需要动态逻辑(如根据参数返回不同值):用函数,但记得加 *args, **kwargs 签名,否则参数不匹配会静默失败

为什么 autospec=True 不是默认选项,但它值得手动开?

autospec=True 会让 mock 自动继承原对象的签名(参数名、数量、是否可变参),调用时参数错位或多余会立刻报错,而不是静默忽略——这对防止“mock 写错了但测试还绿”的情况特别有用。但它不是默认,是因为会额外反射导入目标对象,可能拖慢测试启动,且对某些动态构造的对象(如部分 SQLAlchemy 模型)会失败。

建议在关键集成测试或团队新项目初期就启用,尤其当你发现 mock 调用看起来没问题,但真实运行却报 TypeError: unexpected keyword argument 时,大概率就是缺 autospec。

  • 启用方式:所有 patch 和 Mock 构造都加 autospec=True,例如 mock.patch('foo.bar', autospec=True)
  • 如果遇到 AttributeError: __spec__,说明目标对象不支持反射,降级为手动定义 spec=SomeClass
  • 它不会帮你校验返回值类型,只管调用签名——别指望它替你 catch 业务逻辑错误
mock 分层这事,本质是平衡「控制力」和「维护成本」。越往上层(比如业务逻辑层)mock,测试越快越稳定;但离真实调用链越远,就越容易漏掉路径差异或副作用。最麻烦的往往不是怎么 mock,而是忘了 mock 后没验证是否真被调用了——别只 assert 返回值,顺手加一句 mock_obj.assert_called_once()。

热门AI工具

更多
火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

码上飞
码上飞 Hot

一款基于自然语言的AI应用开发平台,可根据用户描述自动生成应用程序,帮助个人和团队快速完成应用原型与开发。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

相关专题

更多
json数据格式
json数据格式

JSON是一种轻量级的数据交换格式。本专题为大家带来json数据格式相关文章,帮助大家解决问题。

2255

2023.08.07

json是什么
json是什么

JSON是一种轻量级的数据交换格式,具有简洁、易读、跨平台和语言的特点,JSON数据是通过键值对的方式进行组织,其中键是字符串,值可以是字符串、数值、布尔值、数组、对象或者null,在Web开发、数据交换和配置文件等方面得到广泛应用。本专题为大家提供json相关的文章、下载、课程内容,供大家免费下载体验。

3622

2023.08.23

jquery怎么操作json
jquery怎么操作json

操作的方法有:1、“$.parseJSON(jsonString)”2、“$.getJSON(url, data, success)”;3、“$.each(obj, callback)”;4、“$.ajax()”。更多jquery怎么操作json的详细内容,可以访问本专题下面的文章。

1216

2023.10.13

go语言处理json数据方法
go语言处理json数据方法

本专题整合了go语言中处理json数据方法,阅读专题下面的文章了解更多详细内容。

4179

2025.09.10

java多态详细介绍
java多态详细介绍

本专题整合了java多态相关内容,阅读专题下面的文章了解更多详细内容。

375

2025.11.27

lambda表达式
lambda表达式

Lambda表达式是一种匿名函数的简洁表示方式,它可以在需要函数作为参数的地方使用,并提供了一种更简洁、更灵活的编码方式,其语法为“lambda 参数列表: 表达式”,参数列表是函数的参数,可以包含一个或多个参数,用逗号分隔,表达式是函数的执行体,用于定义函数的具体操作。本专题为大家提供lambda表达式相关的文章、下载、课程内容,供大家免费下载体验。

1111

2023.09.15

python lambda函数
python lambda函数

本专题整合了python lambda函数用法详解,阅读专题下面的文章了解更多详细内容。

345

2025.11.08

Python lambda详解
Python lambda详解

本专题整合了Python lambda函数相关教程,阅读下面的文章了解更多详细内容。

614

2026.01.05

Docker Desktop怎么设置端口映射
Docker Desktop怎么设置端口映射

本专题聚焦Docker Desktop容器端口映射与访问配置,整理从创建容器、设置端口绑定到验证本地服务访问的操作流程。针对端口被占用、映射配置错误、容器服务无法访问等情况,也提供相应的检查方向,方便用户完成端口配置。

0

2026.11.03

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn