aioesphomeapi/tests/test_host_resolver.py

271 lines
9.3 KiB
Python
Raw Normal View History

from __future__ import annotations
import asyncio
2021-07-12 20:09:17 +02:00
import socket
from ipaddress import IPv6Address, ip_address
from unittest.mock import AsyncMock, MagicMock, patch
2021-07-12 20:09:17 +02:00
import pytest
from zeroconf.asyncio import AsyncServiceInfo, AsyncZeroconf
2021-07-12 20:09:17 +02:00
import aioesphomeapi.host_resolver as hr
from aioesphomeapi.core import APIConnectionError, ResolveAPIError
from aioesphomeapi.zeroconf import ZeroconfManager
2021-07-12 20:09:17 +02:00
@pytest.fixture
def addr_infos():
return [
hr.AddrInfo(
family=socket.AF_INET,
type=socket.SOCK_STREAM,
proto=socket.IPPROTO_TCP,
sockaddr=hr.IPv4Sockaddr(address="10.0.0.42", port=6052),
),
hr.AddrInfo(
family=socket.AF_INET6,
type=socket.SOCK_STREAM,
proto=socket.IPPROTO_TCP,
sockaddr=hr.IPv6Sockaddr(
address="2001:db8:85a3::8a2e:370:7334",
port=6052,
flowinfo=0,
scope_id=0,
),
),
]
@pytest.mark.asyncio
2023-10-16 01:05:23 +02:00
async def test_resolve_host_zeroconf(async_zeroconf: AsyncZeroconf, addr_infos):
info = MagicMock(auto_spec=AsyncServiceInfo)
ipv6 = IPv6Address("2001:db8:85a3::8a2e:370:7334%0")
info.ip_addresses_by_version.side_effect = [
[ip_address(b"\n\x00\x00*")],
[ipv6],
2021-07-12 20:09:17 +02:00
]
2023-10-16 01:05:23 +02:00
info.async_request = AsyncMock(return_value=True)
with (
patch("aioesphomeapi.host_resolver.AsyncServiceInfo", return_value=info),
patch("aioesphomeapi.zeroconf.AsyncZeroconf", return_value=async_zeroconf),
):
2023-10-16 01:05:23 +02:00
ret = await hr._async_resolve_host_zeroconf("asdf", 6052)
2021-07-12 20:09:17 +02:00
2023-10-16 01:05:23 +02:00
info.async_request.assert_called_once()
2021-07-12 20:09:17 +02:00
async_zeroconf.async_close.assert_called_once_with()
assert ret == addr_infos
@pytest.mark.asyncio
async def test_resolve_host_passed_zeroconf(addr_infos, async_zeroconf):
zeroconf_manager = ZeroconfManager()
2023-10-16 01:05:23 +02:00
info = MagicMock(auto_spec=AsyncServiceInfo)
ipv6 = IPv6Address("2001:db8:85a3::8a2e:370:7334%0")
info.ip_addresses_by_version.side_effect = [
[ip_address(b"\n\x00\x00*")],
[ipv6],
2023-10-16 01:05:23 +02:00
]
info.async_request = AsyncMock(return_value=True)
with patch("aioesphomeapi.host_resolver.AsyncServiceInfo", return_value=info):
ret = await hr._async_resolve_host_zeroconf(
"asdf", 6052, zeroconf_manager=zeroconf_manager
2023-10-16 01:05:23 +02:00
)
info.async_request.assert_called_once()
assert ret == addr_infos
2021-07-12 20:09:17 +02:00
2023-10-16 01:05:23 +02:00
@pytest.mark.asyncio
async def test_resolve_host_zeroconf_empty(async_zeroconf: AsyncZeroconf):
with patch(
"aioesphomeapi.host_resolver.AsyncServiceInfo.async_request"
) as mock_async_request:
ret = await hr._async_resolve_host_zeroconf("asdf.local", 6052)
assert mock_async_request.call_count == 1
2021-07-12 20:09:17 +02:00
assert ret == []
@pytest.mark.asyncio
async def test_resolve_host_zeroconf_fails(async_zeroconf: AsyncZeroconf):
with (
patch(
"aioesphomeapi.host_resolver.AsyncServiceInfo.async_request",
side_effect=Exception("no buffers"),
),
pytest.raises(ResolveAPIError, match="no buffers"),
):
await hr._async_resolve_host_zeroconf("asdf.local", 6052)
@pytest.mark.asyncio
@patch("aioesphomeapi.host_resolver._async_resolve_host_getaddrinfo", return_value=[])
async def test_resolve_host_zeroconf_fails_end_to_end(async_zeroconf: AsyncZeroconf):
with (
patch(
"aioesphomeapi.host_resolver.AsyncServiceInfo.async_request",
side_effect=Exception("no buffers"),
),
pytest.raises(ResolveAPIError, match="no buffers"),
):
await hr.async_resolve_host(["asdf.local"], 6052)
2021-07-12 20:09:17 +02:00
@pytest.mark.asyncio
async def test_resolve_host_getaddrinfo(addr_infos):
event_loop = asyncio.get_running_loop()
2021-10-13 10:15:30 +02:00
with patch.object(event_loop, "getaddrinfo") as mock:
mock.return_value = [
(
socket.AF_INET,
socket.SOCK_STREAM,
socket.IPPROTO_TCP,
"canon1",
("10.0.0.42", 6052),
),
(
socket.AF_INET6,
socket.SOCK_STREAM,
socket.IPPROTO_TCP,
"canon2",
("2001:db8:85a3::8a2e:370:7334", 6052, 0, 0),
),
(-1, socket.SOCK_STREAM, socket.IPPROTO_TCP, "canon3", ("10.0.0.42", 6052)),
]
ret = await hr._async_resolve_host_getaddrinfo("example.com", 6052)
2021-07-12 20:09:17 +02:00
2021-10-13 10:15:30 +02:00
assert ret == addr_infos
2021-07-12 20:09:17 +02:00
@pytest.mark.asyncio
async def test_resolve_host_getaddrinfo_oserror():
event_loop = asyncio.get_running_loop()
2021-10-13 10:15:30 +02:00
with patch.object(event_loop, "getaddrinfo") as mock:
mock.side_effect = OSError()
with pytest.raises(APIConnectionError):
await hr._async_resolve_host_getaddrinfo("example.com", 6052)
2021-07-12 20:09:17 +02:00
@pytest.mark.asyncio
@patch("aioesphomeapi.host_resolver._async_resolve_host_zeroconf")
@patch("aioesphomeapi.host_resolver._async_resolve_host_getaddrinfo")
async def test_resolve_host_mdns(resolve_addr, resolve_zc, addr_infos):
resolve_zc.return_value = addr_infos
ret = await hr.async_resolve_host(["example.local"], 6052)
2021-07-12 20:09:17 +02:00
resolve_zc.assert_called_once_with("example", 6052, zeroconf_manager=None)
2021-07-12 20:09:17 +02:00
resolve_addr.assert_not_called()
assert ret == addr_infos
2021-07-12 20:09:17 +02:00
@pytest.mark.asyncio
@patch("aioesphomeapi.host_resolver._async_resolve_host_zeroconf")
@patch("aioesphomeapi.host_resolver._async_resolve_host_getaddrinfo")
async def test_resolve_host_mdns_empty(resolve_addr, resolve_zc, addr_infos):
resolve_zc.return_value = []
resolve_addr.return_value = addr_infos
ret = await hr.async_resolve_host(["example.local"], 6052)
2021-07-12 20:09:17 +02:00
resolve_zc.assert_called_once_with("example", 6052, zeroconf_manager=None)
2021-10-13 10:15:30 +02:00
resolve_addr.assert_called_once_with("example.local", 6052)
assert ret == addr_infos
2021-07-12 20:09:17 +02:00
@pytest.mark.asyncio
@patch("aioesphomeapi.host_resolver.AsyncServiceInfo.async_request", return_value=False)
@patch("aioesphomeapi.host_resolver._async_resolve_host_getaddrinfo")
async def test_resolve_host_mdns_no_results(resolve_addr, addr_infos):
resolve_addr.return_value = addr_infos
with pytest.raises(ResolveAPIError):
await hr.async_resolve_host(["example.local"], 6052)
2021-07-12 20:09:17 +02:00
@pytest.mark.asyncio
@patch("aioesphomeapi.host_resolver._async_resolve_host_zeroconf")
@patch("aioesphomeapi.host_resolver._async_resolve_host_getaddrinfo")
async def test_resolve_host_addrinfo(resolve_addr, resolve_zc, addr_infos):
resolve_addr.return_value = addr_infos
ret = await hr.async_resolve_host(["example.com"], 6052)
2021-07-12 20:09:17 +02:00
resolve_zc.assert_not_called()
2021-10-13 10:15:30 +02:00
resolve_addr.assert_called_once_with("example.com", 6052)
assert ret == addr_infos
2021-07-12 20:09:17 +02:00
@pytest.mark.asyncio
@patch("aioesphomeapi.host_resolver._async_resolve_host_zeroconf")
@patch("aioesphomeapi.host_resolver._async_resolve_host_getaddrinfo")
async def test_resolve_host_addrinfo_empty(resolve_addr, resolve_zc, addr_infos):
resolve_addr.return_value = []
with pytest.raises(APIConnectionError):
await hr.async_resolve_host(["example.com"], 6052)
2021-07-12 20:09:17 +02:00
resolve_zc.assert_not_called()
2021-10-13 10:15:30 +02:00
resolve_addr.assert_called_once_with("example.com", 6052)
@pytest.mark.asyncio
@patch("aioesphomeapi.host_resolver._async_resolve_host_zeroconf")
@patch("aioesphomeapi.host_resolver._async_resolve_host_getaddrinfo")
async def test_resolve_host_with_address(resolve_addr, resolve_zc):
resolve_zc.return_value = []
resolve_addr.return_value = addr_infos
ret = await hr.async_resolve_host(["127.0.0.1"], 6052)
resolve_zc.assert_not_called()
resolve_addr.assert_not_called()
assert ret == [
hr.AddrInfo(
family=socket.AddressFamily.AF_INET,
type=socket.SocketKind.SOCK_STREAM,
proto=6,
sockaddr=hr.IPv4Sockaddr(address="127.0.0.1", port=6052),
)
]
@pytest.mark.asyncio
async def test_resolve_host_zeroconf_service_info_oserror(
async_zeroconf: AsyncZeroconf, addr_infos
):
info = MagicMock(auto_spec=AsyncServiceInfo)
info.ip_addresses_by_version.return_value = [
ip_address(b"\n\x00\x00*"),
ip_address(b" \x01\r\xb8\x85\xa3\x00\x00\x00\x00\x8a.\x03ps4"),
]
info.async_request = AsyncMock(return_value=True)
with (
patch(
"aioesphomeapi.host_resolver.AsyncServiceInfo.async_request",
side_effect=OSError("out of buffers"),
),
patch("aioesphomeapi.zeroconf.AsyncZeroconf", return_value=async_zeroconf),
pytest.raises(ResolveAPIError, match="out of buffers"),
):
await hr._async_resolve_host_zeroconf("asdf", 6052)
@pytest.mark.asyncio
async def test_resolve_host_create_zeroconf_oserror(
async_zeroconf: AsyncZeroconf, addr_infos
):
info = MagicMock(auto_spec=AsyncServiceInfo)
info.ip_addresses_by_version.return_value = [
ip_address(b"\n\x00\x00*"),
ip_address(b" \x01\r\xb8\x85\xa3\x00\x00\x00\x00\x8a.\x03ps4"),
]
info.async_request = AsyncMock(return_value=True)
with (
patch(
"aioesphomeapi.zeroconf.AsyncZeroconf",
side_effect=OSError("out of buffers"),
),
pytest.raises(ResolveAPIError, match="out of buffers"),
):
await hr._async_resolve_host_zeroconf("asdf", 6052)
def test_scope_id_to_int():
assert hr._scope_id_to_int("123") == 123
assert hr._scope_id_to_int("eth0") == 0
assert hr._scope_id_to_int(None) == 0