從模擬補丁/ Python中

Question

我試圖讓mock.patch對下面的一段示例代碼import語句：

from mock import patch 
from collections import defaultdict 

with patch('collections.defaultdict'): 
    d = defaultdict() 
    print 'd:', d 

此輸出以下：

d: defaultdict(None, {}) 

這意味着該defaultdict未修補。

如果我採用了直板import語句它的工作原理更換從/ import語句：

from mock import patch 
import collections 

with patch('collections.defaultdict'): 
d = collections.defaultdict() 
print 'd:', d 

輸出是：

d: <MagicMock name='defaultdict()' id='139953944084176'> 

有什麼辦法來修補從/導入使用電話嗎？

謝謝

Answer 1

如果你在同一模塊中修補的東西，你可以用__main__：

from mock import patch 
from collections import defaultdict 

with patch('__main__.defaultdict'): 
    d = defaultdict() 
    print 'd:', d 

如果你嘲笑的東西導入模塊，不過，你會想使用該模塊的名稱，以便正確引用（或名稱）進行了修補：

# foo.py 

from collections import defaultdict 

def bar(): 
    return defaultdict() 


# foo_test.py  

from mock import patch 
from foo import bar 

with patch('foo.defaultdict'): 
    print bar() 

的這裏的一點是，補丁要的完整路徑是修補的東西。這在修補當前模塊中的某些內容時看起來有點奇怪，因爲人們通常不會使用__main__（或者必須參考當前模塊）。

Answer 2

patch作品通過修補名稱。如果使用名稱defaultdict（在本地命名空間中）訪問對象，則無法通過修補名稱collections.defaultdict來實現任何功能。請參閱文檔http://www.voidspace.org.uk/python/mock/patch.html#id1。