Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2025-08-09 20:24:02 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Raise kasa TimeoutError instead of _RetryableError on asyncio.TimeoutError

Browse Source
This commit is contained in:
Steven B
2025-01-06 12:39:46 +00:00
parent 9fd2f28420
commit 546ae633f9
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/protocols/test_iotprotocol.py
View File

@@ -16,7 +16,7 @@ import pytest
from kasa.credentials import Credentials
from kasa.device import Device
from kasa.deviceconfig import DeviceConfig
from kasa.exceptions import KasaException, _RetryableError
from kasa.exceptions import KasaException, TimeoutError
from kasa.iot import IotDevice
from kasa.protocols.iotprotocol import IotProtocol, _deprecated_TPLinkSmartHomeProtocol
from kasa.protocols.protocol import (
@@ -444,7 +444,7 @@ async def test_protocol_handles_timeout_failure_during_write(
protocol = protocol_class(transport=transport_class(config=config))
mocker.patch("asyncio.open_connection", side_effect=aio_mock_writer)
with pytest.raises(
_RetryableError,
TimeoutError,
match="Timeout after 5 seconds sending request to the device 127.0.0.1:9999: Simulated timeout",
):
await protocol.query({})
@@ -490,7 +490,7 @@ async def test_protocol_handles_timeout_failure_during_connection(
mocker.patch("asyncio.open_connection", side_effect=aio_mock_writer)
with pytest.raises(
_RetryableError,
TimeoutError,
match="Timeout after 5 seconds connecting to the device: 127.0.0.1:9999: Simulated timeout",
):
await protocol.query({})