10-Mock模拟接口返回数据

1、安装mock

方法一：pip安装

命令行直接输入：pip install mock

方法二：官网下载mock安装包安装

下载安装包后，解压，命令行进入解压目录，执行python setup.py install

2、mock模拟返回值

1）不封装，mock模拟返回值

# encoding:utf-8
from mock import mock
import unittest
from base.requestsDemo import SendMain

# 创建测试类,继承unittest.TestCase
class MockTest(unittest.TestCase):
    def setUp(self):
        self.run = SendMain()

    def test_1(self):
        # 定义请求的post数据并返回
        postData = {'username': 'wangling', 'password': ''}
        # 定义请求的post url
        postUrl = 'http://127.0.0.1:8000/postDataHandle/'
        # 定义请求的响应结果
        respnse_data = postData
        # 响应数据
        mockData = mock.Mock(return_value=respnse_data)
        # mockData是send_main方法的模拟值
        self.run.send_main = mockData
        res = self.run.send_main(postUrl, 'POST', postData)
        print res

    def test_2(self):
        getData = {'username': 'xiaohua', 'mobile': '', 'data': 'haha'}
        getUrl = 'http://127.0.0.1:8000/getDataHandle'
        mockData = mock.Mock(return_value=getData)
        self.run.send_main = mockData
        res = self.run.send_main(getUrl, 'GET', getData)
        print res

if __name__ == '__main__':
    unittest.main()

2）封装，mock模拟返回值

可以将mock模拟返回结果封装一下，使代码重用性更高，具体如下：

# coding:utf-8
from mock import mock

# 模拟mock封装
def mock_test(mock_method,url, method,response_data,request_data=None):
    mock_method = mock.Mock(return_value=response_data)
    res = mock_method(url, method, request_data)
    return res

调用如下：

# encoding:utf-8
import unittest
from base.requestsDemo import SendMain
from base.mockdemoRe import mock_test

# 创建测试类,继承unittest.TestCase
class MockTest(unittest.TestCase):
    def setUp(self):
        self.run = SendMain()

    def test_2(self):
        getData = {'username': 'xiaohua', 'mobile': '', 'data': 'haha'}
        getUrl = 'http://127.0.0.1:8000/getDataHandle'
        respnseDate = getData
        mock_test(self.run.send_main,getUrl,'GET',respnseDate,getData)
        res = self.run.send_main(getUrl, 'GET', getData)
        print res

if __name__ == '__main__':
    unittest.main()