This commit is contained in:
J. Nick Koston 2024-02-16 17:48:14 -06:00
parent 7a8589b1de
commit b76116f5bf
No known key found for this signature in database

View File

@ -30,6 +30,7 @@ from aioesphomeapi.core import (
InvalidAuthAPIError, InvalidAuthAPIError,
RequiresEncryptionAPIError, RequiresEncryptionAPIError,
ResolveAPIError, ResolveAPIError,
SocketClosedAPIError,
TimeoutAPIError, TimeoutAPIError,
) )
@ -461,6 +462,7 @@ async def test_finish_connection_times_out(
[ [
(OSError("Socket error"), HandshakeAPIError), (OSError("Socket error"), HandshakeAPIError),
(APIConnectionError, APIConnectionError), (APIConnectionError, APIConnectionError),
(SocketClosedAPIError, SocketClosedAPIError),
(asyncio.TimeoutError, TimeoutAPIError), (asyncio.TimeoutError, TimeoutAPIError),
(asyncio.CancelledError, APIConnectionError), (asyncio.CancelledError, APIConnectionError),
], ],