mirror of
https://github.com/home-assistant/core.git
synced 2026-08-07 13:55:32 +01:00
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
"""Test Notion setup."""
|
|
|
|
from copy import deepcopy
|
|
from typing import Any
|
|
|
|
from aionotion.listener.models import ListenerKind
|
|
import pytest
|
|
|
|
from homeassistant.components.notion.const import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
# The id of the bridge in the bridge fixture; the sensor must reference it to link.
|
|
BRIDGE_ID = 12345
|
|
BRIDGE_HARDWARE_ID = "0x0000000000000012"
|
|
SENSOR_HARDWARE_ID = "0x0000000000000034"
|
|
|
|
|
|
@pytest.fixture(name="data_bridge")
|
|
def data_bridge_fixture(data_bridge: dict[str, Any]) -> dict[str, Any]:
|
|
"""Give the bridge a hardware id distinct from the sensor."""
|
|
data = deepcopy(data_bridge)
|
|
data["base_stations"][0]["hardware_id"] = BRIDGE_HARDWARE_ID
|
|
return data
|
|
|
|
|
|
@pytest.fixture(name="data_sensor")
|
|
def data_sensor_fixture(data_sensor: dict[str, Any]) -> dict[str, Any]:
|
|
"""Link the sensor to the bridge and give it a distinct hardware id."""
|
|
data = deepcopy(data_sensor)
|
|
data["sensors"][0]["hardware_id"] = SENSOR_HARDWARE_ID
|
|
data["sensors"][0]["bridge"]["id"] = BRIDGE_ID
|
|
data["sensors"][0]["bridge"]["hardware_id"] = BRIDGE_HARDWARE_ID
|
|
return data
|
|
|
|
|
|
@pytest.fixture(name="data_listener")
|
|
def data_listener_fixture(data_listener: dict[str, Any]) -> dict[str, Any]:
|
|
"""Use a listener kind that maps to an entity so a sensor device is created."""
|
|
data = deepcopy(data_listener)
|
|
data["listeners"][0]["definition_id"] = ListenerKind.HINGED_WINDOW.value
|
|
return data
|
|
|
|
|
|
@pytest.mark.usefixtures("setup_config_entry")
|
|
async def test_device_via_device_links(
|
|
hass: HomeAssistant,
|
|
config_entry: MockConfigEntry,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test that a sensor device links to its bridge via via_device_id."""
|
|
bridge_device = device_registry.async_get_device_by_identifier(
|
|
(DOMAIN, BRIDGE_HARDWARE_ID), config_entry.entry_id
|
|
)
|
|
assert bridge_device is not None
|
|
|
|
sensor_device = device_registry.async_get_device_by_identifier(
|
|
(DOMAIN, SENSOR_HARDWARE_ID), config_entry.entry_id
|
|
)
|
|
assert sensor_device is not None
|
|
assert sensor_device.via_device_id == bridge_device.id
|