From b76116f5bf9a0b0378899ab6b049d513fb5d2f3b Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Fri, 16 Feb 2024 17:48:14 -0600 Subject: [PATCH] tweaks --- tests/test_connection.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_connection.py b/tests/test_connection.py index 1fd776c..3b004c5 100644 --- a/tests/test_connection.py +++ b/tests/test_connection.py @@ -30,6 +30,7 @@ from aioesphomeapi.core import ( InvalidAuthAPIError, RequiresEncryptionAPIError, ResolveAPIError, + SocketClosedAPIError, TimeoutAPIError, ) @@ -461,6 +462,7 @@ async def test_finish_connection_times_out( [ (OSError("Socket error"), HandshakeAPIError), (APIConnectionError, APIConnectionError), + (SocketClosedAPIError, SocketClosedAPIError), (asyncio.TimeoutError, TimeoutAPIError), (asyncio.CancelledError, APIConnectionError), ],