Pytest raise测试失败,对于requests.raise_for_status()

Question

我最近开始使用pytest,甚至最近开始使用mock来模拟请求库.我已经发出了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

Answer 1

您指示 pytest 预期应该在APIClient.get_records该方法定义中引发异常，但在该方法定义中您已经捕获了该异常并只是进行了打印。

异常实际上正在发生，并且通过在控制台输出中查看打印结果证明了这一点。

相反，您应该检查该方法raise_for_status被调用的模拟。