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

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

公司主营业务：网站设计制作、成都网站制作、移动网站开发等业务。帮助企业客户真正实现互联网宣传，提高企业的竞争能力。创新互联是一支青春激扬、勤奋敬业、活力青春激扬、勤奋敬业、活力澎湃、和谐高效的团队。公司秉承以“开放、自由、严谨、自律”为核心的企业文化，感谢他们对我们的高要求，感谢他们从不同领域给我们带来的挑战，让我们激情的团队有机会用头脑与智慧不断的给客户带来惊喜。创新互联推出旅顺口免费做网站回馈大家。

 1. 前言

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

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

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

2. Mock 介绍

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

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

它的优势包含：

  • 前、后端并行开发
  • 模拟无法访问的资源
  • 隔离系统，避免脏数据干扰测试结果

3.1 mock

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

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

项目地址：

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

假设 Product 类中有 2 个方法

  • get_product_status_by_id
  • buy_product

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

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

Mock 的步骤如下：

  • 导入使用 mock 中的 patch 方法
  • 作为测试方法的装饰器，对 get_product_status_by_id 方法进行 Mock，方法参数为 Mock 对象
  • 测试方法中，对该 Mock 对象设置一个返回值
  • 调用并断言
 
 
 
    
  
  
  
  1. from mock import patch 
    2. 
  
  
  
  2. from mock_.product_impl import Product 
    3. 
  
  
  
  3.  
    4. 
  
  
  
  4. @patch('mock_.product_impl.Product.get_product_status_by_id') 
    5. 
  
  
  
  5. def test_succuse(mock_get_product_status_by_id): 
    6. 
  
  
  
  6.     # Mock方法，指定一个返回值 
    7. 
  
  
  
  7.     mock_get_product_status_by_id.return_value = {"id": 1, "name": "苹果", "num": 23} 
    8. 
  
  
  
  8.  
    9. 
  
  
  
  9.     product = Product() 
    10. 
  
  
  
  10.  
    11. 
  
  
  
  11.     assert product.buy_product(1).get("status") == 0 
    12.

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

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

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

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

3.2 unittest.mock

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

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

Mock 步骤如下：

  • 导入 unittest 框架中的 mock 文件
  • 实例化 Product 对象
  • mock.Mock(return_value=*) 方法
  • 对 get_product_status_by_id 方法进行 Mock
  • 调用并断言
 
 
 
    
  
  
  
  1. import unittest 
    2. 
  
  
  
  2. from unittest import mock 
    3. 
  
  
  
  3.  
    4. 
  
  
  
  4. from unittest_mock.product_impl import Product 
    5. 
  
  
  
  5.  
    6. 
  
  
  
  6. class TestProduct(unittest.TestCase): 
    7. 
  
  
  
  7.  
    8. 
  
  
  
  8.     def test_success(self): 
    9. 
  
  
  
  9.         # 成功结果 
    10. 
  
  
  
  10.         mock_success_value = {"id": 1, "name": "苹果", "num": 23} 
    11. 
  
  
  
  11.  
    12. 
  
  
  
  12.         product = Product() 
    13. 
  
  
  
  13.  
    14. 
  
  
  
  14.         product.get_product_status_by_id = mock.Mock(return_value=mock_success_value) 
    15. 
  
  
  
  15.  
    16. 
  
  
  
  16.         # 调用实际函数 
    17. 
  
  
  
  17.         assert product.buy_product(1).get("status") == 0 
    18. 
  
  
  
  18.  
    19. 
  
  
  
  19. if __name__ == "__main__": 
    20. 
  
  
  
  20.     unittest.main() 
    21.

3.3 pytest.mock

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

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

 
 
 
    
  
  
  
  1. # pytest依赖 
    2. 
  
  
  
  2. pip3 install pytest 
    3.

Mock 步骤如下：

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

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

4. 最后

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

网站名称：Python中Mock到底该怎么玩？一篇文章告诉你
网站地址：http://www.csdahua.cn/qtweb/news15/103065.html

网站建设、网络推广公司-快上网，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 快上网

成都快上网为您推荐相关内容

解决方案知识

同城分类信息

重庆app软件开发　　 　边坡柔性防护网　　 　成都未宇聚　　 　重庆网站建设　　 　广汉锦华　　 　企业手机网站建设　　 　力比特机械　　 　德阳柴油发电机组　　 　成都楼顶广告牌　　 　成都机柜租用　　 　企业网站设计　　 　微信小程序公众号开发　　 　绵阳网站建设　　 　德阳机房托管　　 　营山产后修复　　 　合江网站建设　　 　成都网站制作　　 　内江做网站　　 　重庆托管服务器　　 　开元建筑学校