17 lines
614 B
Python
17 lines
614 B
Python
"""Helpers for Home Assistant tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from custom_components.ninlight.const import DOMAIN
|
|
from tests_unit.samples import MAC_RGB
|
|
|
|
|
|
def entity_id_for(hass: HomeAssistant, platform: str, key: str, mac: str = MAC_RGB) -> str:
|
|
"""Look entities up by unique_id instead of guessing slugs."""
|
|
entity_id = er.async_get(hass).async_get_entity_id(platform, DOMAIN, f"{mac}_{key}")
|
|
assert entity_id is not None, f"no {platform} entity with key {key}"
|
|
return entity_id
|