1. 背景
在实际产品开发过程中,某个服务或前端依赖一个服务接口,该接口可能依赖多个底层服务或模块,或第三方接口,比如说服务 A 依赖服务B,服务B又依赖服务 C,如下图所示:
这种依赖的问题会导致原本的需求目的是要验证服务A,但由于所依赖的服务B或者服务C不稳定或者未开发完成,导致工作无法正常开展。
那作为测试工程师,面对这样的情形,我们该怎么办呢?解决这类问题的核心的思路:引入依赖服务替身,更通俗的叫法,引入Mock服务。
今天就结合unittest框架,给大家分享一些关于Mock的一些常见使用。
2. Mock是什么
可能还有些读者之前并没有接触过Mock,不清楚Mock是个啥。
Mock简单来理解,就是在测试过程中,对于某些不容易构造或者不容易获取的对象,用一个虚拟的对象来创建以便测试。而这个虚拟的对象就是mock对象。mock对象就是真实对象在调试期间的代替品。
有时也将Mock服务称之为测试服务替身,或者测试服务档板,下图很形象的描述了Mock的作用。
3. Mock能做什么
就Mock功能而言,本身适用场景较多,但在实际项目中,引入Mock常用来解决的几类,概括起来,主要有:
- 接口间的相互依赖
- 单元测试
- 第三方接口调用
1.前后端联调
比如你是一个前端页面开发,现在需要开发一个功能:下一个订单,支付页面的接口,根据支付结果,支付成功,展示支付成功页,支付失败,展示支付失败页。要完成此功能,你需要调用后端的接口,根据返回给你的结果,来展示不同的页面。此时后端接口还没开发好,作为一个前端开发总不能等别人开发好了,你再开发,那你只有加班的命了。为了同步开发完成任务,此时,你可以根据接口文档的规定,把接口的地址和入参传过去,然后自己mock接口的不同返回界面,来完成前端的开发任务。
2.单元测试
由于单元测试仅针对当前单元进行测试,这就要求所有的内部或者外部依赖都应该是稳定的,采用mock的方法模拟跟本单元依赖的其他单元,可以将测试重点放在当前单元功能,排除外界因素干扰,提升测试精准度。
3.第三方接口依赖
在做接口自动化的时候,有时候需要调用第三方的接口,但是别人公司的接口服务不受你的控制,有可能别人提供的测试环境今天服务给你开着,别人就关掉了,给自动化接口测试带来很多的麻烦,此时就可以通过mock来模拟接口的返回数据,比如模拟各种第三方异常时的返回。
4. Mock实现方式
Mock虽然是作为依赖服务的替身,但并不需要原原本本去构造实现一个完整的服务逻辑,比如现在有一个A服务依赖B服务,需要通过Mock来替换B服务(做一个假的B服务替身)。
那么我们做一个 Mock 服务其实就是做了一个简单的服务 B,它不需要实现原有服务 B 负载的处理逻辑,只要能按服务A需要服务B返回的处理逻辑给出对应返回数据就可以了。
目前常见服务或接口协议主要两种,一种是RPC,另一种是HTTP/HTTPS,mock原理都类似,要么是修改原服务地址为Mock服务地址,要么是拦截原服务的请求Mock返回值,总之就是构造一个假的服务,替代原有服务。
5. Mock市面上常见的解决方案
如果你不想自己动手构建一套Mock解决方案,市面上也提供了很多现存的Mock方案。常用的有:EasyMock
、Mockito
、WireMock
、JMockit
、Mock
、Moco
。
如果你团队技术基础相对比较薄弱,推荐你看看Moco
这个方案,官网如下:
https://github.com/dreamhead/moco/
接下来,重点介绍Python系下Mock方案的使用。
6. Python下unittest.mock使用
unittest.mock是一个用于在Python中进行单元测试的库,顾名思义这个库的主要功能是模拟一些东西。它的主要功能是使用mock对象替代掉指定的Python对象,以达到模拟对象的行为。
需要注意的是在Python2.x版本中,Mock需要单独安装
pip install -U mock
从Python 3.3以后的版本mock已经合并到unittest模块中了,是unittest单元测试的一部分,直接导入过来就行
from unittest import mock
官方文档:
https://docs.python.org/dev/library/unittest.mock.html
unittest.mock模块中最常用的是Mock类。
Mock类库是一个专门用于在unittest过程中制作(伪造)和修改(篡改)测试对象的类库,避免这些对象在单元测试过程中依赖外部资源(网络资源,数据库连接,其它服务以及耗时过长等)
案例:如下场景:支付是一个独立的接口,由其它开发提供,根据支付的接口返回状态去显示失败,还是成功,这个是你需要实现的功能,代码存放在pay.py脚本中:
# !/usr/bin/python3 # -*- coding: utf-8 -*- # @Author : Mike Zhou # @Email : 公众号:测试开发技术 # @File : pay.py def zhifu(): '''假设这里是一个支付的功能,未开发完 支付成功返回:{"result": "success", "msg":"支付成功"} 支付失败返回:{"result": "fail", "msg":"余额不足"} ''' pass def zhifu_statues(): '''根据支付的结果success or fail,判断跳转到对应页面''' result = zhifu() try: if result["result"] == "success": return "支付成功" elif result["result"] == "fail": return "支付失败" else: return "未知错误异常" except: return "Error, 服务端返回异常!"
在zhifu_statues方法中,依赖了zhifu方法,但由于zhifu支付方法的接口是由另外一个同事开发,正常情况下,你同事开发的进度你是无法控制的,需要等他开发完了你才能进行联调你所负责的zhifu_statues接口,因此我们可以通过引入Mock来解决这个问题。
引入mock后单元测试用例代码
# !/usr/bin/python3 # -*- coding: utf-8 -*- # @Author : Mike Zhou # @Email : 公众号:测试开发技术 import unittest from unittest import mock import pay class TestZhifuStatues(unittest.TestCase): '''单元测试用例''' def test_01(self): '''测试支付成功场景''' # mock一个支付成功的数据 pay.zhifu = mock.Mock(return_value={"result": "success", "msg":"支付成功"}) # 根据支付结果测试页面跳转 statues = pay.zhifu_statues() print(statues) self.assertEqual(statues, "支付成功") def test_02(self): '''测试支付失败场景''' # mock一个支付失败的数据 pay.zhifu = mock.Mock(return_value={"result": "fail", "msg": "余额不足"}) # 根据支付结果测试页面跳转 statues = pay.zhifu_statues() print(statues) self.assertEqual(statues, "支付失败") if __name__ == "__main__": unittest.main()
上述代码引入Mock
后,我们就可以顺利完成对支付成功和支付异常两类场景的验证工作。(实际你可以补充更多)
mock
中还有另一种实现方式,通过patch
装饰器的使用,patch
作为函数装饰器,为您创建模拟并将其传递到装饰函数。
用mock.patch实现如下:
# !/usr/bin/python3 # -*- coding: utf-8 -*- # @Author : Mike Zhou # @Email : 公众号:测试开发技术 import unittest from unittest import mock import pay class TestZhifuStatues(unittest.TestCase): '''单元测试用例''' @mock.patch("pay.zhifu") def test_001(self, mock_zhifu): '''测试支付成功场景''' # 方法一:mock一个支付成功的数据 # pay.zhifu = mock.Mock(return_value={"result": "success", "msg":"支付成功"}) # print(pay.zhifu()) # 方法二:mock.path装饰器模拟返回结果 mock_zhifu.return_value = {"result": "success", "msg":"支付成功"} # # 根据支付结果测试页面跳转 statues = pay.zhifu_statues() print(statues) self.assertEqual(statues, "支付成功") @mock.patch("pay.zhifu") def test_002(self, mock_zhifu): '''测试支付失败场景''' # mock一个支付失败的数据 mock_zhifu.return_value = {"result": "fail", "msg": "余额不足"} # 根据支付结果测试页面跳转 statues = pay.zhifu_statues() self.assertEqual(statues, "支付失败") if __name__ == "__main__": unittest.main()