Python的单元测试和模拟：检查如果一个函数被调用，但随后停止测试

Question

比方说，我想确保某些标志等得到适当的调度，使之内我的图书馆深，特定的功能被称为：

high_order_function_call(**kwargs) 深内包含library_function_call() 我想确保它实际上被调用。

此给出的典型例子使用mock.patch：

@mock.patch('library') 
def test_that_stuff_gets_called(self, mock_library): 
    high_order_function_call(some_example_keyword='foo') 
    mock_library.library_function_call.assert_called_with(42) 

现在，在这种情况下，我必须等待所有的东西在high_order_function_call整个执行。如果我想要执行停止并在达到mock_library.library_function_call时立即跳回我的单元测试？

Answer 1

您可以尝试在调用中使用异常提升副作用，然后在测试中捕获该异常。

from mock import Mock, patch 
import os.path 

class CallException(Exception): 
    pass 

m = Mock(side_effect=CallException('Function called!')) 
def caller_test(): 
    os.path.curdir() 
    raise RuntimeError("This should not be called!") 

@patch("os.path.curdir", m) 
def test_called(): 
    try: 
     os.path.curdir() 
    except CallException: 
     print "Called!" 
     return 
    assert "Exception not called!" 

if __name__ == "__main__": 
    test_called()