tweaks

2024-02-16 17:47:07 -06:00 · 2024-02-16 17:47:07 -06:00 · 7a8589b1de
parent 626d70d2e9
commit 7a8589b1de
1 changed files with 1 additions and 0 deletions
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@ -460,6 +460,7 @@ async def test_finish_connection_times_out(
    ("exception_map"),
    [
        (OSError("Socket error"), HandshakeAPIError),
+        (APIConnectionError, APIConnectionError),
        (asyncio.TimeoutError, TimeoutAPIError),
        (asyncio.CancelledError, APIConnectionError),
    ],