File: test_list.py

package info (click to toggle)
python-aioguardian 2023.12.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 772 kB
  • sloc: python: 1,387; sh: 45; makefile: 13
file content (89 lines) | stat: -rw-r--r-- 2,744 bytes parent folder | download
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
89
"""Test the list command."""
from unittest.mock import MagicMock

import pytest

from aioguardian import Client
from aioguardian.errors import CommandError
from tests.common import load_fixture


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "command_response", [load_fixture("wifi_list_failure_response.json").encode()]
)
async def test_list_failure(mock_datagram_client: MagicMock) -> None:
    """Test the wifi_list command failing.

    Args:
        mock_datagram_client: A mocked UDP client.
    """
    with mock_datagram_client:
        with pytest.raises(CommandError) as err:
            async with Client("192.168.1.100") as client:
                _ = await client.wifi.list()

        assert str(err.value) == (
            "WIFI_LIST command failed (response: {'command': 38, 'status': 'error'})"
        )


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "command_response", [load_fixture("wifi_list_success_response.json").encode()]
)
async def test_list_success(mock_datagram_client: MagicMock) -> None:
    """Test the wifi_list command succeeding.

    Args:
        mock_datagram_client: A mocked UDP client.
    """
    with mock_datagram_client:
        async with Client("192.168.1.100") as client:
            await client.wifi.scan()
            wifi_list_response = await client.wifi.list()

        assert wifi_list_response["command"] == 38
        assert wifi_list_response["status"] == "ok"
        assert wifi_list_response["data"] == {
            "record_count": 1,
            "records": [
                {
                    "bssid": "60:31:97:BE:53:5D",
                    "rssi": -89,
                    "channel": 1,
                    "authmode": 4,
                    "ssid": "CenturyLink0526",
                }
            ],
        }


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "command_response", [load_fixture("wifi_list_success_response.json").encode()]
)
async def test_list_success_cached(mock_datagram_client: MagicMock) -> None:
    """Test the wifi_list command succeeding after a cache.

    Args:
        mock_datagram_client: A mocked UDP client.
    """
    with mock_datagram_client:
        async with Client("192.168.1.100") as client:
            wifi_list_response = await client.wifi.list()

        assert wifi_list_response["command"] == 38
        assert wifi_list_response["status"] == "ok"
        assert wifi_list_response["data"] == {
            "record_count": 1,
            "records": [
                {
                    "bssid": "60:31:97:BE:53:5D",
                    "rssi": -89,
                    "channel": 1,
                    "authmode": 4,
                    "ssid": "CenturyLink0526",
                }
            ],
        }