928 lines
28 KiB
Python
928 lines
28 KiB
Python
"""Test the Network Configuration."""
|
|
|
|
from ipaddress import IPv4Address
|
|
from typing import Any
|
|
from unittest.mock import MagicMock, Mock, patch
|
|
|
|
import ifaddr
|
|
import pytest
|
|
|
|
from homeassistant.components import network
|
|
from homeassistant.components.network.const import (
|
|
ATTR_ADAPTERS,
|
|
ATTR_CONFIGURED_ADAPTERS,
|
|
DOMAIN,
|
|
MDNS_TARGET_IP,
|
|
STORAGE_KEY,
|
|
STORAGE_VERSION,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.typing import WebSocketGenerator
|
|
|
|
_NO_LOOPBACK_IPADDR = "192.168.1.5"
|
|
_LOOPBACK_IPADDR = "127.0.0.1"
|
|
|
|
|
|
def _mock_socket(sockname):
|
|
mock_socket = MagicMock()
|
|
mock_socket.getsockname = Mock(return_value=sockname)
|
|
return mock_socket
|
|
|
|
|
|
def _mock_cond_socket(sockname):
|
|
class CondMockSock(MagicMock):
|
|
def connect(self, addr):
|
|
"""Mock connect that stores addr."""
|
|
# pylint: disable-next=attribute-defined-outside-init
|
|
self._addr = addr[0]
|
|
|
|
def getsockname(self):
|
|
"""Return addr if it matches the mock sockname."""
|
|
if self._addr == sockname:
|
|
return [sockname]
|
|
raise AttributeError
|
|
|
|
return CondMockSock()
|
|
|
|
|
|
def _mock_socket_exception(exc):
|
|
mock_socket = MagicMock()
|
|
mock_socket.getsockname = Mock(side_effect=exc)
|
|
return mock_socket
|
|
|
|
|
|
def _generate_mock_adapters():
|
|
mock_lo0 = Mock(spec=ifaddr.Adapter)
|
|
mock_lo0.nice_name = "lo0"
|
|
mock_lo0.ips = [ifaddr.IP("127.0.0.1", 8, "lo0")]
|
|
mock_lo0.index = 0
|
|
mock_eth0 = Mock(spec=ifaddr.Adapter)
|
|
mock_eth0.nice_name = "eth0"
|
|
mock_eth0.ips = [ifaddr.IP(("2001:db8::", 1, 1), 8, "eth0")]
|
|
mock_eth0.index = 1
|
|
mock_eth1 = Mock(spec=ifaddr.Adapter)
|
|
mock_eth1.nice_name = "eth1"
|
|
mock_eth1.ips = [ifaddr.IP("192.168.1.5", 23, "eth1")]
|
|
mock_eth1.index = 2
|
|
mock_vtun0 = Mock(spec=ifaddr.Adapter)
|
|
mock_vtun0.nice_name = "vtun0"
|
|
mock_vtun0.ips = [ifaddr.IP("169.254.3.2", 16, "vtun0")]
|
|
mock_vtun0.index = 3
|
|
return [mock_eth0, mock_lo0, mock_eth1, mock_vtun0]
|
|
|
|
|
|
async def test_async_detect_interfaces_setting_non_loopback_route(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test without default interface config and the route returns a non-loopback address."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([_NO_LOOPBACK_IPADDR]),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
network_obj = hass.data[DOMAIN]
|
|
assert network_obj.configured_adapters == []
|
|
|
|
assert network_obj.adapters == [
|
|
{
|
|
"auto": False,
|
|
"index": 1,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 8,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
}
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"index": 0,
|
|
"auto": False,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "127.0.0.1", "network_prefix": 8}],
|
|
"ipv6": [],
|
|
"name": "lo0",
|
|
},
|
|
{
|
|
"index": 2,
|
|
"auto": True,
|
|
"default": True,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"index": 3,
|
|
"auto": False,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_async_detect_interfaces_setting_loopback_route(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test without default interface config and the route returns a loopback address."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([_LOOPBACK_IPADDR]),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
network_obj = hass.data[DOMAIN]
|
|
assert network_obj.configured_adapters == []
|
|
assert network_obj.adapters == [
|
|
{
|
|
"index": 1,
|
|
"auto": True,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 8,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
}
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"index": 0,
|
|
"auto": False,
|
|
"default": True,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "127.0.0.1", "network_prefix": 8}],
|
|
"ipv6": [],
|
|
"name": "lo0",
|
|
},
|
|
{
|
|
"index": 2,
|
|
"auto": True,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"index": 3,
|
|
"auto": False,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_async_detect_interfaces_setting_empty_route(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test without default interface config and the route returns nothing."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([]),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
network_obj = hass.data[DOMAIN]
|
|
assert network_obj.configured_adapters == []
|
|
assert network_obj.adapters == [
|
|
{
|
|
"auto": True,
|
|
"index": 1,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 8,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
}
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 0,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "127.0.0.1", "network_prefix": 8}],
|
|
"ipv6": [],
|
|
"name": "lo0",
|
|
},
|
|
{
|
|
"auto": True,
|
|
"index": 2,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 3,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_async_detect_interfaces_setting_exception(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test without default interface config and the route throws an exception."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket_exception(AttributeError),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
network_obj = hass.data[DOMAIN]
|
|
assert network_obj.configured_adapters == []
|
|
assert network_obj.adapters == [
|
|
{
|
|
"auto": True,
|
|
"index": 1,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 8,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
}
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 0,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "127.0.0.1", "network_prefix": 8}],
|
|
"ipv6": [],
|
|
"name": "lo0",
|
|
},
|
|
{
|
|
"auto": True,
|
|
"index": 2,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 3,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_interfaces_configured_from_storage(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test settings from storage are preferred over auto configure."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth0", "eth1", "vtun0"]},
|
|
}
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([_NO_LOOPBACK_IPADDR]),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
network_obj = hass.data[DOMAIN]
|
|
assert network_obj.configured_adapters == ["eth0", "eth1", "vtun0"]
|
|
|
|
assert network_obj.adapters == [
|
|
{
|
|
"auto": False,
|
|
"index": 1,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 8,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
}
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 0,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "127.0.0.1", "network_prefix": 8}],
|
|
"ipv6": [],
|
|
"name": "lo0",
|
|
},
|
|
{
|
|
"auto": True,
|
|
"index": 2,
|
|
"default": True,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 3,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_interfaces_configured_from_storage_websocket_update(
|
|
hass: HomeAssistant,
|
|
hass_ws_client: WebSocketGenerator,
|
|
hass_storage: dict[str, Any],
|
|
) -> None:
|
|
"""Test settings from storage can be updated via websocket api."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth0", "eth1", "vtun0"]},
|
|
}
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([_NO_LOOPBACK_IPADDR]),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
network_obj = hass.data[DOMAIN]
|
|
assert network_obj.configured_adapters == ["eth0", "eth1", "vtun0"]
|
|
ws_client = await hass_ws_client(hass)
|
|
await ws_client.send_json({"id": 1, "type": "network"})
|
|
|
|
response = await ws_client.receive_json()
|
|
|
|
assert response["success"]
|
|
assert response["result"][ATTR_CONFIGURED_ADAPTERS] == ["eth0", "eth1", "vtun0"]
|
|
assert response["result"][ATTR_ADAPTERS] == [
|
|
{
|
|
"auto": False,
|
|
"index": 1,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 8,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
}
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 0,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "127.0.0.1", "network_prefix": 8}],
|
|
"ipv6": [],
|
|
"name": "lo0",
|
|
},
|
|
{
|
|
"auto": True,
|
|
"index": 2,
|
|
"default": True,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 3,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
await ws_client.send_json(
|
|
{"id": 2, "type": "network/configure", "config": {ATTR_CONFIGURED_ADAPTERS: []}}
|
|
)
|
|
response = await ws_client.receive_json()
|
|
assert response["result"][ATTR_CONFIGURED_ADAPTERS] == []
|
|
|
|
await ws_client.send_json({"id": 3, "type": "network"})
|
|
response = await ws_client.receive_json()
|
|
assert response["result"][ATTR_CONFIGURED_ADAPTERS] == []
|
|
assert response["result"][ATTR_ADAPTERS] == [
|
|
{
|
|
"auto": False,
|
|
"index": 1,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 8,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
}
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 0,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "127.0.0.1", "network_prefix": 8}],
|
|
"ipv6": [],
|
|
"name": "lo0",
|
|
},
|
|
{
|
|
"auto": True,
|
|
"index": 2,
|
|
"default": True,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 3,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_async_get_source_ip_matching_interface(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test getting the source ip address with interface matching."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth1"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket(["192.168.1.5"]),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
assert await network.async_get_source_ip(hass, MDNS_TARGET_IP) == "192.168.1.5"
|
|
|
|
|
|
async def test_async_get_source_ip_interface_not_match(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test getting the source ip address with interface does not match."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["vtun0"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket(["192.168.1.5"]),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
assert await network.async_get_source_ip(hass, MDNS_TARGET_IP) == "169.254.3.2"
|
|
|
|
|
|
async def test_async_get_source_ip_cannot_determine_target(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test getting the source ip address when getsockname fails."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth1"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([None]),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
assert await network.async_get_source_ip(hass, MDNS_TARGET_IP) == "192.168.1.5"
|
|
|
|
|
|
async def test_async_get_ipv4_broadcast_addresses_default(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test getting ipv4 broadcast addresses when only the default address is enabled."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth1"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket(["192.168.1.5"]),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
assert await network.async_get_ipv4_broadcast_addresses(hass) == {
|
|
IPv4Address("255.255.255.255")
|
|
}
|
|
|
|
|
|
async def test_async_get_ipv4_broadcast_addresses_multiple(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any]
|
|
) -> None:
|
|
"""Test getting ipv4 broadcast addresses when multiple adapters are enabled."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth1", "vtun0"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([_LOOPBACK_IPADDR]),
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=_generate_mock_adapters(),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
assert await network.async_get_ipv4_broadcast_addresses(hass) == {
|
|
IPv4Address("255.255.255.255"),
|
|
IPv4Address("192.168.1.255"),
|
|
IPv4Address("169.254.255.255"),
|
|
}
|
|
|
|
|
|
async def test_async_get_source_ip_no_enabled_addresses(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any], caplog: pytest.LogCaptureFixture
|
|
) -> None:
|
|
"""Test getting the source ip address when all adapters are disabled."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth1"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=[],
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket(["192.168.1.5"]),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
assert await network.async_get_source_ip(hass, MDNS_TARGET_IP) == "192.168.1.5"
|
|
|
|
assert "source address detection may be inaccurate" in caplog.text
|
|
|
|
|
|
async def test_async_get_source_ip_cannot_be_determined_and_no_enabled_addresses(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any], caplog: pytest.LogCaptureFixture
|
|
) -> None:
|
|
"""Test getting the source ip address when all adapters are disabled and getting it fails."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth1"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=[],
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_socket([None]),
|
|
),
|
|
):
|
|
assert not await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
with pytest.raises(HomeAssistantError):
|
|
await network.async_get_source_ip(hass, MDNS_TARGET_IP)
|
|
|
|
|
|
async def test_async_get_source_ip_no_ip_loopback(
|
|
hass: HomeAssistant, hass_storage: dict[str, Any], caplog: pytest.LogCaptureFixture
|
|
) -> None:
|
|
"""Test getting the source ip address when all adapters are disabled no target is specified."""
|
|
hass_storage[STORAGE_KEY] = {
|
|
"version": STORAGE_VERSION,
|
|
"key": STORAGE_KEY,
|
|
"data": {ATTR_CONFIGURED_ADAPTERS: ["eth1"]},
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.util.ifaddr.get_adapters",
|
|
return_value=[],
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.util.socket.socket",
|
|
return_value=_mock_cond_socket(_LOOPBACK_IPADDR),
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
assert await network.async_get_source_ip(hass) == "127.0.0.1"
|
|
|
|
|
|
_ADAPTERS_WITH_MANUAL_CONFIG = [
|
|
{
|
|
"auto": True,
|
|
"index": 1,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [],
|
|
"ipv6": [
|
|
{
|
|
"address": "2001:db8::",
|
|
"network_prefix": 64,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
},
|
|
{
|
|
"address": "fe80::1234:5678:9abc:def0",
|
|
"network_prefix": 64,
|
|
"flowinfo": 1,
|
|
"scope_id": 1,
|
|
},
|
|
],
|
|
"name": "eth0",
|
|
},
|
|
{
|
|
"auto": True,
|
|
"index": 2,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "192.168.1.5", "network_prefix": 23}],
|
|
"ipv6": [],
|
|
"name": "eth1",
|
|
},
|
|
{
|
|
"auto": True,
|
|
"index": 3,
|
|
"default": False,
|
|
"enabled": True,
|
|
"ipv4": [{"address": "172.16.1.5", "network_prefix": 23}],
|
|
"ipv6": [
|
|
{
|
|
"address": "fe80::dead:beef:dead:beef",
|
|
"network_prefix": 64,
|
|
"flowinfo": 1,
|
|
"scope_id": 3,
|
|
}
|
|
],
|
|
"name": "eth2",
|
|
},
|
|
{
|
|
"auto": False,
|
|
"index": 4,
|
|
"default": False,
|
|
"enabled": False,
|
|
"ipv4": [{"address": "169.254.3.2", "network_prefix": 16}],
|
|
"ipv6": [],
|
|
"name": "vtun0",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_async_get_announce_addresses(hass: HomeAssistant) -> None:
|
|
"""Test addresses for mDNS/etc announcement."""
|
|
first_ip = "172.16.1.5"
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.async_get_source_ip",
|
|
return_value=first_ip,
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.async_get_adapters",
|
|
return_value=_ADAPTERS_WITH_MANUAL_CONFIG,
|
|
),
|
|
):
|
|
actual = await network.async_get_announce_addresses(hass)
|
|
assert actual[0] == first_ip and actual == [
|
|
first_ip,
|
|
"2001:db8::",
|
|
"fe80::1234:5678:9abc:def0",
|
|
"192.168.1.5",
|
|
"fe80::dead:beef:dead:beef",
|
|
]
|
|
|
|
first_ip = "192.168.1.5"
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.async_get_source_ip",
|
|
return_value=first_ip,
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.async_get_adapters",
|
|
return_value=_ADAPTERS_WITH_MANUAL_CONFIG,
|
|
),
|
|
):
|
|
actual = await network.async_get_announce_addresses(hass)
|
|
|
|
assert actual[0] == first_ip and actual == [
|
|
first_ip,
|
|
"2001:db8::",
|
|
"fe80::1234:5678:9abc:def0",
|
|
"172.16.1.5",
|
|
"fe80::dead:beef:dead:beef",
|
|
]
|
|
|
|
|
|
async def test_async_get_announce_addresses_no_source_ip(hass: HomeAssistant) -> None:
|
|
"""Test addresses for mDNS/etc announcement without source ip."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.network.async_get_source_ip",
|
|
return_value=None,
|
|
),
|
|
patch(
|
|
"homeassistant.components.network.async_get_adapters",
|
|
return_value=_ADAPTERS_WITH_MANUAL_CONFIG,
|
|
),
|
|
):
|
|
actual = await network.async_get_announce_addresses(hass)
|
|
assert actual == [
|
|
"2001:db8::",
|
|
"fe80::1234:5678:9abc:def0",
|
|
"192.168.1.5",
|
|
"172.16.1.5",
|
|
"fe80::dead:beef:dead:beef",
|
|
]
|
|
|
|
|
|
async def test_websocket_network_url(
|
|
hass: HomeAssistant, hass_ws_client: WebSocketGenerator
|
|
) -> None:
|
|
"""Test the network/url websocket command."""
|
|
assert await async_setup_component(hass, "network", {})
|
|
|
|
client = await hass_ws_client(hass)
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.helpers.network._get_internal_url", return_value="internal"
|
|
),
|
|
patch("homeassistant.helpers.network._get_cloud_url", return_value="cloud"),
|
|
):
|
|
await client.send_json({"id": 1, "type": "network/url"})
|
|
msg = await client.receive_json()
|
|
assert msg["success"]
|
|
assert msg["result"] == {
|
|
"internal": "internal",
|
|
"external": "cloud",
|
|
"cloud": "cloud",
|
|
}
|
|
|
|
# Test with no cloud URL
|
|
with (
|
|
patch(
|
|
"homeassistant.helpers.network._get_internal_url", return_value="internal"
|
|
),
|
|
):
|
|
await client.send_json({"id": 2, "type": "network/url"})
|
|
msg = await client.receive_json()
|
|
assert msg["success"]
|
|
assert msg["result"] == {
|
|
"internal": "internal",
|
|
"external": None,
|
|
"cloud": None,
|
|
}
|