Ejemplo n.º 1
0
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
Ejemplo n.º 2
0
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'