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

什么是mock？

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

mock的作用

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

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

举个栗子

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

场景源码：case.py

1. # -*-coding:utf-8 -*-

3. def login():

4. # 登录接口，尚未开发完成

5. # 登录成功返回：{"result": "success", "message": "登录成功"}

6. # 登录失败返回：{"result": "fail", "message":"账号或密码错误"}

7. # 登录异常返回：{"code": "404", "message": "找不到页面"}

8. # message返回失败原因

10. pass

13. def get_user_info():

14. # 根据登录的结果success or fail，判断跳转到对应页面

16. result = login()

17. print(result)

18. try:

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

20. return "登录成功"

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

22. return "登录失败"

23. else:

24. return "未知失败"

25. except:

26. return "服务端异常"

单元测试用例设计：

1. # -*-coding:utf-8 -*-

2. from unittest import mock

3. import unittest

4. import case

6. class TestLogin(unittest.TestCase):

7. # 单元测试用例

9. def test_login_success(self):

10. # 测试登录成功场景

11. # mock一个支付成功的数据

13. case.login = mock.Mock(return_value={"result": "success", "message": "登录成功"})

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

15. statues = case.get_user_info()

16. self.assertEqual(statues, "登录成功")

18. def test_login_fail(self):

19. # 测试登录失败场景

20. # mock一个登录失败的数据

22. case.login = mock.Mock(return_value={"result": "fail", "message": "账号或密码错误"})

23. # 根据登录结果测试页面跳转

24. statues = case.get_user_info()

25. self.assertEqual(statues, "登录失败")

27. def test_login_fail2(self):

28. # 测试登录失败场景

29. # mock一个登录失败的数据

31. case.login = mock.Mock(return_value={"result": "", "message": "账号或者密码为空"})

32. # 根据登录结果测试页面跳转

33. statues = case.get_user_info()

34. self.assertEqual(statues, "未知失败")

36. def test_login_error(self):

37. # 测试登录异常场景

38. # mock一个登录异常的数据

40. case.login = mock.Mock(return_value={"code": "404", "message": "找不到页面"})

41. # 根据登录结果测试页面跳转

42. statues = case.get_user_info()

43. self.assertEqual(statues, "服务端异常")

45. if __name__ == "__main__":

46. unittest.main()

执行unittest单元测试用例：

 

学习安排上

作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。【保证100%免费】

视频文档获取方式：

这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取