mirror of
https://github.com/home-assistant/core.git
synced 2026-05-23 08:50:09 +01:00
epenet
76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
"""Test the UniFi Discovery init."""
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.unifi_discovery.const import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from . import (
|
|
UNIFI_DISCOVERY_MAPPINGPROXY_SERVICES,
|
|
UNIFI_DISCOVERY_NO_MAC,
|
|
_patch_discovery,
|
|
)
|
|
|
|
|
|
async def test_setup_starts_discovery(hass: HomeAssistant) -> None:
|
|
"""Test that async_setup starts discovery and dispatches flows."""
|
|
with _patch_discovery():
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
# The scanner should have dispatched a discovery flow for the Protect consumer
|
|
flows = hass.config_entries.flow.async_progress_by_handler("unifiprotect")
|
|
assert len(flows) == 1
|
|
assert flows[0]["context"]["source"] == config_entries.SOURCE_INTEGRATION_DISCOVERY
|
|
|
|
|
|
async def test_setup_no_devices(hass: HomeAssistant) -> None:
|
|
"""Test setup with no devices found."""
|
|
with _patch_discovery(no_device=True):
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
flows = hass.config_entries.flow.async_progress_by_handler("unifiprotect")
|
|
assert len(flows) == 0
|
|
|
|
|
|
async def test_setup_device_without_mac(hass: HomeAssistant) -> None:
|
|
"""Test that devices without hw_addr are skipped."""
|
|
with _patch_discovery(device=UNIFI_DISCOVERY_NO_MAC):
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
flows = hass.config_entries.flow.async_progress_by_handler("unifiprotect")
|
|
assert len(flows) == 0
|
|
|
|
|
|
async def test_dependency_loads_discovery(
|
|
hass: HomeAssistant,
|
|
) -> None:
|
|
"""Test that loading unifiprotect triggers unifi_discovery as dependency."""
|
|
with _patch_discovery():
|
|
assert await async_setup_component(hass, "unifiprotect", {})
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
# unifi_discovery should have been loaded as a dependency and started scanning
|
|
flows = hass.config_entries.flow.async_progress_by_handler("unifiprotect")
|
|
assert len(flows) == 1
|
|
assert flows[0]["context"]["source"] == config_entries.SOURCE_INTEGRATION_DISCOVERY
|
|
|
|
|
|
async def test_discovery_does_not_deepcopy_device(hass: HomeAssistant) -> None:
|
|
"""Test discovery works without deepcopy.
|
|
|
|
In production asdict() deep-copies Enum keys in the services dict which
|
|
can crash on Python 3.14+ because Enum.__members__ is a mappingproxy that
|
|
cannot be pickled. We force the crash reliably by using MappingProxyType
|
|
as the services value.
|
|
"""
|
|
with _patch_discovery(device=UNIFI_DISCOVERY_MAPPINGPROXY_SERVICES):
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
flows = hass.config_entries.flow.async_progress_by_handler("unifiprotect")
|
|
assert len(flows) == 1
|
|
assert flows[0]["context"]["source"] == config_entries.SOURCE_INTEGRATION_DISCOVERY
|