通过re.compile()创建一个正则表达式模式,您可以将其输入到请求模拟程序中,如文档所示:
import re
import pytest
import requests
import requests_mock
def test_owm_client_raises_error_for_timeout_error():
with requests_mock.Mocker() as m:
# This is the regex pattern. Currently, it will accept any text after the base URL below. Update this if your requirements are stricter.
matcher = re.compile('http://api.openweathermap.org/geo/1.0/direct/.*')
# For simplicity, let's say we will just raise ValueError for the target URL
m.get(
matcher,
exc=ValueError("Mocked error!"),
)
# Test if the exception ValueError will be raised if we accessed the target URLs
with pytest.raises(ValueError):
requests.get("http://api.openweathermap.org/geo/1.0/direct/")
with pytest.raises(ValueError):
requests.get("http://api.openweathermap.org/geo/1.0/direct/abcde")
with pytest.raises(ValueError):
requests.get("http://api.openweathermap.org/geo/1.0/direct/12345")
with pytest.raises(ValueError):
requests.get("http://api.openweathermap.org/geo/1.0/direct/abcde/12345")
with pytest.raises(ValueError):
requests.get("http://api.openweathermap.org/geo/1.0/direct/abcde/12345?any=thing&in=here")
# Test if the mocked request wouldn't be used if the URL is different. Thus, the exception should show that the URL wasn't mocked.
with pytest.raises(requests_mock.exceptions.NoMockAddress):
requests.get("http://api.openweathermap.org/geo/1.0/direct2/")
with pytest.raises(requests_mock.exceptions.NoMockAddress):
requests.get("http://api.openweathermap.org/geo/1.0/direct2/abcde")
输出:
$ pytest -q
.
1 passed in 0.07s