Question

我正在尝试测试依赖于第三方的代码，并希望使用Monkeypatch复制我希望请求返回的内容。这是我所拥有的代码的最小示例。

import requests

def get_urls(*urls):
    results = []
    for url in urls:
        results.append(requests.get(url).text)

对于我的测试，我有如下内容：

from my_package import get_urls

def test_get_urls():
    urls = ("https://example.com/a", "https://example.com/b", "https://example.com/c")
    assert len(get_urls(urls)) == 3

如何使用requests.get猴子对monkeypatch.setattr的每个呼叫进行补丁处理？模拟程序包似乎可以do this using side effects。我该如何使用pytest？

Answer 1

使用monkeypatch覆盖方法调用时，可以将该属性设置为自定义函数。这是一种基于url实现不同行为的方法：

URL_MAP = {
    'https://example.com/a': json.dumps({1: 2}),
    'https://example.com/b': json.dumps({3: 4})
}

def fake_req_get(url, *args, **kwargs):
    return url_map.get(url, '{}')

def test_get_urls(monkeypatch):
    monkeypatch.setattr('requests.get', fake_req_get)
    urls = ("https://example.com/a", "https://example.com/b", "https://example.com/c")
    assert get_urls(urls)[0] == URL_MAP["https://example.com/a"]

多次更改Monkeypatch setattr

1 个答案: