unittest单元测试之mock

<h1 id="什么是mock">什么是mock</h1>

unittest.mock是一个用于在Python中进行单元测试的库，Mock翻译过来就是模拟的意思，顾名思义这个库的主要功能是模拟一些东西。
它的主要功能是使用mock对象替代掉指定的Python对象，以达到模拟对象的行为。

学过python自动化的对unittest并不陌生，unittest其实是单元测试框架，
但对于单元测试，估计很多小伙伴都不懂，单元测试才是自动化测试的至高境界，其中mock是单元测试的脊髓所在

mock能做什么：

• 1.前后端联调，如果你是一个前端页面开发，现在需要开发一个功能：
  下一个订单，支付页面的接口，根据支付结果，支付成功，展示支付成功页，支付失败，展示支付失败页。
  要完成此功能，你需要调用后端的接口，根据返回给你的结果，来展示不同的页面。此时后端接口还没开发好，
  作为一个前端开发总不能等别人开发好了，你再开发，那你只有加班的命了。
  为了同步开发完成任务，此时，你可以根据接口文档的规定，把接口的地址和入参传过去，然后自己mock接口的不同返回界面，来完成前端的开发任务

• 2.单元测试，单元测试的目的是测试某个小小单元的功能，但现实中开发的函数或方法都是有依赖关系的，比如b函数的参数，需要调用a函数的返回结果,但是我前面已经测试a函数了
  这种情况下，就不需要再测一次a函数了，此时就可以用mock模块来模拟调用这部分内容，并给出返回结果

• 3.第三方接口依赖，在做接口自动化的时候，有时候需要调用第三方的接口，但是别人公司的接口服务不受你的控制，有可能别人提供的测试环境今天服务给你开着，别人就关掉了，
  给自动化接口测试带来很多的麻烦，此时就可以自己写一个mock-server来模拟接口的返回数据

mock环境准备

1.python2.x的版本，mock是一个独立的模块，需要用pip安装

pip install -U mock

3.从Python 3.3以后的版本mock已经合并到unittest模块中了，是unittest单元测试的一部分，直接导入过来就行

from unittest import mock

依赖关系

1.如下场景：支付是一个独立的接口，由其它开发提供，根据支付的接口返回状态去显示失败，还是成功，这个是你需要实现的功能

也就是说你写一个b功能，你的同事写一个a功能，你的b功能需要根据a功能的结果去判断，然后实现对应的功能。这就是存在依赖关系，你同事开发的进度你是无法控制的
你要是等他开发完了，你再开发，那你就坐等加班吧.

2.以下是自己写的 zhifu_statues()函数功能，大概设计如下，保存为temple.py文件

# 保存为temple.py
# coding:utf-8

# 作者：上海-悠悠 QQ交流群：588402570
def zhifu():

'''假设这里是一个支付的功能,未开发完

支付成功返回：{"result": "success", "reason":"null"}

支付失败返回：{"result": "fail", "reason":"余额不足"}

reason返回失败原因

'''

pass
def zhifu_statues():

'''根据支付的结果success or fail，判断跳转到对应页面'''

result = zhifu()

print(result)

try:

if result["result"] == "success":

return "支付成功"

elif result["result"] == "fail":

print("失败原因：%s" % result["reason"])

return "支付失败"

else:

return "未知错误异常"

except:

return "Error, 服务端返回异常!"

3.单元测试用例设计

# coding:utf-8
from unittest import mock
import unittest
import temple
# 作者：上海-悠悠 QQ交流群：588402570
class Test_zhifu_statues(unittest.TestCase):

'''单元测试用例'''

def test_01(self):

'''测试支付成功场景'''

# mock一个支付成功的数据

temple.zhifu = mock.Mock(return_value={"result": "success", "reason":"null"})

# 根据支付结果测试页面跳转

statues = temple.zhifu_statues()

print(statues)

self.assertEqual(statues, "支付成功")
<span class="hljs-function"><span class="hljs-keyword">def</span> <span class="hljs-title">test_02</span><span class="hljs-params">(self)</span>:</span>
    <span class="hljs-string">'''测试支付失败场景'''</span>
    <span class="hljs-comment"># mock一个支付成功的数据</span>
    temple.zhifu = mock.Mock(return_value={<span class="hljs-string">"result"</span>: <span class="hljs-string">"fail"</span>, <span class="hljs-string">"reason"</span>: <span class="hljs-string">"余额不足"</span>})
    <span class="hljs-comment"># 根据支付结果测试页面跳转</span>
    statues = temple.zhifu_statues()
    self.assertEqual(statues, <span class="hljs-string">"支付失败"</span>)

if name == "main":

unittest.main()

4.单元测试是为了保证函数的每个分支都测到，以上zhifu_statues()函数有四个分支，也就是说要写四个案例，这里我只写了2个，还有2个案例，你们可以自己练习下