我最近开始使用pytest,最近开始使用mock来模拟requests库。我已经发出了request.Response对象没关系,对于200状态代码,它工作正常。我在这里尝试做的是使用raise_for_status()来检查超出速率限制的错误,并测试它是否用pytest处理异常。
我正在使用Mock side_effect选项,它似乎触发了我希望的异常,但是pytest似乎并没有意识到这已经发生并且未通过测试。
有什么想法?我确定这是我很想念的东西!
我上课的代码是:
class APIClient:
def get_records(self, url):
try:
r = requests.get(url)
r.raise_for_status()
return r.json()
except requests.HTTPError as e:
print("Handling the exception")
在测试课上,我得到了:
@pytest.fixture
def http_error_response(rate_limit_json):
mock_response = mock.Mock()
mock_response.json.return_value = rate_limit_json
mock_response.status_code = 429
mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError
return mock_response
class TestRecovery(object):
@mock.patch('requests.get')
def test_throws_exception_for_rate_limit_error\
(self, mock_get, api_query_object, http_error_response):
mock_get.return_value = http_error_response
print(http_error_response.raise_for_status.side_effect)
url = api_query_object.get_next_url()
with pytest.raises(requests.exceptions.HTTPError):
api_query_object.get_records(url)
我得到的输出是:
with pytest.raises(requests.exceptions.HTTPError):
> api_query_object.get_records(url)
E Failed: DID NOT RAISE
---------------------- Captured stdout call ----------------------
<class 'requests.exceptions.HTTPError'>
Handling the exception
答案 0 :(得分:0)
您正在指示pytest期待应该在APIClient.get_records
中引发的异常,但是在该方法定义中,您已经在捕获该异常,并且只是执行打印。
该异常实际上正在发生,并且可以通过在控制台输出中看到打印结果来证明。
相反,您应该通过模拟检查方法raise_for_status
被调用。