在Python单元测试中修补多个方法的首选方法

在我的单元测试中测试对第四个方法（ _handle_command ）的调用之前，我需要用模拟方法修补三个方法（ _send_reply ， _reset_watchdog和_handle_set_watchdog ）。

从查看模拟软件包的文档，我可以通过几种方法来了解它：

用patch.multiple作为装饰器

@patch.multiple(MBG120Simulator,
                _send_reply=DEFAULT,
                _reset_watchdog=DEFAULT,
                _handle_set_watchdog=DEFAULT,
                autospec=True)
def test_handle_command_too_short_v1(self,
                                     _send_reply,
                                     _reset_watchdog,
                                     _handle_set_watchdog):
    simulator = MBG120Simulator()
    simulator._handle_command('XA99')
    _send_reply.assert_called_once_with(simulator, 'X?')
    self.assertFalse(_reset_watchdog.called)
    self.assertFalse(_handle_set_watchdog.called)
    simulator.stop()

用patch.multiple作为上下文管理器

def test_handle_command_too_short_v2(self):
    simulator = MBG120Simulator()

    with patch.multiple(simulator,
                        _send_reply=DEFAULT,
                        _reset_watchdog=DEFAULT,
                        _handle_set_watchdog=DEFAULT,
                        autospec=True) as mocks:
        simulator._handle_command('XA99')
        mocks['_send_reply'].assert_called_once_with('X?')
        self.assertFalse(mocks['_reset_watchdog'].called)
        self.assertFalse(mocks['_handle_set_watchdog'].called)
        simulator.stop()

使用多个patch.object装饰

@patch.object(MBG120Simulator, '_send_reply', autospec=True)
@patch.object(MBG120Simulator, '_reset_watchdog', autospec=True)
@patch.object(MBG120Simulator, '_handle_set_watchdog', autospec=True)
def test_handle_command_too_short_v3(self,
                                     _handle_set_watchdog_mock,
                                     _reset_watchdog_mock,
                                     _send_reply_mock):
    simulator = MBG120Simulator()
    simulator._handle_command('XA99')
    _send_reply_mock.assert_called_once_with(simulator, 'X?')
    self.assertFalse(_reset_watchdog_mock.called)
    self.assertFalse(_handle_set_watchdog_mock.called)
    simulator.stop()

使用create_autospec手动替换方法

def test_handle_command_too_short_v4(self):
    simulator = MBG120Simulator()

    # Mock some methods.
    simulator._send_reply = create_autospec(simulator._send_reply)
    simulator._reset_watchdog = create_autospec(simulator._reset_watchdog)
    simulator._handle_set_watchdog = create_autospec(simulator._handle_set_watchdog)

    # Exercise.
    simulator._handle_command('XA99')

    # Check.
    simulator._send_reply.assert_called_once_with('X?')
    self.assertFalse(simulator._reset_watchdog.called)
    self.assertFalse(simulator._handle_set_watchdog.called)

就我个人而言，我认为最后一个阅读是最清楚的，如果嘲笑方法的数量增加，不会导致可怕的长线。 它还避免了必须将simulator作为assert_called_once_with的第一个（ self ）参数assert_called_once_with 。

但我没有发现他们中的任何一个特别好。 尤其是多个patch.object方法，它需要仔细匹配参数顺序和嵌套的装饰。

是否有一些我错过的方法，或者让它更易读？ 当你需要在被测试的实例/类上修补多个方法时，你会做什么？

---

不，你没有错过任何与你所提议的完全不同的东西。

关于可读性，我的味道是装饰者的方式，因为它从测试身体中删除嘲笑的东西...但它只是味道。

您是对的：如果您通过autospec=True修补方法的静态实例，则必须在assert_called_*系列检查方法中使用self。 但是你的情况只是一个小课堂，因为你确切知道你需要修补哪些对象，而且你的修补程序并不需要其他上下文而不是测试方法。

您只需修补您的对象即可在所有测试中使用它：通常在测试中，在调用之前您无法修补实例，并且在这些情况下无法使用create_autospec ：您可以修改方法的静态实例。

如果将实例传递给assert_called_*方法assert_called_*您感到困扰，请考虑使用ANY来打破依赖关系。 最后，我写了数百个这样的测试，而且我从来没有遇到过有关参数顺序的问题。

我的标准方法是在你的测试中

@patch('mbgmodule.MBG120Simulator._send_reply', autospec=True)
@patch('mbgmodule.MBG120Simulator._reset_watchdog', autospec=True)
@patch('mbgmodule.MBG120Simulator._handle_set_watchdog', autospec=True)
def test_handle_command_too_short(self,mock_handle_set_watchdog,
                                          mock_reset_watchdog,
                                          mock_send_reply):
    simulator = MBG120Simulator()
    simulator._handle_command('XA99')
    # You can use ANY instead simulator if you don't know it
    mock_send_reply.assert_called_once_with(simulator, 'X?')
    self.assertFalse(mock_reset_watchdog.called)
    self.assertFalse(mock_handle_set_watchdog_mock.called)
    simulator.stop()

修补不在测试方法代码中

每个模拟都以mock_前缀开头

我更喜欢使用简单的patch调用和绝对路径：它清楚并且整洁你在做什么

最后：也许创建simulator并停止它的setUp()和tearDown()责任和测试应考虑只是为了修补某些方法并执行检查。

我希望答案是有用的，但这个问题没有独特的有效答案，因为可读性不是绝对的概念，而是取决于读者。 而且，即使是关于一般情况的标题，问题的例子也是关于应该修补要测试的对象的方法的特定类别的问题。

[编辑]

我虽然有一段时间关于这个问题，但我发现什么让我感到困扰：您正在尝试对私有方法进行测试和理解。 发生这种情况时，首先应该问的是为什么？ 答案很可能是因为这些方法应该是私人合作者的公共方法（这不是我的话）。

在这种新的情况下，你应该感觉私人合作者，你不能只改变你的对象。 你需要做的是修补一些其他类的静态实例。

链接地址: http://www.djcxy.com/p/84243.html

上一篇: Preferred way of patching multiple methods in Python unit test

下一篇: Unwanted overlay of java script picture slide animation with website header