文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Python 中 Mock 到底该怎么玩?一篇文章告诉你

2024-12-03 13:07

关注

本文转载自微信公众号「AirPython」,作者星安果 。转载本文请联系AirPython公众号。  

 1. 前言

微服务架构下,由于各类服务开发进度的不一致,导致联调工作经常会存在不确定性,进而导致项目延期

在实际工作中,为了保证项目进度,我们经常需要针对部分未完成模块及不稳定模块采用 Mock 方式,以验证已开发完的模块

本篇文章将介绍 Python 实现 Mock 的几种常见方式

2. Mock 介绍

Mock 测试:在测试验证过程中,对于那些尚未完成或不稳定的对象,用一个虚拟对象来替代,以便测试的测试方法

因此,这个虚拟的对象是 Mock 对象,Mock 对象是真实对象在调试期间的代替品

它的优势包含:

3.1 mock

在 Python 3.3 之前使用 mock,需要先安装依赖

  1. # 安装mock依赖 
  2.  
  3. pip3 install mock 

项目地址:

https://github.com/testing-cabal/mock

假设 Product 类中有 2 个方法

其中,get_product_status_by_id 方法还没有实现;buy_product 方法依赖于 get_product_status_by_id 方法的返回值

  1. # product_impl.py 
  2.  
  3. class Product(object): 
  4.  
  5.     def __init__(self): 
  6.         pass 
  7.  
  8.     def get_product_status_by_id(self, product_id): 
  9.         ""
  10.         通过商品id获取产品信息(Mock) 
  11.         :return
  12.         ""
  13.         # 待实现查询数据库的业务逻辑 
  14.         pass 
  15.  
  16.     def buy_product(self, product_id): 
  17.         ""
  18.         购买产品(真实逻辑) 
  19.         :return
  20.         ""
  21.         # 产品信息 
  22.         # {"id":1,"name":"苹果","num":23} 
  23.         product = self.get_product_status_by_id(product_id) 
  24.  
  25.         if product.get("num") >= 1: 
  26.             result = {"status": 0, "msg""购买成功!"
  27.         else
  28.             result = {"status": 1, "msg""购买失败,库存不足!"
  29.  
  30.         return result 

Mock 的步骤如下:

  1. from mock import patch 
  2. from mock_.product_impl import Product 
  3.  
  4. @patch('mock_.product_impl.Product.get_product_status_by_id'
  5. def test_succuse(mock_get_product_status_by_id): 
  6.     # Mock方法,指定一个返回值 
  7.     mock_get_product_status_by_id.return_value = {"id": 1, "name""苹果""num": 23} 
  8.  
  9.     product = Product() 
  10.  
  11.     assert product.buy_product(1).get("status") == 0 

需要注意的是,Mock 此方法的时候,必须制定该方法的完整路径

使用 @patch.object 同样能完成 Mock,不同的是,@patch.object 包含 2 个参数

第一个参数为该方法所在的类;第二个参数为方法名

  1. from mock import patch 
  2.  
  3. from mock_.product_impl import Product 
  4.  
  5. # Mock一个方法 
  6. # @patch.object:对象、方法名 
  7. @patch.object(Product, 'get_product_status_by_id'
  8. def test_succuse(mock_get_product_status_by_id): 
  9.     # Mock方法,指定一个返回值 
  10.     mock_get_product_status_by_id.return_value = {"id": 1, "name""苹果""num": 23} 
  11.  
  12.     product = Product() 
  13.  
  14.     assert product.buy_product(1).get("status") == 0 

3.2 unittest.mock

Python 3.3 之后,mock 作为标准库,已经内置到 unittest 中了

还是以 3.1 的场景为例,使用 unittest 编写一个测试用例

Mock 步骤如下:

  1. import unittest 
  2. from unittest import mock 
  3.  
  4. from unittest_mock.product_impl import Product 
  5.  
  6. class TestProduct(unittest.TestCase): 
  7.  
  8.     def test_success(self): 
  9.         # 成功结果 
  10.         mock_success_value = {"id": 1, "name""苹果""num": 23} 
  11.  
  12.         product = Product() 
  13.  
  14.         product.get_product_status_by_id = mock.Mock(return_value=mock_success_value) 
  15.  
  16.         # 调用实际函数 
  17.         assert product.buy_product(1).get("status") == 0 
  18.  
  19. if __name__ == "__main__"
  20.     unittest.main() 

3.3 pytest.mock

相比 unittest,pytest 由于强大的插件支持,用户群体可能更大!

如果项目本身使用的框架是 pytest,则 Mock 更建议使用 pytest-mock 这个插件

  1. # pytest依赖 
  2. pip3 install pytest 

Mock 步骤如下:

  1. import pytest 
  2.  
  3. from pytest_mock_.product_impl import Product 
  4.  
  5. def test_buy_product_success(mocker): 
  6.     ""
  7.     购买成功Mock 
  8.     :param mocker: 
  9.     :return
  10.     ""
  11.     # 实例化一个产品对象 
  12.     product = Product() 
  13.  
  14.     # 对Product中的方法的返回值进行Mock 
  15.     mock_value = {"id": 1, "name""苹果""num": 23} 
  16.  
  17.     # Mock方法 
  18.     # 注意:需要指定方法的完整路径 
  19.     # mocker.patch 的第一个参数必须是模拟对象的具体路径,第二个参数用来指定返回值 
  20.     product.get_product_status_by_id = mocker.patch("product_impl.Product.get_product_status_by_id"
  21.                                                     return_value=mock_value) 
  22.  
  23.     # 调用购买产品的方法 
  24.     result = product.buy_product(1) 
  25.  
  26.     assert result.get("status") == 0 

需要注意的是,mocker.patch 方法第一个参数必须是 Mock 对象的完整路径

4. 最后

文中对 Python 中常见的 Mock 方案进行了讲解,实际应用中,建议根据项目实际情况进行选型

 

来源: AirPython内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯