Python中Mock如何使用
Python中Mock如何使用,针对这个问题,这篇文章详细介绍了相对应的分析和解答,希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。
成都创新互联公司是一家专注于网站设计、网站建设与策划设计,永顺网站建设哪家好?成都创新互联公司做网站,专注于网站建设十载,网设计领域的专业建站公司;建站业务涵盖:永顺等地区。永顺做网站价格咨询:13518219792
1. 前言
微服务架构下,由于各类服务开发进度的不一致,导致联调工作经常会存在不确定性,进而导致项目延期
在实际工作中,为了保证项目进度,我们经常需要针对部分未完成模块及不稳定模块采用 Mock 方式,以验证已开发完的模块
本篇文章将介绍 Python 实现 Mock 的几种常见方式
2. Mock 介绍
Mock 测试:在测试验证过程中,对于那些尚未完成或不稳定的对象,用一个虚拟对象来替代,以便测试的测试方法
因此,这个虚拟的对象是 Mock 对象,Mock 对象是真实对象在调试期间的代替品
它的优势包含:
前、后端并行开发
模拟无法访问的资源
隔离系统,避免脏数据干扰测试结果
3.1 mock
在 Python 3.3 之前使用 mock,需要先安装依赖
# 安装mock依赖 pip3 install mock
假设 Product 类中有 2 个方法
get_product_status_by_id
buy_product
其中,get_product_status_by_id 方法还没有实现;buy_product 方法依赖于 get_product_status_by_id 方法的返回值
# product_impl.py class Product(object): def __init__(self): pass def get_product_status_by_id(self, product_id): """ 通过商品id获取产品信息(Mock) :return: """ # 待实现查询数据库的业务逻辑 pass def buy_product(self, product_id): """ 购买产品(真实逻辑) :return: """ # 产品信息 # {"id":1,"name":"苹果","num":23} product = self.get_product_status_by_id(product_id) if product.get("num") >= 1: result = {"status": 0, "msg": "购买成功!"} else: result = {"status": 1, "msg": "购买失败,库存不足!"} return result
Mock 的步骤如下:
导入使用 mock 中的 patch 方法
作为测试方法的装饰器,对 get_product_status_by_id 方法进行 Mock,方法参数为 Mock 对象
测试方法中,对该 Mock 对象设置一个返回值
调用并断言
from mock import patch from mock_.product_impl import Product @patch('mock_.product_impl.Product.get_product_status_by_id') def test_succuse(mock_get_product_status_by_id): # Mock方法,指定一个返回值 mock_get_product_status_by_id.return_value = {"id": 1, "name": "苹果", "num": 23} product = Product() assert product.buy_product(1).get("status") == 0
需要注意的是,Mock 此方法的时候,必须制定该方法的完整路径
使用 @patch.object 同样能完成 Mock,不同的是,@patch.object 包含 2 个参数
第一个参数为该方法所在的类;第二个参数为方法名
from mock import patch from mock_.product_impl import Product # Mock一个方法 # @patch.object:对象、方法名 @patch.object(Product, 'get_product_status_by_id') def test_succuse(mock_get_product_status_by_id): # Mock方法,指定一个返回值 mock_get_product_status_by_id.return_value = {"id": 1, "name": "苹果", "num": 23} product = Product() assert product.buy_product(1).get("status") == 0
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
调用并断言
import unittest from unittest import mock from unittest_mock.product_impl import Product class TestProduct(unittest.TestCase): def test_success(self): # 成功结果 mock_success_value = {"id": 1, "name": "苹果", "num": 23} product = Product() product.get_product_status_by_id = mock.Mock(return_value=mock_success_value) # 调用实际函数 assert product.buy_product(1).get("status") == 0 if __name__ == "__main__": unittest.main()
3.3 pytest.mock
相比 unittest,pytest 由于强大的插件支持,用户群体可能更大!
如果项目本身使用的框架是 pytest,则 Mock 更建议使用 pytest-mock 这个插件
# pytest依赖 pip3 install pytest
Mock 步骤如下:
使用 pytest 编写测试方法,参数为 mocker
实例化 Product 对象
使用 mocker.patch() 方法对 get_product_status_by_id 方法进行 Mock,并设置返回值
调用并断言
import pytest from pytest_mock_.product_impl import Product def test_buy_product_success(mocker): """ 购买成功Mock :param mocker: :return: """ # 实例化一个产品对象 product = Product() # 对Product中的方法的返回值进行Mock mock_value = {"id": 1, "name": "苹果", "num": 23} # Mock方法 # 注意:需要指定方法的完整路径 # mocker.patch 的第一个参数必须是模拟对象的具体路径,第二个参数用来指定返回值 product.get_product_status_by_id = mocker.patch("product_impl.Product.get_product_status_by_id", return_value=mock_value) # 调用购买产品的方法 result = product.buy_product(1) assert result.get("status") == 0
关于Python中Mock如何使用问题的解答就分享到这里了,希望以上内容可以对大家有一定的帮助,如果你还有很多疑惑没有解开,可以关注创新互联行业资讯频道了解更多相关知识。
标题名称:Python中Mock如何使用
标题链接:http://myzitong.com/article/ipgppj.html