三年杂谈 | 测试之Mock - 51Testing软件测试论坛

# a simple module with a simple class
　　class AClass(object):
　　 def method(self, an_argument):
　　 print "I typed {}".format(an_argument)

复制代码

>>> from mock import MagicMock
　　>>> import amodule
　　>>> thing = amodule.AClass()
　　# method方法替换成MagicMock实例
　　# 这里传入参数指定了return_value，因此调用MagicMock伪造的method方法，将会返回3！
　　>>> thing.method = MagicMock(return_value=3)
　　>>> print thing.method('an_arg')
　　3
　　# assert_called_with 即验证，我们最近一次调用method方法，是传入了'others'来调用的
　　# 但是，显然不是:( 因此这行会报错
　　>>> print thing.method.assert_called_with('others')
　　...
　　AssertionError: Expected call: mock('others')
　　Actual call: mock('an_arg')

复制代码

import mock
　　import requests
　　import json
　　import unittest
　　def get_request():
　　 req = requests.get("http://www.baidu.com")
　　 # do sth...
　　 return req.status_code == 200
　　class TestClient(unittest.TestCase):
　　 @mock.patch('requests.get')
　　 def test_get_request(self, mock_request):
　　 mock_request.return_value = mock.MagicMock(
　　 status_code=200, response=json.dumps({'key': 'value'})
　　 )
　　 self.assertEqual(get_request(), True)
　　if __name__ == '__main__':
　　 unittest.main()

复制代码

# result
　　$ python request_get_mock.py
　　.
　　----------------------------------------------------------------------
　　Ran 1 test in 0.001s
　　OK

复制代码

# coding=utf-8
　　import unittest
　　import retrying
　　import mock
　　import requests
　　import json
　　class RequestError(Exception):
　　 def __init__(self, server):
　　 self.message = "Request failed to {}.".format(server)
　　class Blog(object):
　　 '''
　　 Blog类包含一些博客文章相关的操作封装
　　类在初始化后，post方法提供对指定文章链接内容的获取并返回JSON的格式
　　 '''
　　 def __init__(self, name):
　　 self.name = name
　　 @retrying.retry(stop_max_attempt_number=3, wait_fixed=100)
　　 def post(self, article_path):
　　 '''
　　根据获取文章的内容
　　 '''
　　 response = requests.get("http://{name}/{path}".format(
　　 name=self.name,
　　 path=article_path))
　　 if response.status_code >= 200 and response.status_code < 300:
　　 return response.text
　　 else:
　　 raise RequestError(self.name)
　　 def __repr__(self):
　　 return '<Blog: {}>'.format(self.name)
　　class TestBlog(unittest.TestCase):
　　 '''
　　我们需要测试哪些内容？根据K大的http://docs.python-guide.org/en/latest/writing/tests/
　　我们关注本身的功能的覆盖，那么主要需要测试：
　　 1. 正常请求时是否能获得正确的JSON数据
　　 2. 异常返回时是否抛出RequestError异常
　　 3. 遇到错误是否会重试
　　 '''
　　 @mock.patch('requests.get')
　　 def test_blog_post(self, mock_get):
　　 mocked_content = json.dumps({
　　 'status': True, 'content': 'This is the post content'
　　 })
　　 mocked_success = mock.MagicMock(
　　 status_code=200,
　　 headers={'content-type': "application/json"},
　　 text=mocked_content)
　　 mock_get.return_value = mocked_success
　　 # new blog instance and call `post`.
　　 # confirm return value on succcess.
　　 b = Blog(name="devopstarter.info")
　　 content = b.post("three-years-later-test-mock")
　　 self.assertEqual(content, mocked_content)
　　 mock_get.reset_mock() # reset mock, include called count etc.
　　 mocked_error = mock.MagicMock(
　　 status_code=500,
　　 headers={'content-type': "application/json"},
　　 text="Internal server error")
　　 # confirm raise exception on failure.
　　 # due to mock object return_value is always error, retry is also error.
　　 mock_get.return_value = mocked_error
　　 with self.assertRaises(RequestError):
　　 b.post("three-years-later-test-mock")
　　 # confirm the retry works in failure case, call_count exceeded 3!
　　 self.assertEqual(mock_get.call_count, 3)
　　 # confirm the retry works while ONLY one time failed.
　　 # `side_effect` helped us complete this goal!
　　 mock_get.reset_mock() # reset mock, include called count etc.
　　 mock_get.side_effect = [
　　 requests.ConnectionError('Test error'),
　　 mocked_success
　　 ]
　　 content = b.post("three-years-later-test-mock")
　　 self.assertEqual(content, mocked_content)
　　if __name__ == '__main__':
　　 unittest.main()

复制代码

# mymodule.py
　　def Hello():
　　 print "Hello, Jacky!"

复制代码

# monkeypatch.py
　　import sys
　　import mymodule
　　# patched method `Hello`
　　def Hello():
　　 print "Hello, Monkey!"
　　# monkey patch
　　class monkeypatch(object):
　　 def __init__(self):
　　 mymod = self._get_global_module("mymodule")
　　 # Methods to replace
　　 self.methods = ('Hello', )
　　 # Store the original methods
　　 self.orig_methods = dict(
　　 (m, mymod.__dict__[m]) for m in self.methods)
　　 # Monkey patch
　　 g = globals()
　　 for m in self.methods:
　　 # mymod's method has been replaced!
　　 mymod.__dict__[m] = g[m]
　　 def _get_global_module(self, mod):
　　 return sys.modules[mod]
　　 def __enter__(self):
　　 return self
　　 def __exit__(self, *args):
　　 mymod = self._get_global_module("mymodule")
　　 # restore the method
　　 for m in self.methods:
　　 mymod.__dict__[m] = self.orig_methods[m]
　　if __name__ == '__main__':
　　 monkeypatch()
　　 # under this hood, mymodule's Hello has been monkeypatched.
　　 mymodule.Hello()

复制代码

# actual call
　　$ python monkeypatch.py
　　Hello, Monkey!

复制代码

In [6]: import mock
　　In [7]: m = mock.MagicMock()
　　In [8]: isinstance(m, mock.MagicMock)
　　Out[8]: True
　　In [9]: isinstance(m.m, mock.MagicMock)
　　Out[9]: True
　　In [10]: isinstance(m.m.m, mock.MagicMock)
　　Out[10]: True

复制代码

...
　　 def _get_child_mock(self, **kw):
　　 """Create the child mocks for attributes and return value.
　　 By default child mocks will be the same type as the parent.
　　 Subclasses of Mock may want to override this to customize the way
　　 child mocks are made.
　　 For non-callable mocks the callable variant will be used (rather than
　　 any custom subclass)."""
　　...

复制代码

# https://github.com/testing-cabal/mock/blob/master/mock/mock.py#L1203
　　# _importer 方法实现的即动态根据"class.module"引入相应的原始模块
　　# https://github.com/testing-cabal/mock/blob/master/mock/mock.py#L1270
　　# patch是通过猴子补丁实现方法和类的mock，没错，就是前面介绍到的monkeypatch，这也是mock模块的主要实现思想！
　　# 其他一些代码也比较有趣，奈何功力有限，读的不是很清晰，就不误人子弟了，:)
　　...

复制代码