1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
"""Test generic client characteristics."""
import asyncio
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from aioguardian import Client
from aioguardian.errors import SocketError
from tests.common import load_fixture
@pytest.mark.asyncio
@pytest.mark.parametrize("recv_response", [AsyncMock(side_effect=asyncio.TimeoutError)])
async def test_command_timeout(mock_datagram_client: MagicMock) -> None:
"""Test that a timeout during command execution throws an exception.
Args:
mock_datagram_client: A mocked UDP client.
"""
with mock_datagram_client, patch("asyncio.sleep"):
with pytest.raises(SocketError) as err:
async with Client("192.168.1.100") as client:
await client.system.ping()
assert str(err.value) == "SYSTEM_PING command timed out"
@pytest.mark.asyncio
async def test_command_timeout_successful_retry(
mock_datagram_client: MagicMock,
) -> None:
"""Test that a timeout with a successful retry works.
Args:
mock_datagram_client: A mocked UDP client.
"""
with mock_datagram_client, patch("asyncio.sleep"):
mock_datagram_client.recv.side_effect = [
asyncio.TimeoutError,
(load_fixture("ping_success_response.json").encode(), "192.168.1.100"),
]
async with Client("192.168.1.100") as client:
ping_response = await client.system.ping()
assert ping_response["command"] == 0
assert ping_response["status"] == "ok"
assert ping_response["data"]["uid"] == "ABCDEF123456"
@pytest.mark.asyncio
async def test_command_without_socket_connect() -> None:
"""Test that executing a command without an open connection throws an exception."""
client = Client("192.168.1.100")
with pytest.raises(SocketError) as err:
await client.system.ping()
assert str(err.value) == "You aren't connected to the device yet"
@pytest.mark.asyncio
async def test_connect_timeout() -> None:
"""Test that a timeout during connection throws an exception."""
with patch("asyncio_dgram.connect", AsyncMock(side_effect=asyncio.TimeoutError)):
with pytest.raises(SocketError) as err:
async with Client("192.168.1.100") as client:
await client.system.ping()
assert str(err.value) == "Connection to device timed out"
@pytest.mark.asyncio
@pytest.mark.parametrize(
"command_response", [load_fixture("ping_success_response.json").encode()]
)
async def test_raw_command_success(mock_datagram_client: MagicMock) -> None:
"""Test a successful raw command.
Args:
mock_datagram_client: A mocked UDP client.
"""
with mock_datagram_client:
async with Client("192.168.1.100") as client:
ping_response = await client.execute_raw_command(0)
assert ping_response["command"] == 0
assert ping_response["status"] == "ok"
assert ping_response["data"]["uid"] == "ABCDEF123456"
|
