mirror of https://github.com/pypa/pip
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
|
import pytest
|
||
|
|
||
|
from pip._internal.exceptions import NetworkConnectionError
|
||
|
from pip._internal.network.utils import raise_for_status
|
||
|
from tests.lib.requests_mocks import MockResponse
|
||
|
|
||
|
|
||
|
@pytest.mark.parametrize(("status_code", "error_type"), [
|
||
|
(401, "Client Error"),
|
||
|
(501, "Server Error"),
|
||
|
])
|
||
|
def test_raise_for_status_raises_exception(status_code, error_type):
|
||
|
contents = b'downloaded'
|
||
|
resp = MockResponse(contents)
|
||
|
resp.status_code = status_code
|
||
|
resp.url = "http://www.example.com/whatever.tgz"
|
||
|
resp.reason = "Network Error"
|
||
|
with pytest.raises(NetworkConnectionError) as exc:
|
||
|
raise_for_status(resp)
|
||
|
assert str(exc.info) == (
|
||
|
"{} {}: Network Error for url:"
|
||
|
" http://www.example.com/whatever.tgz".format(
|
||
|
status_code, error_type)
|
||
|
)
|
||
|
|
||
|
|
||
|
def test_raise_for_status_does_not_raises_exception():
|
||
|
contents = b'downloaded'
|
||
|
resp = MockResponse(contents)
|
||
|
resp.status_code = 201
|
||
|
resp.url = "http://www.example.com/whatever.tgz"
|
||
|
resp.reason = "No error"
|
||
|
return_value = raise_for_status(resp)
|
||
|
assert return_value is None
|