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

时间:2016-02-20 12:41:24

标签: python unit-testing pytest python-mock

我最近开始使用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

1 个答案:

答案 0 :(得分:0)

您正在指示pytest期待应该在APIClient.get_records中引发的异常,但是在该方法定义中,您已经在捕获该异常,并且只是执行打印。

该异常实际上正在发生,并且可以通过在控制台输出中看到打印结果来证明。

相反,您应该通过模拟检查方法raise_for_status被调用。