mirror of https://github.com/pypa/pip
36 lines
1.1 KiB
Python
36 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: int, error_type: str) -> None:
|
|
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 excinfo:
|
|
raise_for_status(resp)
|
|
assert str(excinfo.value) == (
|
|
f"{status_code} {error_type}: Network Error for url:"
|
|
" http://www.example.com/whatever.tgz"
|
|
)
|
|
|
|
|
|
def test_raise_for_status_does_not_raises_exception() -> None:
|
|
contents = b"downloaded"
|
|
resp = MockResponse(contents)
|
|
resp.status_code = 201
|
|
resp.url = "http://www.example.com/whatever.tgz"
|
|
resp.reason = "No error"
|
|
raise_for_status(resp)
|