Python如何在另一个函数中模拟一个函数

时间:2018-03-05 13:19:43

标签: python python-3.x unit-testing mocking python-unittest

我对破坏单元测试的功能进行了一些更改。以前我有一个包含函数request_url的common.py:

npm update --save/--save-dev

这就是测试:

import requests

def request_url(method, url):
    return _request_url(method, url)

def _request_url(method, url):
    return requests.request(method, url)

在功能发生变化之后,我首先发起了一个会话,安装了一个TLS适配器,然后调用会话的请求函数,而不是简单地调用requests.request:

@mock.patch("requests.request", autospec=True)
def test_request_url_200(self, mock_request):

    response = mock.MagicMock()
    response.status_code = 200
    method = mock.sentinel.method
    path = "url"

    result = common.request_url(
        method,
        path
    )

    self.assertListEqual([
        mock.call(
            mock.sentinel.method,
            "url"
        ),
    ], list(mock_request.mock_calls))

    self.assertListEqual([mock.call.raise_for_status()], list(response.mock_calls))
    self.assertEqual(mock_request.return_value, result)

在这里,我不确定如何完全模拟session.request,因为可以通过会话变量获得。我尝试修补 requests.session.request ,但这不起作用。

有没有人知道如何嘲笑这个功能?

谢谢!

1 个答案:

答案 0 :(得分:3)

我认为@mock.patch(...)中的原因。您设置了autospec=True,但您的mock_request并未返回数据(在您的情况下每次都是Mock)。 The documentation说:

  

如果设置autospec = True,则将使用规范创建模拟   从被替换的对象。模拟的所有属性也将   具有对象的相应属性的规范   更换。 ...

尝试致电:

print(mock_request.return_value) # <MagicMock name='request()()' id='4343989392'>
# or
# print(mock_request.return_value.return_value) # <MagicMock name='request()()()' id='4344191568'>

正如您所看到的,requests.request已被嘲笑&#39;但是没有返回任何数据。您可以使用return_valueside_effect设置预期数据。这是一个例子:

from unittest import TestCase
import mock
from common import request_url


class MyTestExample(TestCase):

    def test_request_url_1(self):
        mocked_request = mock.patch('requests.request', side_effect=['one', 'two', 'tree', ])
        mocked_request.start()
        # request_url(...) will return item from list
        self.assertEqual('one', request_url('test', 'url'))
        self.assertEqual('two', request_url('test', 'url'))
        self.assertEqual('tree', request_url('test', 'url'))

        mocked_request.stop()

    def test_request_url_2(self):
        result = {'my': 'dict'}
        mocked_request = mock.patch('requests.request', return_value=result)
        mocked_request.start()
        # request_url(...) will return static data 
        self.assertEqual(result, request_url('test', 'url'))
        self.assertEqual(result, request_url('test', 'url'))
        self.assertEqual(result, request_url('test', 'url'))

        mocked_request.stop()

因此,您只需要描述预期的数据。您也可以mock使用httpretty

希望这会有所帮助。