Changing monkeypatch setattr multiple times

Question

I am trying to test code that depends on a third party and would like to use monkeypatch to replicate what I expect a request will return. Here is a minimal example of the code that I have.

import requests

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

For my tests, I have something like the following:

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

How can I monkeypatch each of the calls to requests.get using monkeypatch.setattr? The mock package seems to be able to do this using side effects. How do I do this with pytest?

score 5 · Accepted Answer · edited May 07 '21 at 21:34

5

When you override a method call using monkeypatch, you can set that attribute to be a custom function. Here is one method of implementing different behaviors based on 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"]

edited May 07 '21 at 21:34

baumannalexj

466
7
18

answered Feb 28 '20 at 16:23

jordanm

29,762
6
59
72

This is exactly what I was looking for. Thank you! – Jack Moody Feb 28 '20 at 16:32
This helped me as well, very neat, thanks! – nojohnny101 Feb 03 '21 at 18:03
1

This solution is good, but I am wondering if it is allowed to use monkeypatch.setattr multiple times in a test – Michael S May 12 '21 at 15:04

Changing monkeypatch setattr multiple times

1 Answers1