python-botocore/tests/unit/test_http_client_exception_mapping.py

import pytest

from botocore import exceptions as botocore_exceptions
from botocore.vendored.requests import exceptions as requests_exceptions
from botocore.vendored.requests.packages.urllib3 import exceptions as urllib3_exceptions


@pytest.mark.parametrize(
    "new_exception, old_exception",
    (
        (botocore_exceptions.ReadTimeoutError, requests_exceptions.ReadTimeout),
        (botocore_exceptions.ReadTimeoutError, urllib3_exceptions.ReadTimeoutError),
        (botocore_exceptions.ConnectTimeoutError, requests_exceptions.ConnectTimeout),
        (botocore_exceptions.ProxyConnectionError, requests_exceptions.ProxyError),
        (botocore_exceptions.SSLError, requests_exceptions.SSLError),
    ),
)
def test_http_client_exception_mapping(new_exception, old_exception):
    # assert that the new exception can still be caught by the old vendored one
    with pytest.raises(old_exception):
        raise new_exception(endpoint_url=None, proxy_url=None, error=None)