Python中的Mock

发布时间:2022-03-10 09:51:22 人气:45 作者:多测师

  1. 前言

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

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

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

  2. Mock 介绍

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

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

  它的优势包含:

  ·前、后端并行开发

  ·模拟无法访问的资源

  ·隔离系统,避免脏数据干扰测试结果

  3.1 mock

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

  # 安装mock依赖

  pip3 install mock

  项目地址:

  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 方法的返回值。

  # 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 编写一个测试用例

Python中的Mock

  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

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

  4. 最后

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

  以上内容为大家介绍了Python中的Mock,希望对大家有所帮助,如果想要了解更多Python相关知识,请关注多测师。https://www.e70w.com/xwzx/

返回列表
在线客服
联系方式

热线电话

17727591462

上班时间

周一到周五

二维码
线