什麼是嘲諷Python中的類的多個方法的乾淨方式？

Question

在編寫測試時，我經常需要模擬幾個類的方法。目前我正在通過嵌套with陳述包含mock引用例如

from ... import A 


def test_sample(self) 
    instance = A() 
    with mock(A, 'function_1', return_value=1): 
     with mock(A, 'function_2', return_value=2): 
      with mock(A, 'function_3', return_value=3): 
       assert A.function_4, 10 

是否有一個整潔的/推薦的方式做到這一點？能夠刪除幾個嵌套的呼叫將是很好的！

在這個過程中，可能有也可能沒有其他類沒有被模擬的方法。

Answer 1

你不需要有嵌套上下文管理器 - 你可以修補一氣呵成所有的方法：

def test_sample(self) 
    instance = A() 
    with (mock(A, 'function_1', return_value=1), 
      mock(A, 'function_2', return_value=2), 
      mock(A, 'function_3', return_value=3)): 

     assert A.function_4, 10 

或者您可以使用patch.multiple：

from unittest import patch 

def test_sample(self) 
    instance = A() 
    with patch.multiple(A, function_1=2, function_2=2, function_3=3) as patched_values: 
     assert A.function_4, 10