如果方法在另一個方法中被調用兩次，如何assert_called_with第一個調用？

Question

例如，在t.py

def a(obj): 
    print obj 

def b(): 
    a(1) 
    a(2) 

則：

from t import b 

with patch('t.a') as m: 
    b() 
    m.assert_called_with(1) 

我得到：

AssertionError: Expected call: a(1) 
Actual call: a(2) 

Answer 1

最簡單的方法是從mock.call_args_list得到第一個項目，並檢查它是否叫做1：

call_args_list

這是爲了在序列 模擬對象（因此列表的長度是倍它已被調用的數目）進行的所有呼叫的列表。

assert m.call_args_list[0] == call(1) 

其中call從mock進口：from mock import call。

另外，mock_calls也可代替call_args_list。

另一種選擇是使用assert_any_call()：

m.assert_any_call(1)