[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参数的值,则将该信息作为失败的错误信息返回。

序号 断言方法 断言描述
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
上一篇:Shell脚本编程基础之shell脚本算术运算


下一篇:JavaScript浮点计算