Python单元测试之mock使用

什么是mock？比如当我们需要用一个接口时，这个接口还没有实现或者依赖第三方服务，为了保证当前功能的开发和测试，就需要使用mock模拟这些接口。

Python中使用mock对象替代掉指定的Python对象，实现控制Python对象的行为。mock模块在Python 3.3以后合并到unittest模块中了，可以直接通过导入使用。

Mock基本使用

Mock对象就是mock模块中的一个类的实例，能在整个测试套件中模拟大量的方法。创建后，就可以指定返回值并设置所需的属性，也可以断言调用了哪些方法/属性及其参数。

class Mock(spec=None, side_effect=None, return_value=DEFAULT, wraps=None, name=None, spec_set=None, **kwargs)

Mock类主要的几个参数：

• name:命名一个mock对象，只是起到标识作用，可以通过print查看。
• return_value: 定义mock方法的返回值，可以指定一个值（或者对象），当mock对象被调用时，返回return_value指定的值。
• side_effect: 这个参数指向一个可调用对象，接收一个可迭代序列。可以抛出异常或者动态改变值。当传递这个参数的时候return_value 参数就会失效。

2. from unittest import mock

4. result1 = mock.Mock(name='mock名称')

6. print(result1)

8. mock_value1 = mock.Mock(return_value="返回值1")

10. print(mock_value1())

12. mock_value2 = mock.Mock(return_value="返回值2",side_effect= [1,2,3])

14. print(mock_value2())

16. print(mock_value2())

18. print(mock_value2())

Mock 步骤如下：

• 导入 unittest 框架中的 mock
• 找到要替换的对象A，可以是一个类、函数或者类实例
• 实例化mock对象，设置mock对象的行为，比如调用的时候返回的值，被访问成员的时候返回什么值等。
• 使用mock对象替换对象A
• 调用并断言

mock一个未开发的接口

 

mock一个依赖关系的功能

实际工作中，我们也会遇到这样的场景，测试A模块，然后A模块依赖于B模块的调用，这时就可以借助mock在单元测试中分别测试正常返回和异常返回的情况。

访问baidu的功能，visit_baidu()方法依赖send_request的返回结果。

2. import requests

6. def send_request(url):

8. r = requests.get(url)

10. return r.status_code

14. def visit_baidu():

16. url = 'http://www.baidu.com'

18. return send_request(url)

用mock对象在单元测试中分别测试正常返回和异常返回的情况

2. from unittest import mock

4. import unittest

6. import demo

10. class TestReq(unittest.TestCase):

12. def test_request_01(self):

14. # 实例化mock对象，指定返回值，替换原有对象

16. demo.send_request = mock.Mock(return_value='200')

18. print(demo.send_request())

20. self.assertEqual(demo.visit_baidu(), '200')

23. def test_request_02(self):

25. # 实例化mock对象，指定返回值，替换原有对象

27. demo.send_request = mock.Mock(return_value='404')

29. print(demo.send_request())

31. self.assertEqual(demo.visit_baidu(), '404')

35. if __name__ == '__main__':

37. unittest.main(verbosity=2)

 

Mock的高级用法

mock库提供了patch函数来简化mock对象对原对象的替换，该函数会返回一个mock内部的类实例，它可以控制mock的范围，可以作为装饰器或者上下文管理器使用。

mock.patch（target，new = DEFAULT，spec = None，create = False，spec_set = None，autospec = None，new_callable = None，** kwargs ）

mock装饰器使用格式

• @patch("module名字.方法名")
• @patch.object(类名, "方法名")

patch作为装饰器，需要把你想模拟的函数写在里面，然后在后面的单元测试案例中为它赋一个具体实例，再用return_value 来指定模拟函数返回的结果。

改造上面的单元测试：

2. from unittest import mock

4. import unittest

6. import demo

10. class TestReq(unittest.TestCase):

12. #在测试的参数里对该Mock对象设置一个参数

14. @mock.patch("demo.send_request")

16. def test_request_01(self,mock_request):

18. # 指定一个返回值

20. mock_request.return_value='200'

22. self.assertEqual(demo.visit_baidu(), '200')

25. @mock.patch("demo.send_request")

27. def test_request_02(self,mock_request):

29. # 指定一个返回值

31. mock_request.return_value='404'

33. self.assertEqual(demo.visit_baidu(), '404')

37. if __name__ == '__main__':

39. unittest.main(verbosity=2)

总结：

感谢每一个认真阅读我文章的人！！！

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

软件测试面试文档

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

 

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