python技术栈 之 单元测试中mock的使用

一、什么是mock？

mock测试就是在测试过程中，对于某些不容易构造或者不容易获取的对象，用一个虚拟的对象来创建以便测试的测试方法。

二、mock的作用

特别是开发过程中上下游未完成的工序导致当前无法测试，需要虚拟某些特定对象以便测试。

unittest是python内置的单元测试库，在做接口测试时，如果开发的接口未开发出来，我们如果想要测试接口联调，又不能干等着，这时可以使用unittest.mock模拟接口返回，进行接口测试。

三、举个栗子

示例：1、新增登录和获取个人信息两个接口，分别由A和B开发。2、A登录接口尚未开发完成，B获取个人信息接口已经开发完成。3、已知登录接口返回3种状态：登录成功，登录失败，登录异常。

场景源码：case.py

如果你想学习接口自动化测试，我这边给你推荐一套视频，这个视频可以说是B站播放全网第一的接口自动化测试教程，同时在线人数到达1000人，并且还有笔记可以领取及各路大神技术交流：798478386      

【已更新】B站讲的最详细的Python接口自动化测试实战教程全集（实战最新版）_哔哩哔哩_bilibili【已更新】B站讲的最详细的Python接口自动化测试实战教程全集（实战最新版）共计200条视频，包括：1.【接口自动化】目前软件测试的市场行情以及测试人员能力标准。、2.【接口自动化】全面熟练Requests库以及底层方法调用逻辑、3.【接口自动化】接口自动化实战及正则和JsonPath提取器的应用等，UP主更多精彩视频，请关注UP账号。https://www.bilibili.com/video/BV17p4y1B77x/?spm_id_from=333.337

# -*-coding:utf-8 -*-def login():# 登录接口，尚未开发完成# 登录成功返回：{"result": "success", "message": "登录成功"}# 登录失败返回：{"result": "fail", "message":"账号或密码错误"}# 登录异常返回：{"code": "404", "message": "找不到页面"}# message返回失败原因passdef get_user_info():# 根据登录的结果success or fail，判断跳转到对应页面result = login()print(result)try:if result["result"] == "success":return "登录成功"elif result["result"] == "fail":return "登录失败"else:return "未知失败"except:return "服务端异常"

单元测试用例设计：

# -*-coding:utf-8 -*-
from unittest import mock
import unittest
import caseclass TestLogin(unittest.TestCase):# 单元测试用例def test_login_success(self):# 测试登录成功场景# mock一个支付成功的数据case.login = mock.Mock(return_value={"result": "success", "message": "登录成功"})# 根据支付结果测试页面跳转statues = case.get_user_info()self.assertEqual(statues, "登录成功")def test_login_fail(self):# 测试登录失败场景# mock一个登录失败的数据case.login = mock.Mock(return_value={"result": "fail", "message": "账号或密码错误"})# 根据登录结果测试页面跳转statues = case.get_user_info()self.assertEqual(statues, "登录失败")def test_login_fail2(self):# 测试登录失败场景# mock一个登录失败的数据case.login = mock.Mock(return_value={"result": "", "message": "账号或者密码为空"})# 根据登录结果测试页面跳转statues = case.get_user_info()self.assertEqual(statues, "未知失败")def test_login_error(self):# 测试登录异常场景# mock一个登录异常的数据case.login = mock.Mock(return_value={"code": "404", "message": "找不到页面"})# 根据登录结果测试页面跳转statues = case.get_user_info()self.assertEqual(statues, "服务端异常")if __name__ == "__main__":unittest.main()

执行unittest单元测试用例：