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
|
"""Test network discovery."""
from socket import inet_aton
from unittest.mock import Mock, patch
import pytest
from zeroconf import ServiceStateChange, Zeroconf
from devolo_plc_api import Device, network
from devolo_plc_api.device_api import SERVICE_TYPE
from devolo_plc_api.zeroconf import ZeroconfServiceInfo
from . import TestData
from .mocks.zeroconf import MockServiceBrowser
class TestNetwork:
"""Test devolo_plc_api.network functions."""
@pytest.mark.asyncio
async def test_async_discover_network(self, test_data: TestData, mock_info_from_service: Mock):
"""Test discovering the network asynchronously."""
with (
patch("devolo_plc_api.network.ServiceBrowser", MockServiceBrowser),
patch("devolo_plc_api.network.Zeroconf.get_service_info", return_value=""),
):
serial_number = test_data.device_info[SERVICE_TYPE].properties["SN"]
mock_info_from_service.return_value = ZeroconfServiceInfo(
address=inet_aton(test_data.ip),
properties=test_data.device_info[SERVICE_TYPE].properties,
)
discovered = await network.async_discover_network(timeout=0.1)
assert serial_number in discovered
assert isinstance(discovered[serial_number], Device)
def test_discover_network(self, test_data: TestData, mock_info_from_service: Mock):
"""Test discovering the network synchronously."""
with (
patch("devolo_plc_api.network.ServiceBrowser", MockServiceBrowser),
patch("devolo_plc_api.network.Zeroconf.get_service_info", return_value=""),
):
serial_number = test_data.device_info[SERVICE_TYPE].properties["SN"]
mock_info_from_service.return_value = ZeroconfServiceInfo(
address=inet_aton(test_data.ip),
properties=test_data.device_info[SERVICE_TYPE].properties,
)
discovered = network.discover_network(timeout=0.1)
assert serial_number in discovered
assert isinstance(discovered[serial_number], Device)
def test_add_wrong_state(self):
"""Test early return on wrong state changes."""
with patch("devolo_plc_api.network.Zeroconf.get_service_info") as gsi:
network._add(Zeroconf(), SERVICE_TYPE, SERVICE_TYPE, ServiceStateChange.Removed, devices={}, timeout=3)
assert gsi.call_count == 0
def test_no_devices(self):
"""Test discovery with no devices."""
with (
patch("devolo_plc_api.network.ServiceBrowser", MockServiceBrowser),
patch("devolo_plc_api.network.Zeroconf.get_service_info", return_value=None),
):
discovered = network.discover_network(timeout=0.1)
assert not discovered
@pytest.mark.parametrize("mt", ["2600", "2601"])
def test_hcu(self, test_data: TestData, mt: str, mock_info_from_service: Mock):
"""Test ignoring Home Control Central Units."""
with (
patch("devolo_plc_api.network.ServiceBrowser", MockServiceBrowser),
patch("devolo_plc_api.network.Zeroconf.get_service_info", return_value=""),
):
mock_info_from_service.return_value = ZeroconfServiceInfo(address=test_data.ip.encode(), properties={"MT": mt})
discovered = network.discover_network(timeout=0.1)
assert not discovered
|
