测试Django模拟 - 检查函数/方法调用

Question

我想检查do_a调用do_b。我在做这样的：

代码：

def do_a(...): 
    ... 
    do_b(...) 
    ... 

测试：

def test_do_a(self): 
     ... 

     with patch('...do_b', new_callable=do_nothing()) as mock_do_b: 
      do_a(...) 

     mock_do_b.assert_called_once_with(...) 

而且do_nothing：

def do_nothing(): 
    pass 

这是工作正常，但我不得不使用do_nothing（），我觉得哈克。有没有办法做同样的测试没有额外的无用函数do_nothing（）？

Answer 1

您可以使用补丁作为装饰

@patch('...do_b') 
def test_do_a(self, mock_do_b): 
    do_a(...) 
    mock_do_b.assert_called_once_with(...)