mirror of
https://github.com/home-assistant/core.git
synced 2026-08-07 13:55:32 +01:00
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
"""Test DROP device registry linkage."""
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.drop_connect.const import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
from .common import config_entry_hub, config_entry_softener
|
|
|
|
HUB_IDENTIFIER = (DOMAIN, "DROP-1_C0FFEE_255")
|
|
SOFTENER_IDENTIFIER = (DOMAIN, "DROP-1_C0FFEE_0")
|
|
|
|
|
|
@pytest.mark.usefixtures("mqtt_mock")
|
|
async def test_sub_device_links_to_hub(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test a sub-device links to its hub via via_device_id when the hub exists."""
|
|
hub_entry = config_entry_hub()
|
|
hub_entry.add_to_hass(hass)
|
|
assert await hass.config_entries.async_setup(hub_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
softener_entry = config_entry_softener()
|
|
softener_entry.add_to_hass(hass)
|
|
assert await hass.config_entries.async_setup(softener_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
hub_device = device_registry.async_get_device_by_identifier(
|
|
HUB_IDENTIFIER, hub_entry.entry_id
|
|
)
|
|
assert hub_device is not None
|
|
|
|
softener_device = device_registry.async_get_device_by_identifier(
|
|
SOFTENER_IDENTIFIER, softener_entry.entry_id
|
|
)
|
|
assert softener_device is not None
|
|
assert softener_device.via_device_id == hub_device.id
|
|
|
|
|
|
@pytest.mark.usefixtures("mqtt_mock")
|
|
async def test_sub_device_without_hub(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test a sub-device set up without its hub is not linked and does not crash."""
|
|
softener_entry = config_entry_softener()
|
|
softener_entry.add_to_hass(hass)
|
|
assert await hass.config_entries.async_setup(softener_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
softener_device = device_registry.async_get_device_by_identifier(
|
|
SOFTENER_IDENTIFIER, softener_entry.entry_id
|
|
)
|
|
assert softener_device is not None
|
|
assert softener_device.via_device_id is None
|