如何嘲笑非呼叫反对在Python单元测试库

Question

方法我已经有了一个真正有用的蟒蛇方法是这样的：

def stop_widget(): 
    original_widget = load_widget_from_file() 
    if original_widget: 
    original_widget.close() 

当我想测试它以

确保我叫close()，我做：

@patch('load_widget_from_file') 
def test_stop_widget_with_original_widget(self, lwff_mock): 
    mock_widget = create_autospec(Widget) 
    lwff_mock.return_value = mock_widget 
    stop_widget() 
    mock_widget.close.assert_called_once_with() 

，但我该怎么办时，我想测试不调用close时load_widget_from_file返回值不计算为真？

如果我试图使另一个单元测试有：

@patch('load_widget_from_file') 
def test_stop_widget_with_original_widget(self, lwff_mock): 
    mock_widget = None 
    lwff_mock.return_value = mock_widget 
    stop_widget() 
    mock_widget.close.assert_not_called() 

这将炸毁。

Answer 1

您需要创建第二个模拟对象（而不是设置mock_widget = None），但小部件需要是falsey以防止函数进入if条件。

在Python，对象始终是“truthy”，除非有一个零长度，或它具有__bool__ method that returns False：

对象.__布尔__（个体）

调用，以实现真值测试和内置操作bool();应该返回False或True。如果未定义此方法，则调用__len__()（如果已定义），并且如果该对象的结果非零，则该对象被视为真。如果一个类既没有定义__len__()也没有定义__bool__()，它的所有实例都被认为是真实的。

大概是“正确的方式”到__bool__方法添加到您的模拟对象将use MagicMock，它附带了很多的Python的“魔术方法”预先定义的。但你也必须将返回值更改为False，如下所示：

@patch('load_widget_from_file') 
def test_stop_widget_with_original_widget(self, lwff_mock): 
    mock_widget = MagicMock() 
    mock_widget.__bool__.return_value = False 
    lwff_mock.return_value = mock_widget 
    stop_widget() 
    mock_widget.close.assert_not_called()