03cd88c336
* Map httpcore exceptions for `Response.iter_*` methods * Tweak * Change wording Co-authored-by: Florimond Manca <florimond.manca@gmail.com> Co-authored-by: Florimond Manca <florimond.manca@gmail.com> Co-authored-by: Tom Christie <tom@tomchristie.com>
69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
from typing import Any
|
|
|
|
import httpcore
|
|
import pytest
|
|
|
|
import httpx
|
|
from httpx._exceptions import HTTPCORE_EXC_MAP
|
|
|
|
|
|
def test_httpcore_all_exceptions_mapped() -> None:
|
|
"""
|
|
All exception classes exposed by HTTPCore are properly mapped to an HTTPX-specific
|
|
exception class.
|
|
"""
|
|
not_mapped = [
|
|
value
|
|
for name, value in vars(httpcore).items()
|
|
if isinstance(value, type)
|
|
and issubclass(value, Exception)
|
|
and value not in HTTPCORE_EXC_MAP
|
|
]
|
|
|
|
if not_mapped: # pragma: nocover
|
|
pytest.fail(f"Unmapped httpcore exceptions: {not_mapped}")
|
|
|
|
|
|
def test_httpcore_exception_mapping(server) -> None:
|
|
"""
|
|
HTTPCore exception mapping works as expected.
|
|
"""
|
|
|
|
# Make sure we don't just map to `NetworkError`.
|
|
with pytest.raises(httpx.ConnectError):
|
|
httpx.get("http://doesnotexist")
|
|
|
|
# Make sure streaming methods also map exceptions.
|
|
url = server.url.copy_with(path="/slow_stream_response")
|
|
timeout = httpx.Timeout(None, read=0.1)
|
|
with httpx.stream("GET", url, timeout=timeout) as stream:
|
|
with pytest.raises(httpx.ReadTimeout):
|
|
stream.read()
|
|
|
|
# Make sure it also works with custom transports.
|
|
class MockTransport(httpcore.SyncHTTPTransport):
|
|
def request(self, *args: Any, **kwargs: Any) -> Any:
|
|
raise httpcore.ProtocolError()
|
|
|
|
client = httpx.Client(transport=MockTransport())
|
|
with pytest.raises(httpx.ProtocolError):
|
|
client.get("http://testserver")
|
|
|
|
|
|
def test_httpx_exceptions_exposed() -> None:
|
|
"""
|
|
All exception classes defined in `httpx._exceptions`
|
|
are exposed as public API.
|
|
"""
|
|
|
|
not_exposed = [
|
|
value
|
|
for name, value in vars(httpx._exceptions).items()
|
|
if isinstance(value, type)
|
|
and issubclass(value, Exception)
|
|
and not hasattr(httpx, name)
|
|
]
|
|
|
|
if not_exposed: # pragma: nocover
|
|
pytest.fail(f"Unexposed HTTPX exceptions: {not_exposed}")
|