def test_connection_raises_for_request_retry_failure(): responses.add(responses.GET, 'http://127.0.0.1:4003/spongebob', body=requests.exceptions.RequestException()) connection = Connection('http://127.0.0.1:4003') with pytest.raises(ArkHTTPException) as exception: connection.get('spongebob') assert len(responses.calls) == 3
def test_connection_request_retry_successful(): responses.add(responses.GET, 'http://127.0.0.1:4003/spongebob', body=requests.exceptions.RequestException()) responses.add(responses.GET, 'http://127.0.0.1:4003/spongebob', body=requests.exceptions.RequestException()) responses.add(responses.GET, 'http://127.0.0.1:4003/spongebob', json={'success': True}, status=200) connection = Connection('http://127.0.0.1:4003') data = connection.get('spongebob') assert data == {'success': True} assert len(responses.calls) == 3 assert responses.calls[0].request.url == 'http://127.0.0.1:4003/spongebob'