魔術模擬 assert_Called_once 與 assert_Called_once_with 奇怪的行為

Question

我注意到在 python 中使用assert_called_once和assert_called_once_with有一個奇怪的行為。 這是我真正的簡單測試：

文件模塊/a.py

from .b import B

class A(object):
    def __init__(self):
        self.b = B("hi")

    def call_b_hello(self):
        print(self.b.hello())

文件模塊/b.py

class B(object):
    def __init__(self, string):
        print("created B")
        self.string = string;

    def hello(self):
        return self.string

這些是我的測試：

import unittest
from mock import patch
from module.a import A    

class MCVETests(unittest.TestCase):
    @patch('module.a.B')   
    def testAcallBwithMockPassCorrect(self, b1):
        a = A()
        b1.assert_called_once_with("hi")
        a.call_b_hello()
        a.b.hello.assert_called_once()

    @patch('module.a.B')
    def testAcallBwithMockPassCorrectWith(self, b1):
        a = A()
        b1.assert_called_once_with("hi")
        a.call_b_hello()
        a.b.hello.assert_called_once_with()

    @patch('module.a.B')
    def testAcallBwithMockFailCorrectWith(self, b1):
        a = A()
        b1.assert_called_once_with("hi")
        a.b.hello.assert_called_once_with()

    @patch('module.a.B')
    def testAcallBwithMockPassWrong(self, b1):
        a = A()
        b1.assert_called_once_with("hi")
        a.b.hello.assert_called_once()

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

我在函數名稱中提到的問題是：

• 測試 1 正確通過
• 測試 2 正確通過
• 測試 3 正確失敗（我已刪除對 b 的調用）
• 測試 4 通過我不知道為什么。

難道我做錯了什么？ 我不確定，但閱讀文檔docs python ：

assert_Called_once(*args, **kwargs)

斷言模擬只被調用了一次。

Answer 1

這是舊的，但對於其他登陸這里的人......

對於 python < 3.6， assert_called_once不是一回事，所以你實際上是在做一個不會出錯的assert_called_once函數調用

請參閱： http : //engineroom.trackmaven.com/blog/mocking-mistakes/

您可以改為檢查呼叫計數。