mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 12:59:34 +00:00
Deprecate IPv6 zeroconf setting in favor of the network integration (#51173)
This commit is contained in:
J. Nick Koston
GitHub
@@ -1,7 +1,8 @@
|
||||
"""Test Zeroconf component setup process."""
|
||||
from unittest.mock import call, patch
|
||||
|
||||
from zeroconf import InterfaceChoice, IPVersion, ServiceInfo, ServiceStateChange
|
||||
from zeroconf import InterfaceChoice, IPVersion, ServiceStateChange
|
||||
from zeroconf.asyncio import AsyncServiceInfo
|
||||
|
||||
from homeassistant.components import zeroconf
|
||||
from homeassistant.components.zeroconf import CONF_DEFAULT_INTERFACE, CONF_IPV6
|
||||
@@ -24,29 +25,8 @@ PROPERTIES = {
|
||||
HOMEKIT_STATUS_UNPAIRED = b"1"
|
||||
HOMEKIT_STATUS_PAIRED = b"0"
|
||||
|
||||
_ROUTE_NO_LOOPBACK = (
|
||||
{
|
||||
"attrs": [
|
||||
("RTA_TABLE", 254),
|
||||
("RTA_DST", "224.0.0.251"),
|
||||
("RTA_OIF", 4),
|
||||
("RTA_PREFSRC", "192.168.1.5"),
|
||||
],
|
||||
},
|
||||
)
|
||||
_ROUTE_LOOPBACK = (
|
||||
{
|
||||
"attrs": [
|
||||
("RTA_TABLE", 254),
|
||||
("RTA_DST", "224.0.0.251"),
|
||||
("RTA_OIF", 4),
|
||||
("RTA_PREFSRC", "127.0.0.1"),
|
||||
],
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
def service_update_mock(zeroconf, services, handlers, *, limit_service=None):
|
||||
def service_update_mock(ipv6, zeroconf, services, handlers, *, limit_service=None):
|
||||
"""Call service update handler."""
|
||||
for service in services:
|
||||
if limit_service is not None and service != limit_service:
|
||||
@@ -56,7 +36,7 @@ def service_update_mock(zeroconf, services, handlers, *, limit_service=None):
|
||||
|
||||
def get_service_info_mock(service_type, name):
|
||||
"""Return service info for get_service_info."""
|
||||
return ServiceInfo(
|
||||
return AsyncServiceInfo(
|
||||
service_type,
|
||||
name,
|
||||
addresses=[b"\n\x00\x00\x14"],
|
||||
@@ -70,7 +50,7 @@ def get_service_info_mock(service_type, name):
|
||||
|
||||
def get_service_info_mock_without_an_address(service_type, name):
|
||||
"""Return service info for get_service_info without any addresses."""
|
||||
return ServiceInfo(
|
||||
return AsyncServiceInfo(
|
||||
service_type,
|
||||
name,
|
||||
addresses=[],
|
||||
@@ -86,7 +66,7 @@ def get_homekit_info_mock(model, pairing_status):
|
||||
"""Return homekit info for get_service_info for an homekit device."""
|
||||
|
||||
def mock_homekit_info(service_type, name):
|
||||
return ServiceInfo(
|
||||
return AsyncServiceInfo(
|
||||
service_type,
|
||||
name,
|
||||
addresses=[b"\n\x00\x00\x14"],
|
||||
@@ -104,7 +84,7 @@ def get_zeroconf_info_mock(macaddress):
|
||||
"""Return info for get_service_info for an zeroconf device."""
|
||||
|
||||
def mock_zc_info(service_type, name):
|
||||
return ServiceInfo(
|
||||
return AsyncServiceInfo(
|
||||
service_type,
|
||||
name,
|
||||
addresses=[b"\n\x00\x00\x14"],
|
||||
@@ -122,7 +102,7 @@ def get_zeroconf_info_mock_manufacturer(manufacturer):
|
||||
"""Return info for get_service_info for an zeroconf device."""
|
||||
|
||||
def mock_zc_info(service_type, name):
|
||||
return ServiceInfo(
|
||||
return AsyncServiceInfo(
|
||||
service_type,
|
||||
name,
|
||||
addresses=[b"\n\x00\x00\x14"],
|
||||
@@ -136,14 +116,14 @@ def get_zeroconf_info_mock_manufacturer(manufacturer):
|
||||
return mock_zc_info
|
||||
|
||||
|
||||
async def test_setup(hass, mock_zeroconf):
|
||||
async def test_setup(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
with patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -162,13 +142,15 @@ async def test_setup(hass, mock_zeroconf):
|
||||
|
||||
# Test instance is set.
|
||||
assert "zeroconf" in hass.data
|
||||
assert await hass.components.zeroconf.async_get_instance() is mock_zeroconf
|
||||
assert (
|
||||
await hass.components.zeroconf.async_get_async_instance() is mock_async_zeroconf
|
||||
)
|
||||
|
||||
|
||||
async def test_setup_with_overly_long_url_and_name(hass, mock_zeroconf, caplog):
|
||||
async def test_setup_with_overly_long_url_and_name(hass, mock_async_zeroconf, caplog):
|
||||
"""Test we still setup with long urls and names."""
|
||||
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.get_url",
|
||||
return_value="https://this.url.is.way.too.long/very/deep/path/that/will/make/us/go/over/the/maximum/string/length/and/would/cause/zeroconf/to/fail/to/startup/because/the/key/and/value/can/only/be/255/bytes/and/this/string/is/a/bit/longer/than/the/maximum/length/that/we/allow/for/a/value",
|
||||
@@ -177,7 +159,7 @@ async def test_setup_with_overly_long_url_and_name(hass, mock_zeroconf, caplog):
|
||||
"location_name",
|
||||
"\u00dcBER \u00dcber German Umlaut long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string long string",
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo.request",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo.request",
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
||||
@@ -187,12 +169,12 @@ async def test_setup_with_overly_long_url_and_name(hass, mock_zeroconf, caplog):
|
||||
assert "German Umlaut" in caplog.text
|
||||
|
||||
|
||||
async def test_setup_with_default_interface(hass, mock_zeroconf):
|
||||
async def test_setup_with_default_interface(hass, mock_async_zeroconf):
|
||||
"""Test default interface config."""
|
||||
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(
|
||||
@@ -201,30 +183,30 @@ async def test_setup_with_default_interface(hass, mock_zeroconf):
|
||||
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert mock_zeroconf.called_with(interface_choice=InterfaceChoice.Default)
|
||||
assert mock_async_zeroconf.called_with(interface_choice=InterfaceChoice.Default)
|
||||
|
||||
|
||||
async def test_setup_without_default_interface(hass, mock_zeroconf):
|
||||
async def test_setup_without_default_interface(hass, mock_async_zeroconf):
|
||||
"""Test without default interface config."""
|
||||
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(
|
||||
hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {CONF_DEFAULT_INTERFACE: False}}
|
||||
)
|
||||
|
||||
assert mock_zeroconf.called_with()
|
||||
assert mock_async_zeroconf.called_with()
|
||||
|
||||
|
||||
async def test_setup_without_ipv6(hass, mock_zeroconf):
|
||||
async def test_setup_without_ipv6(hass, mock_async_zeroconf):
|
||||
"""Test without ipv6."""
|
||||
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(
|
||||
@@ -233,15 +215,15 @@ async def test_setup_without_ipv6(hass, mock_zeroconf):
|
||||
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert mock_zeroconf.called_with(ip_version=IPVersion.V4Only)
|
||||
assert mock_async_zeroconf.called_with(ip_version=IPVersion.V4Only)
|
||||
|
||||
|
||||
async def test_setup_with_ipv6(hass, mock_zeroconf):
|
||||
async def test_setup_with_ipv6(hass, mock_async_zeroconf):
|
||||
"""Test without ipv6."""
|
||||
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(
|
||||
@@ -250,28 +232,28 @@ async def test_setup_with_ipv6(hass, mock_zeroconf):
|
||||
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert mock_zeroconf.called_with()
|
||||
assert mock_async_zeroconf.called_with()
|
||||
|
||||
|
||||
async def test_setup_with_ipv6_default(hass, mock_zeroconf):
|
||||
async def test_setup_with_ipv6_default(hass, mock_async_zeroconf):
|
||||
"""Test without ipv6 as default."""
|
||||
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert mock_zeroconf.called_with()
|
||||
assert mock_async_zeroconf.called_with()
|
||||
|
||||
|
||||
async def test_zeroconf_match_macaddress(hass, mock_zeroconf):
|
||||
async def test_zeroconf_match_macaddress(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
|
||||
def http_only_service_update_mock(zeroconf, services, handlers):
|
||||
def http_only_service_update_mock(ipv6, zeroconf, services, handlers):
|
||||
"""Call service update handler."""
|
||||
handlers[0](
|
||||
zeroconf,
|
||||
@@ -291,9 +273,9 @@ async def test_zeroconf_match_macaddress(hass, mock_zeroconf):
|
||||
), patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=http_only_service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=http_only_service_update_mock
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_zeroconf_info_mock("FFAADDCC11DD"),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -305,10 +287,10 @@ async def test_zeroconf_match_macaddress(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[0][1][0] == "shelly"
|
||||
|
||||
|
||||
async def test_zeroconf_match_manufacturer(hass, mock_zeroconf):
|
||||
async def test_zeroconf_match_manufacturer(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
|
||||
def http_only_service_update_mock(zeroconf, services, handlers):
|
||||
def http_only_service_update_mock(ipv6, zeroconf, services, handlers):
|
||||
"""Call service update handler."""
|
||||
handlers[0](
|
||||
zeroconf,
|
||||
@@ -324,9 +306,9 @@ async def test_zeroconf_match_manufacturer(hass, mock_zeroconf):
|
||||
), patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=http_only_service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=http_only_service_update_mock
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_zeroconf_info_mock_manufacturer("Samsung Electronics"),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -338,10 +320,10 @@ async def test_zeroconf_match_manufacturer(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[0][1][0] == "samsungtv"
|
||||
|
||||
|
||||
async def test_zeroconf_match_manufacturer_not_present(hass, mock_zeroconf):
|
||||
async def test_zeroconf_match_manufacturer_not_present(hass, mock_async_zeroconf):
|
||||
"""Test matchers reject when a property is missing."""
|
||||
|
||||
def http_only_service_update_mock(zeroconf, services, handlers):
|
||||
def http_only_service_update_mock(ipv6, zeroconf, services, handlers):
|
||||
"""Call service update handler."""
|
||||
handlers[0](
|
||||
zeroconf,
|
||||
@@ -357,9 +339,9 @@ async def test_zeroconf_match_manufacturer_not_present(hass, mock_zeroconf):
|
||||
), patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=http_only_service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=http_only_service_update_mock
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_zeroconf_info_mock("aabbccddeeff"),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -370,10 +352,10 @@ async def test_zeroconf_match_manufacturer_not_present(hass, mock_zeroconf):
|
||||
assert len(mock_config_flow.mock_calls) == 0
|
||||
|
||||
|
||||
async def test_zeroconf_no_match(hass, mock_zeroconf):
|
||||
async def test_zeroconf_no_match(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
|
||||
def http_only_service_update_mock(zeroconf, services, handlers):
|
||||
def http_only_service_update_mock(ipv6, zeroconf, services, handlers):
|
||||
"""Call service update handler."""
|
||||
handlers[0](
|
||||
zeroconf,
|
||||
@@ -389,9 +371,9 @@ async def test_zeroconf_no_match(hass, mock_zeroconf):
|
||||
), patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=http_only_service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=http_only_service_update_mock
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_zeroconf_info_mock("FFAADDCC11DD"),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -402,10 +384,10 @@ async def test_zeroconf_no_match(hass, mock_zeroconf):
|
||||
assert len(mock_config_flow.mock_calls) == 0
|
||||
|
||||
|
||||
async def test_zeroconf_no_match_manufacturer(hass, mock_zeroconf):
|
||||
async def test_zeroconf_no_match_manufacturer(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
|
||||
def http_only_service_update_mock(zeroconf, services, handlers):
|
||||
def http_only_service_update_mock(ipv6, zeroconf, services, handlers):
|
||||
"""Call service update handler."""
|
||||
handlers[0](
|
||||
zeroconf,
|
||||
@@ -421,9 +403,9 @@ async def test_zeroconf_no_match_manufacturer(hass, mock_zeroconf):
|
||||
), patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=http_only_service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=http_only_service_update_mock
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_zeroconf_info_mock_manufacturer("Not Samsung Electronics"),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -434,7 +416,7 @@ async def test_zeroconf_no_match_manufacturer(hass, mock_zeroconf):
|
||||
assert len(mock_config_flow.mock_calls) == 0
|
||||
|
||||
|
||||
async def test_homekit_match_partial_space(hass, mock_zeroconf):
|
||||
async def test_homekit_match_partial_space(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
with patch.dict(
|
||||
zc_gen.ZEROCONF,
|
||||
@@ -444,12 +426,12 @@ async def test_homekit_match_partial_space(hass, mock_zeroconf):
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf,
|
||||
"HaServiceBrowser",
|
||||
"HaAsyncServiceBrowser",
|
||||
side_effect=lambda *args, **kwargs: service_update_mock(
|
||||
*args, **kwargs, limit_service="_hap._tcp.local."
|
||||
),
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_homekit_info_mock("LIFX bulb", HOMEKIT_STATUS_UNPAIRED),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -461,7 +443,7 @@ async def test_homekit_match_partial_space(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[0][1][0] == "lifx"
|
||||
|
||||
|
||||
async def test_homekit_match_partial_dash(hass, mock_zeroconf):
|
||||
async def test_homekit_match_partial_dash(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
with patch.dict(
|
||||
zc_gen.ZEROCONF,
|
||||
@@ -471,12 +453,12 @@ async def test_homekit_match_partial_dash(hass, mock_zeroconf):
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf,
|
||||
"HaServiceBrowser",
|
||||
"HaAsyncServiceBrowser",
|
||||
side_effect=lambda *args, **kwargs: service_update_mock(
|
||||
*args, **kwargs, limit_service="_hap._udp.local."
|
||||
),
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_homekit_info_mock("Rachio-fa46ba", HOMEKIT_STATUS_UNPAIRED),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -488,7 +470,7 @@ async def test_homekit_match_partial_dash(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[0][1][0] == "rachio"
|
||||
|
||||
|
||||
async def test_homekit_match_partial_fnmatch(hass, mock_zeroconf):
|
||||
async def test_homekit_match_partial_fnmatch(hass, mock_async_zeroconf):
|
||||
"""Test matching homekit devices with fnmatch."""
|
||||
with patch.dict(
|
||||
zc_gen.ZEROCONF,
|
||||
@@ -498,12 +480,12 @@ async def test_homekit_match_partial_fnmatch(hass, mock_zeroconf):
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf,
|
||||
"HaServiceBrowser",
|
||||
"HaAsyncServiceBrowser",
|
||||
side_effect=lambda *args, **kwargs: service_update_mock(
|
||||
*args, **kwargs, limit_service="_hap._tcp.local."
|
||||
),
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_homekit_info_mock("YLDP13YL", HOMEKIT_STATUS_UNPAIRED),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -515,7 +497,7 @@ async def test_homekit_match_partial_fnmatch(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[0][1][0] == "yeelight"
|
||||
|
||||
|
||||
async def test_homekit_match_full(hass, mock_zeroconf):
|
||||
async def test_homekit_match_full(hass, mock_async_zeroconf):
|
||||
"""Test configured options for a device are loaded via config entry."""
|
||||
with patch.dict(
|
||||
zc_gen.ZEROCONF,
|
||||
@@ -525,12 +507,12 @@ async def test_homekit_match_full(hass, mock_zeroconf):
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf,
|
||||
"HaServiceBrowser",
|
||||
"HaAsyncServiceBrowser",
|
||||
side_effect=lambda *args, **kwargs: service_update_mock(
|
||||
*args, **kwargs, limit_service="_hap._udp.local."
|
||||
),
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_homekit_info_mock("BSB002", HOMEKIT_STATUS_UNPAIRED),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -542,7 +524,7 @@ async def test_homekit_match_full(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[0][1][0] == "hue"
|
||||
|
||||
|
||||
async def test_homekit_already_paired(hass, mock_zeroconf):
|
||||
async def test_homekit_already_paired(hass, mock_async_zeroconf):
|
||||
"""Test that an already paired device is sent to homekit_controller."""
|
||||
with patch.dict(
|
||||
zc_gen.ZEROCONF,
|
||||
@@ -552,12 +534,12 @@ async def test_homekit_already_paired(hass, mock_zeroconf):
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf,
|
||||
"HaServiceBrowser",
|
||||
"HaAsyncServiceBrowser",
|
||||
side_effect=lambda *args, **kwargs: service_update_mock(
|
||||
*args, **kwargs, limit_service="_hap._tcp.local."
|
||||
),
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_homekit_info_mock("tado", HOMEKIT_STATUS_PAIRED),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -570,7 +552,7 @@ async def test_homekit_already_paired(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[1][1][0] == "homekit_controller"
|
||||
|
||||
|
||||
async def test_homekit_invalid_paring_status(hass, mock_zeroconf):
|
||||
async def test_homekit_invalid_paring_status(hass, mock_async_zeroconf):
|
||||
"""Test that missing paring data is not sent to homekit_controller."""
|
||||
with patch.dict(
|
||||
zc_gen.ZEROCONF,
|
||||
@@ -580,12 +562,12 @@ async def test_homekit_invalid_paring_status(hass, mock_zeroconf):
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf,
|
||||
"HaServiceBrowser",
|
||||
"HaAsyncServiceBrowser",
|
||||
side_effect=lambda *args, **kwargs: service_update_mock(
|
||||
*args, **kwargs, limit_service="_hap._tcp.local."
|
||||
),
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_homekit_info_mock("tado", b"invalid"),
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -597,7 +579,7 @@ async def test_homekit_invalid_paring_status(hass, mock_zeroconf):
|
||||
assert mock_config_flow.mock_calls[0][1][0] == "tado"
|
||||
|
||||
|
||||
async def test_homekit_not_paired(hass, mock_zeroconf):
|
||||
async def test_homekit_not_paired(hass, mock_async_zeroconf):
|
||||
"""Test that an not paired device is sent to homekit_controller."""
|
||||
with patch.dict(
|
||||
zc_gen.ZEROCONF,
|
||||
@@ -606,9 +588,9 @@ async def test_homekit_not_paired(hass, mock_zeroconf):
|
||||
), patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
) as mock_config_flow, patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
) as mock_service_browser, patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_homekit_info_mock(
|
||||
"this_will_not_match_any_integration", HOMEKIT_STATUS_UNPAIRED
|
||||
),
|
||||
@@ -646,19 +628,21 @@ async def test_info_from_service_with_addresses(hass):
|
||||
assert info is None
|
||||
|
||||
|
||||
async def test_get_instance(hass, mock_zeroconf):
|
||||
async def test_get_instance(hass, mock_async_zeroconf):
|
||||
"""Test we get an instance."""
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
assert await hass.components.zeroconf.async_get_instance() is mock_zeroconf
|
||||
assert (
|
||||
await hass.components.zeroconf.async_get_async_instance() is mock_async_zeroconf
|
||||
)
|
||||
hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
|
||||
await hass.async_block_till_done()
|
||||
assert len(mock_zeroconf.ha_close.mock_calls) == 1
|
||||
assert len(mock_async_zeroconf.ha_async_close.mock_calls) == 1
|
||||
|
||||
|
||||
async def test_removed_ignored(hass, mock_zeroconf):
|
||||
async def test_removed_ignored(hass, mock_async_zeroconf):
|
||||
"""Test we remove it when a zeroconf entry is removed."""
|
||||
|
||||
def service_update_mock(zeroconf, services, handlers):
|
||||
def service_update_mock(ipv6, zeroconf, services, handlers):
|
||||
"""Call service update handler."""
|
||||
handlers[0](
|
||||
zeroconf,
|
||||
@@ -680,9 +664,9 @@ async def test_removed_ignored(hass, mock_zeroconf):
|
||||
)
|
||||
|
||||
with patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
) as mock_service_info:
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
@@ -709,24 +693,28 @@ _ADAPTER_WITH_DEFAULT_ENABLED = [
|
||||
]
|
||||
|
||||
|
||||
async def test_async_detect_interfaces_setting_non_loopback_route(hass):
|
||||
async def test_async_detect_interfaces_setting_non_loopback_route(
|
||||
hass, mock_async_zeroconf
|
||||
):
|
||||
"""Test without default interface config and the route returns a non-loopback address."""
|
||||
with patch(
|
||||
"homeassistant.components.zeroconf.models.HaZeroconf"
|
||||
) as mock_zc, patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
with patch("homeassistant.components.zeroconf.HaZeroconf") as mock_zc, patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
), patch.object(
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.network.async_get_adapters",
|
||||
return_value=_ADAPTER_WITH_DEFAULT_ENABLED,
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert mock_zc.mock_calls[0] == call(interfaces=InterfaceChoice.Default)
|
||||
assert mock_zc.mock_calls[0] == call(
|
||||
interfaces=InterfaceChoice.Default, ip_version=IPVersion.V4Only
|
||||
)
|
||||
|
||||
|
||||
_ADAPTERS_WITH_MANUAL_CONFIG = [
|
||||
@@ -764,17 +752,17 @@ _ADAPTERS_WITH_MANUAL_CONFIG = [
|
||||
]
|
||||
|
||||
|
||||
async def test_async_detect_interfaces_setting_empty_route(hass):
|
||||
async def test_async_detect_interfaces_setting_empty_route(hass, mock_async_zeroconf):
|
||||
"""Test without default interface config and the route returns nothing."""
|
||||
with patch(
|
||||
"homeassistant.components.zeroconf.models.HaZeroconf"
|
||||
) as mock_zc, patch.object(hass.config_entries.flow, "async_init"), patch.object(
|
||||
zeroconf, "HaServiceBrowser", side_effect=service_update_mock
|
||||
with patch("homeassistant.components.zeroconf.HaZeroconf") as mock_zc, patch.object(
|
||||
hass.config_entries.flow, "async_init"
|
||||
), patch.object(
|
||||
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.network.async_get_adapters",
|
||||
return_value=_ADAPTERS_WITH_MANUAL_CONFIG,
|
||||
), patch(
|
||||
"homeassistant.components.zeroconf.ServiceInfo",
|
||||
"homeassistant.components.zeroconf.AsyncServiceInfo",
|
||||
side_effect=get_service_info_mock,
|
||||
):
|
||||
assert await async_setup_component(hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {}})
|
||||
|
||||
Reference in New Issue
Block a user