[Python] 在测试中使用 Mock

2022-08-28 13:20:22

Mock概念
Mock类
简单的例子体验下 Mock 的功能特点
一个相对正式的 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参数的值，则将该信息作为失败的错误信息返回。

序号	断言方法	断言描述
1	assertEqual(arg1, arg2, msg=None)	验证arg1=arg2，不等则fail
2	assertNotEqual(arg1, arg2, msg=None)	验证arg1 != arg2, 相等则fail
3	assertTrue(expr, msg=None)	验证expr是true，如果为false，则fail
4	assertFalse(expr,msg=None)	验证expr是false，如果为true，则fail
5	assertIs(arg1, arg2, msg=None)	验证arg1、arg2是同一个对象，不是则fail
6	assertIsNot(arg1, arg2, msg=None)	验证arg1、arg2不是同一个对象，是则fail
7	assertIsNone(expr, msg=None)	验证expr是None，不是则fail
8	assertIsNotNone(expr, msg=None)	验证expr不是None，是则fail
9	assertIn(arg1, arg2, msg=None)	验证arg1是arg2的子串，不是则fail
10	assertNotIn(arg1, arg2, msg=None)	验证arg1不是arg2的子串，是则fail
11	assertIsInstance(obj, cls, msg=None)	验证obj是cls的实例，不是则fail
12	assertNotIsInstance(obj, cls, msg=None)	验证obj不是cls的实例，是则fail

码农公寓