python接口测试之mock(三)

前面介绍了moco的详细的使用，它主要是基于moco-runner-0.11.0-standalone.jar，通过编写json的文件来实现，mock翻译过来就是模拟的意思，也就是说，它是将测试对象所依存的对象替换为虚构对象的库，该虚构对象的调用允许事后查看。Python中有几个库是可以模拟请求和响应的.

1. mock

在python的2.x版本中，mock是属于第三方的库，需要单独的按钮，在python3.3的版本以后，不需要单独的安装，从unitest里直接导入就可以了

接下来我们来看mock库的经常使用的方法以及它的详细的帮助信息，见以下的代码：

class Count:
def add(self):
pass
from unittest import mock
import unittest
class test_add(unittest.TestCase):
def test_add(self):
count = Count()
#找到要替换的对象，这个对象可以是一个类，或者是一个函数，或者是一个类实例
#实例化Mock类得到一个mock对象，并且设置这个mock对象的行为，比如被调用的时候返回什么值，被访问成员的时候返回什么值等
count.add = mock.Mock(return_value=12)
result = count.add(8,4)
self.assertEqual(result,12)
if __name__ == '__main__':
unittest.main()

2. responses库

安装: pip install responses

接下来看看responses的使用方法:

import responses
import requests, unittest
class responsesTest(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
@responses.activate
def test_simple1(self):
responses.add(responses.GET, 'http://twitter.com/api/1/foobar',
json={'error': 'not found'}, status=404)
url = 'http://twitter.com/api/1/foobar'
r = requests.get(url)
# print(r.json())
self.assertEqual(r.json()['error'], 'not found')
if __name__ == '__main__':
unittest.main()

 

import json,responses, requests, unittest
class responseTest(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
@responses.activate
def test_case1(self):
def request_callback(request):
payload = json.loads(request.body)
resp_body = {'value': sum(payload['numbers'])}
headers = {'request-id': '728d329e-0e86-11e4-a748-0c84dc037c13'}
return (200, headers, json.dumps(resp_body))
responses.add_callback(
responses.POST, 'http://calc.com/sum',
callback=request_callback,
content_type='application/json',
)
resp = requests.post(
'http://calc.com/sum',
json.dumps({'numbers': [1, 2, 3]}),
headers={'content-type': 'application/json'},
)
self.assertEqual(resp.json()['value'], 6)
self.assertEqual(resp.status_code,200)
self.assertEqual(resp.headers['request-id'], '728d329e-0e86-11e4-a748-0c84dc037c13')
if __name__ == '__main__':
unittest.main()