一、什么是mock?

  • mock翻译过来有模拟的意思。主要功能是使用mock对象替代掉指定的python对象，以达到模拟对象的行为的目的。
  • 在python里面mock是辅助单元测试的一个模块，在2.x版本里面属于单独的模块，在python3以后已经集成到了unittest模块当中。

二、使用场景

  • 解决依赖：当测试某个接口或者功能模块时，如果被测接口所依赖的其他接口并没有开发完成，这个时候我们就可以使用mock来模拟依赖接口的返回，进行提前测试。
  • 模拟复杂业务接口：如果测试的某个功能，涉及其他系统，而其他系统测试数据构造相对复杂，该系统并不在本次迭代范围以内。此时可以使用mock来模拟这个复杂系统来降低测试成本提高测试效率。
  • 外部接口联调：当测试过程中需要与第三方公司联调，但是测试环境并没有打通的时候，此时可以根据第三方公司的接口文档来mock各种场景，提前进行我方系统的测试。

三、简单使用

导入模块

from unittest import mock

使用mock模拟函数返回
from unittest import mock
def test01():
    """
    test01函数并未实现，通过mock来模拟该函数的返回
    :return:
    """
    pass

test01 = mock.Mock(return_value=100) 创建Mock对象, 作为Test01的实例方法
print(test01()) 当调用test01方法时, 会返回定义Mock对象时指定的return_value值
输出：100
跟据不同传参返回不同的值
from unittest import mock
def test01():
    """
    根据传参模拟不同返回值
    :return:
    """
    pass

def test02(data):
    mock_data={
        "demo01":"支付中",
        "demo02":"支付成功",
        "demo03":"支付失败"
    }
    return mock_data[data]
test01 = mock.Mock(side_effect=test02)
print(test01("demo02"))
输出：支付成功
函数传参个数检查
from unittest import mock
def test01(x,y,z):
    pass
传参符合规范返回success,否则报错
test01=mock.create_autospec(test01,return_value="success")
print(test01(1,2,3))
输出：success
@path.object()装饰器使用
新建mymock.py
def test01():
    print("啥也不干")


新建demo01.py文件
from unittest.mock import patch
import mymock

@patch.object(target=mymock,attribute="test01")
def test02(mock_fun):
    """
    函数内调用模拟方法并返回模拟值
    demo01:类名称/模块名称
    "test01":被模拟的方法
    mock_fun:接收被模拟的函数
    """
    mock_fun.return_value={'status_code':'200'} 为mock对象设置返回值
    result = mymock.test01() 调用函数就相当于调用mock对象
    print(result)
test02()
输出：{'status_code':'200'}
总结
  • mock还有很多自带方法，而且功能非常强大，但是对于我们测试工程师而言，掌握这几个mock方法，平时工作中就足够了。本篇文章仅介绍mock的常用方法，其他方法感兴趣的同学可以去看官方文档。
  • 官方文档地址：https://docs.python.org/3/library/unittest.mock.html