Python 如何在测试中使用 Mock

Mock概念

mock 的意思是模拟，也就是模拟接口返回的信息，用已有的信息替换它需要返回的信息，从实现对所依赖的模块的测试。

一般有两种场景：

• 前端对后端接口的 mock，
• 后端服务之间的测试中涉及的mock，常常发生在单元测试的时候。

前端mock可以通过一些工具来完成：

• 使用抓包工具Fiddler，Charles 来实现，通过修改代理返回的数据，实现多种场景的测试。
• 使用一些API管理工具来模拟，比如yapi，Easy Mock 等
• 当然有编码能力的，也可以使用node.js，python的fastAPI来模拟

后端的 Mock 则是从接口的角度，如果一个接口A返回的数据需要依赖于另一个接口B，当敏捷开发中B接口还未开发完全时候这里会需要用到 Mock。

对于测试人员，对接口测试的时候，部分接口尚未开发完成，在约定了接口定义之后，也可以使用 Mock 来模拟。

在 python3.X 中 Mock 模块已经被集成到unittest里面。

Mock类

class Mock(spec=None,side_effect=None,return_value=DEFAULT,name=None)

• spec:定义Mock对象的属性值，可以是一个列表，字符串，一个对象的实例
• side_effect:可以用来抛出异常或者动态改变返回值，可以覆盖return_value
• return_value:定义mock的返回值
• name:作为mock对象的标识可以在print时候看到

简单的例子体验下 Mock 的功能特点

from unittest import mock
def add(num1,num2):
return num1 + num2 # pass
 
add = mock.Mock(return_value=200) # 创建mock对象
 
print( add(10,20) )

你会发现无论输入的参数是什么，输出结果都是200。等于方法被 Mock 拦截处理了。

一个相对正式的 Mock 例子

正常情况：

import requests
def request_scm():
# res = requests.get('http://www.mysx-scm.com')
res = requests.get('http://baidu.com')
return res.status_code

import unittest
from unittest import mock
class TestScmApi(unittest.TestCase):
 
def testUrl(self):
# request_scm = mock.Mock(return_value=200)
self.assertEqual(request_scm(), 200, msg='testUrl 出现错误')

if __name__ == '__main__':
unittest.main()

可以分别把两个 # 注释移到下一句试试。

一个完整的测试例子

import requests

class scmapi():
def request_scm():
res = requests.get('http://www.mysx-scm.com')
# res = requests.get('http://baidu.com')
return res.status_code

def pay_alipay():
'''
待实现
return 200
'''
return 0
 

import unittest
from unittest import mock

class TestScmApi(unittest.TestCase):
 
needmock = True
def setUpClass():
print("setUpClass():所有方法之前执行")

def tearDownClass():
print("tearDownClass():所有方法之后执行")
 
def setUp(self):
self.scmapi = scmapi()
print("setUp():每个方法之前执行")

def tearDown(self):
print("teardown():每个方法之后执行")
 
def test_request_scm(self):
if self.needmock:
scmapi.request_scm = mock.Mock(return_value=200)
self.assertEqual(scmapi.request_scm(), 200, msg='test_request_scm 出现错误')

def test_pay_alipay(self):
if self.needmock:
scmapi.pay_alipay = mock.Mock(return_value=200)
self.assertEqual(scmapi.pay_alipay(), 200, msg='test_pay_alipay 出现错误')

if __name__ == '__main__':
unittest.main()

断言方法

基本的断言方法提供了测试结果是True还是False。所有的断言方法都有一个msg参数，如果指定msg参数的值，则将该信息作为失败的错误信息返回。