避免将第二个参数传递给使用模拟补丁

Question

我在模仿我的RpcClient类进行所有单元测试，如下所示：

import unittest2
from mock import patch

@patch('listeners.RpcClient')
class SomeTestCase(unittest2.TestCase):

    test_something(self, mock_api):
        ...

    test_something_else(self, mock_api):
        ...

对于我的大多数测试，我不想使用mock对象做任何断言，我只想修补这个类，这样RpcClient就不会尝试连接和激发对我的每个测试的请求（我将它连接到我的一个模型上的post save事件）。你知道吗

我能避免把mock_api传入我的每一个测试吗？

Answer 1

最后我用patcher.start()在setUp中进行了模拟：

def setUp(self):
    self.rpc_patcher = patch('listeners.RpcClient')
    self.MockClass = rpc_patcher.start()

def tearDown(self):
    self.rpc_patcher.stop()

所以我不必装饰我的任何测试用例，也不必为我的测试添加任何额外的参数。你知道吗

更多信息：

http://docs.python.org/dev/library/unittest.mock#patch-methods-start-and-stop

Answer 2

你能给mock\u api设置一个默认参数吗？你知道吗

def test_something(self, mock_api=None):
    ...

def test_something_else(self, mock_api=None):

Answer 3

在调用SUT时，可以使用mock.patch作为上下文管理器。比如：

import unittest2
from mock import patch


class SomeTestCase(unittest2.TestCase):

    def call_sut(self, *args, **kwargs):
        with mock.patch('listeners.RpcClient'):
            # call SUT

    def test_something(self):
        self.call_sut()
        # make assertions

    def test_something_else(self):
        self.call_sut()
        # make assertions