pytestを使用して関数呼び出しを傍受し、パラメータを変更する方法は？

Question

download_page_html(url)は、有効なurlが指すページのHTMLコンテンツをダウンロードします。さまざまな機能を使用して彼らの仕事をdownload_page_html。

私はWikipediaからHTMLをダウンロードするのではなく、ローカルホストからHTMLをダウンロードしたいと考えています。

したがって、さまざまな機能によって行われた呼び出しをdownload_page_htmlに代行し、urlパラメータの値を置き換えることはできますか？

Answer 1

Pythonのstdlib unittest.mock.patchを使用します。デコレータとして使用することも、コンテキストマネージャ（つまり、with文を使用すること）として使用することもできます。

あなたはnewキーワードパラメータを渡す場合、関数はそれにpacthedされる - あなたはちょうどあなたが選ぶ パラメータで元の関数を呼び出すワンライナーを入れて：

from unittest.mock import patch 

from mymodule import download_page_html as original_download_page 

@patch('mymodule.download_page_html', new=lambda page: original_download_page("http://localhost/page.html")) 
def test_my_test_function(...): 
    ... 

Answer 2

あなたが追加する必要がありますあなたの「さまざまな機能」にしてから、関数本体内のオプションdph=None引数は、再割り当てすること必要に応じて、ハードコーディングされたdownload_page_html関数へ：

def my_various_function(url, arg2, kw1=val1, dph=None): 
    if dph is None: 
     dph = download_page_html 
    res = dph(url) 

すると、あなたのデザインワラントsuとは思わない場合

import pytest # no need to import monkeypatch from pytest 

def test_mytest(monkeypatch): 
    url = 'some_url' 
    def mockdlpage(url): 
     return 'the_mock_data' 
    monkeypatch.setattr(yourmodule, 'download_page_html', mockdlpage) 
    data = download_page_html(url)