devolo_plc_api/tests/test_network.py at 5db67810c9930398ef5e7b947ea5c1e290b2cf1a · 2Fake/devolo_plc_api · GitHub

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