Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-22 19:23:34 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Mock asyncio.sleep for klapprotocol tests (#1130)

Browse Source 
Speeds up tests in `test_klapprotocol.py` from 26s to 2s when there's no
sleep between the retries.
This commit is contained in:
Teemu R. 2024-09-27 10:34:30 +02:00 committed by GitHub
parent 69c270055d
commit 8321fd08aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
kasa/tests/test_klapprotocol.py
View File

@ -67,6 +67,7 @@ async def test_protocol_retries_via_client_session(
host = "127.0.0.1" host = "127.0.0.1"
conn = mocker.patch.object(aiohttp.ClientSession, "post", side_effect=error) conn = mocker.patch.object(aiohttp.ClientSession, "post", side_effect=error)
mocker.patch.object(protocol_class, "BACKOFF_SECONDS_AFTER_TIMEOUT", 0) mocker.patch.object(protocol_class, "BACKOFF_SECONDS_AFTER_TIMEOUT", 0)
mocker.patch("asyncio.sleep")
config = DeviceConfig(host) config = DeviceConfig(host)
with pytest.raises(KasaException): with pytest.raises(KasaException):
@ -139,6 +140,8 @@ async def test_protocol_retry_recoverable_error(
"post", "post",
side_effect=aiohttp.ClientOSError("foo"), side_effect=aiohttp.ClientOSError("foo"),
) )
mocker.patch("asyncio.sleep")
config = DeviceConfig(host) config = DeviceConfig(host)
with pytest.raises(KasaException): with pytest.raises(KasaException):
await protocol_class(transport=transport_class(config=config)).query( await protocol_class(transport=transport_class(config=config)).query(