如何使用模擬庫修補Python類

Question

我在爲測試修補類時遇到問題。我正在嘗試修補屬於Scrapy的東西 - HtmlXpathSelector課程。

下面是一些代碼：

from scrapy.selector import HtmlXPathSelector 
from mock import MagicMock, patch 

with patch('scrapy.selector.HtmlXPathSelector') as MockHtml: 
    instance = MockHtml.return_value 
    instance.method.return_value = 'foo' 
    example = HtmlXPathSelector() 
    print type(example) 
    assert example is instance 
    assert example.method == 'foo' 

結果是：

<class 'scrapy.selector.lxmlsel.HtmlXPathSelector'> 
Traceback (most recent call last): 
    File "<stdin>", line 6, in <module> 
AssertionError 
>>> 

這個例子是一樣的一個在素庫教程近了。任何想法爲什麼它不工作？

Answer 1

您不應該修補已在當前測試代碼中導入的類。相反，您需要在相應的模塊中修補該類（您要測試）。所以，如果HtmlXPathSelector在mymodule進口的，你會打補丁：

with patch('mymodule.HtmlXPathSelector') as MockClass: 
    ... 

進一步詳情，請參閱where to patch。

編輯如果你真的需要這一點，你可以在當前模塊中修補類：

with patch('__main__.Class') as MockClass: 

Answer 2

有兩個問題與您的代碼示例。首先是您從scrapy模塊導入了HtmlXPathSelector，然後在事實之後更改該名稱。進口替代選擇，並且使用的名稱從那裏：

from scrapy import selector 
from mock import MagicMock, patch 

with patch('scrapy.selector.HtmlXPathSelector') as MockHtml: 
    instance = MockHtml.return_value 
    instance.method.return_value = 'foo' 
    example = selector.HtmlXPathSelector() 
    print type(example) 
    assert example is instance 
    assert example.method() == 'foo' 

的另一個問題是，你的最後一行是檢查method，當它需要調用的方法爲method()。