diff --git a/kasa/klaptransport.py b/kasa/klaptransport.py index ab33ca18e..8feae98c1 100644 --- a/kasa/klaptransport.py +++ b/kasa/klaptransport.py @@ -57,7 +57,7 @@ from .credentials import Credentials from .deviceconfig import DeviceConfig -from .exceptions import AuthenticationError, KasaException +from .exceptions import AuthenticationError, KasaException, _RetryableError from .httpclient import HttpClient from .json import loads as json_loads from .protocol import DEFAULT_CREDENTIALS, BaseTransport, get_default_credentials, md5 @@ -337,7 +337,7 @@ async def send(self, request: str): # If we failed with a security error, force a new handshake next time. if response_status == 403: self._handshake_done = False - raise AuthenticationError( + raise _RetryableError( f"Got a security error from {self._host} after handshake " + "completed" ) diff --git a/kasa/tests/test_klapprotocol.py b/kasa/tests/test_klapprotocol.py index 7c8054758..b71ea460d 100644 --- a/kasa/tests/test_klapprotocol.py +++ b/kasa/tests/test_klapprotocol.py @@ -417,7 +417,7 @@ async def _return_response(url: URL, params=None, data=None, *_, **__): pytest.param( (200, 200, 403), True, - pytest.raises(AuthenticationError), + pytest.raises(_RetryableError), id="request-403-status", ), pytest.param(