모의 라이브러리를 사용하여 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()로 메소드를 호출해야 할 때.