0

0

Python 单元测试 mock 的分层原则

舞夢輝影

舞夢輝影

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

|

409人浏览过

|

来源于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_methodsetUp 里做 patch——pytest fixture 的作用域控制更清晰
  • fixture 名不要带 mock_ 前缀,比如叫 fake_api_responsemock_requests_get 更表达意图
  • 如果 patch 的是类的实例方法,确保 patch 的是「使用处」的路径,不是定义处(常见错误:@patch('module.Class.method') 错,应为 @patch('caller_module.Class.method')

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

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

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

PpcyAI
PpcyAI

泡泡次元AI-游戏美术AI创作平台,低门槛上手,高度可控,让你的创意秒速落地

下载

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

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

side_effectreturn_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

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

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

441

2023.08.07

json是什么
json是什么

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

544

2023.08.23

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

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

318

2023.10.13

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

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

81

2025.09.10

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

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

22

2025.11.27

lambda表达式
lambda表达式

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

212

2023.09.15

python lambda函数
python lambda函数

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

192

2025.11.08

Python lambda详解
Python lambda详解

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

58

2026.01.05

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

139

2026.02.13

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.3万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.6万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号