0

0

如何正确在 Pytest 中管理 Mock 补丁以避免跨测试污染

霞舞

霞舞

发布时间:2026-02-02 10:34:13

|

273人浏览过

|

来源于php中文网

原创

如何正确在 Pytest 中管理 Mock 补丁以避免跨测试污染

pytest 中使用 `@mock.patch` 类装饰器会导致补丁未自动清理,从而污染后续测试;应改用 `pytest-mock` 插件提供的 `mocker` fixture,在函数级精确控制 mock 生命周期。

在 Pytest 中,直接使用 unittest.mock.patch 的类装饰器(如 @mock.patch(...) 修饰整个测试类)是危险且不推荐的。其根本原因在于:该装饰器会在类定义时立即应用 patch,并默认在类加载后持续生效,而非在测试方法执行完毕后自动还原——尤其当测试跨文件执行(如 pytest 全局运行)时,patch 可能长期驻留于模块命名空间中,导致后续测试(如 sql_loader_test.py 中的用例)意外继承已被 mock 的 SnowFlakeConnector,出现 NonCallableMagicMock 实例复用、ID 相同等“跨测试污染”现象。

✅ 正确做法是使用 pytest-mock 插件提供的 mocker fixture:

pip install pytest-mock

然后在测试中按需、按作用域启用 patch:

# test/kernel_application_test.py
class TestExecuteApplication:
    def test_connects_with_config(self, mocker):
        # ✅ 函数级 patch:仅在本测试方法内生效,退出自动还原
        mock_connector = mocker.patch('customlib.snowflake.SnowFlakeConnector', autospec=True)

        # 调用被测代码(例如初始化应用)
        from customlib.kernel import KernelApplication
        app = KernelApplication()
        app.execute()

        # 断言 mock 行为
        mock_connector.assert_called_once()
        assert mock_connector.return_value.connect.called

    def test_handles_failure_gracefully(self, mocker):
        # ✅ 每个测试独立 patch,互不干扰
        mock_connector = mocker.patch('customlib.snowflake.SnowFlakeConnector', autospec=True)
        mock_connector.side_effect = ConnectionError("Network down")

        # ... 触发异常逻辑并断言

同样地,在 test/sql_loader_test.py 中也应遵循相同模式:

Calicat
Calicat

ProcessOn推出的一站式AI产品设计协作平台

下载
# test/sql_loader_test.py
class TestSqlLoader:
    def test_loads_from_query(self, mocker):
        # ✅ 即使 kernel_application_test.py 已运行过,此处仍是干净的原始类
        mock_connector = mocker.patch('customlib.snowflake.SnowFlakeConnector', autospec=True)
        mock_cursor = mock_connector.return_value.cursor.return_value
        mock_cursor.fetchall.return_value = [("data1",), ("data2",)]

        loader = SqlLoader()
        result = loader.load("SELECT * FROM table")

        assert len(result) == 2

⚠️ 注意事项:

  • 不要继承 unittest.TestCase:Pytest 原生不依赖 unittest 运行机制,继承它不仅无益,还可能干扰 fixture 注入和生命周期管理;
  • 避免全局/模块级 patch:如 @patch(...) 修饰模块或使用 patch(...).start() + stop() 手动管理,极易遗漏还原;
  • 优先使用 mocker.patch 而非 mock.patch:mocker fixture 内置自动 cleanup 逻辑,确保每个测试函数结束后所有 patch 均被撤销;
  • 若需类级别 patch(极少数场景),可使用 mocker.patch 在 setup_method 或 pytest.fixture(scope="function") 中声明,但务必坚持“最小作用域”原则。

总结:Pytest 的 mock 管理哲学是「显式、局部、自动还原」。放弃类装饰器式的“一劳永逸”,转而拥抱 mocker fixture 的函数级精准控制,是保障测试隔离性与可靠性的关键实践。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
function是什么
function是什么

function是函数的意思,是一段具有特定功能的可重复使用的代码块,是程序的基本组成单元之一,可以接受输入参数,执行特定的操作,并返回结果。本专题为大家提供function是什么的相关的文章、下载、课程内容,供大家免费下载体验。

485

2023.08.04

js函数function用法
js函数function用法

js函数function用法有:1、声明函数;2、调用函数;3、函数参数;4、函数返回值;5、匿名函数;6、函数作为参数;7、函数作用域;8、递归函数。本专题提供js函数function用法的相关文章内容,大家可以免费阅读。

163

2023.10.07

Golang WebAssembly(WASM)开发入门
Golang WebAssembly(WASM)开发入门

本专题系统讲解 Golang 在 WebAssembly(WASM)开发中的实践方法,涵盖 WASM 基础原理、Go 编译到 WASM 的流程、与 JavaScript 的交互方式、性能与体积优化,以及典型应用场景(如前端计算、跨平台模块)。帮助开发者掌握 Go 在新一代 Web 技术栈中的应用能力。

0

2026.02.02

PHP Swoole 高性能服务开发
PHP Swoole 高性能服务开发

本专题聚焦 PHP Swoole 扩展在高性能服务端开发中的应用,系统讲解协程模型、异步IO、TCP/HTTP/WebSocket服务器、进程与任务管理、常驻内存架构设计。通过实战案例,帮助开发者掌握 使用 PHP 构建高并发、低延迟服务端应用的工程化能力。

0

2026.02.02

Java JNI 与本地代码交互实战
Java JNI 与本地代码交互实战

本专题系统讲解 Java 通过 JNI 调用 C/C++ 本地代码的核心机制,涵盖 JNI 基本原理、数据类型映射、内存管理、异常处理、性能优化策略以及典型应用场景(如高性能计算、底层库封装)。通过实战示例,帮助开发者掌握 Java 与本地代码混合开发的完整流程。

0

2026.02.02

go语言 注释编码
go语言 注释编码

本专题整合了go语言注释、注释规范等等内容,阅读专题下面的文章了解更多详细内容。

61

2026.01.31

go语言 math包
go语言 math包

本专题整合了go语言math包相关内容,阅读专题下面的文章了解更多详细内容。

52

2026.01.31

go语言输入函数
go语言输入函数

本专题整合了go语言输入相关教程内容,阅读专题下面的文章了解更多详细内容。

25

2026.01.31

golang 循环遍历
golang 循环遍历

本专题整合了golang循环遍历相关教程,阅读专题下面的文章了解更多详细内容。

10

2026.01.31

热门下载

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

精品课程

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

共578课时 | 55.6万人学习

国外Web开发全栈课程全集
国外Web开发全栈课程全集

共12课时 | 1.0万人学习

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

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