生產類構造函數需要額外參數時，爲什麼unittest.mock失敗？

Question

我遇到了一個問題，我認爲這可能是我使用的庫的一個錯誤。不過，我對python，unittest和unittest.mock庫相當陌生，所以這可能只是我理解中的一個漏洞。

同時加入測試，一些生產代碼，我遇到了一個錯誤，我已經產生再現這一問題的最少的樣品：

import unittest 
import mock 

class noCtorArg: 
    def __init__(self): 
     pass 
    def okFunc(self): 
     raise NotImplemented 


class withCtorArg: 
    def __init__(self,obj): 
     pass 
    def notOkFunc(self): 
     raise NotImplemented 
    def okWithArgFunc(self, anArgForMe): 
     raise NotImplemented 

class BasicTestSuite(unittest.TestCase): 
    """Basic test Cases.""" 

    # passes 
    def test_noCtorArg_okFunc(self): 
     mockSUT = mock.MagicMock(spec=noCtorArg) 
     mockSUT.okFunc() 
     mockSUT.assert_has_calls([mock.call.okFunc()]) 

    # passes 
    def test_withCtorArg_okWithArgFuncTest(self): 
     mockSUT = mock.MagicMock(spec=withCtorArg) 
     mockSUT.okWithArgFunc("testing") 
     mockSUT.assert_has_calls([mock.call.okWithArgFunc("testing")]) 

    # fails 
    def test_withCtorArg_doNotOkFuncTest(self): 
     mockSUT = mock.MagicMock(spec=withCtorArg) 
     mockSUT.notOkFunc() 
     mockSUT.assert_has_calls([mock.call.notOkFunc()]) 


if __name__ == '__main__': 
    unittest.main() 

如何我進行的測試，輸出如下：

E:\work>python -m unittest testCopyFuncWithMock 
.F. 
====================================================================== 
FAIL: test_withCtorArg_doNotOkFuncTest (testCopyFuncWithMock.BasicTestSuite) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "testCopyFuncWithMock.py", line 38, in test_withCtorArg_doNotOkFuncTest 
    mockSUT.assert_has_calls([mock.call.notOkFunc()]) 
    File "C:\Python27\lib\site-packages\mock\mock.py", line 969, in assert_has_calls 
    ), cause) 
    File "C:\Python27\lib\site-packages\six.py", line 718, in raise_from 
    raise value 
AssertionError: Calls not found. 
Expected: [call.notOkFunc()] 
Actual: [call.notOkFunc()] 

---------------------------------------------------------------------- 
Ran 3 tests in 0.004s 

FAILED (failures=1) 

我使用python 2.7.11，通過pip安裝模擬版本2.0.0。

對於我在做什麼的任何建議是錯誤的？或者這看起來像庫中的錯誤？

Answer 1

有趣的是，您選擇執行斷言的方式掩蓋了您的問題。

嘗試，而不是這樣的：

mockSUT.assert_has_calls(calls=[mock.call.notOkFunc()]) 

做到這一點：

mockSUT.assert_has_calls(calls=[mock.call.notOkFunc()], any_order=True) 

你會看到實際的異常：

TypeError("'obj' parameter lacking default value") 

這是因爲你試圖實例類別withCtorArg的一個實例，其參數obj沒有發def lt值。如果你曾試圖居然直接實例化它，你會看到：

TypeError: __init__() takes exactly 2 arguments (1 given) 

不過，既然你讓mock庫處理模擬對象的實例化，錯誤發生在那裏 - 你得到了TypeError例外。

修改相關類：

class withCtorArg: 
    def __init__(self, obj = None): 
     pass 
    def notOkFunc(self): 
     pass 
    def okWithArgFunc(self, anArgForMe): 
     pass 

和OBJ增加一個默認值沒有解決的問題。

Answer 2

我不認爲我可以明確地解釋爲什麼會出現這種情況，我仍然懷疑Mock庫中存在一個錯誤，因爲這個問題只發生在被調用函數沒有參數的測試用例上。感謝advance512指出真正的錯誤是隱藏的！

然而，爲了解決這個問題，而不必修改生產代碼我會用下面的辦法：

# passes 
@mock.patch ('mymodule.noCtorArg') 
def test_noCtorArg_okFunc(self, noCtorArgMock): 
    mockSUT = noCtorArg.return_value 
    mockSUT.okFunc() 
    mockSUT.assert_has_calls([mock.call.okFunc()]) 

# passes 
@mock.patch ('mymodule.withCtorArg') 
def test_withCtorArg_okWithArgFuncTest(self, withCtorArgMock): 
    mockSUT = withCtorArg.return_value 
    mockSUT.okWithArgFunc("testing") 
    mockSUT.assert_has_calls([mock.call.okWithArgFunc("testing")]) 

# now passes 
@mock.patch ('mymodule.withCtorArg') 
def test_withCtorArg_doNotOkFuncTest(self, withCtorArgMock): 
    mockSUT = withCtorArg.return_value 
    mockSUT.notOkFunc() 
    mockSUT.assert_has_calls([mock.call.notOkFunc()], any_order=True) 

編輯：此

一個問題是，模擬沒有spec設置。這意味着SUT能夠調用原始類定義中不存在的方法。

的另一種方法是包裝類來嘲笑：

class withCtorArg: 
    def __init__(self,obj): 
     pass 
    def notOkFunc(self): 
     raise NotImplemented 
    def okWithArgFunc(self, anArgForMe): 
     raise NotImplemented 

class wrapped_withCtorArg(withCtorArg): 
    def __init__(self): 
     super(None) 

class BasicTestSuite(unittest.TestCase): 
    """Basic test Cases.""" 

    # now passes 
    def test_withCtorArg_doNotOkFuncTest(self): 
     mockSUT = mock.MagicMock(spec=wrapped_withCtorArg) 
     mockSUT.notOkFunc() 
     #mockSUT.doesntExist() #causes the test to fail. "Mock object has no attribute 'doesntExist'" 
     assert isinstance (mockSUT, withCtorArg) 
     mockSUT.assert_has_calls([mock.call.notOkFunc()], any_order=True)