108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
"""The tests for hue logbook."""
|
|
from homeassistant.components.hue.const import ATTR_HUE_EVENT, CONF_SUBTYPE, DOMAIN
|
|
from homeassistant.components.hue.v1.hue_event import CONF_LAST_UPDATED
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
CONF_DEVICE_ID,
|
|
CONF_EVENT,
|
|
CONF_ID,
|
|
CONF_TYPE,
|
|
CONF_UNIQUE_ID,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .conftest import setup_platform
|
|
|
|
from tests.components.logbook.common import MockRow, mock_humanify
|
|
|
|
# v1 event
|
|
SAMPLE_V1_EVENT = {
|
|
CONF_DEVICE_ID: "fe346f17a9f8c15be633f9cc3f3d6631",
|
|
CONF_EVENT: 18,
|
|
CONF_ID: "hue_tap",
|
|
CONF_LAST_UPDATED: "2019-12-28T22:58:03",
|
|
CONF_UNIQUE_ID: "00:00:00:00:00:44:23:08-f2",
|
|
}
|
|
# v2 event
|
|
SAMPLE_V2_EVENT = {
|
|
CONF_DEVICE_ID: "f974028e7933aea703a2199a855bc4a3",
|
|
CONF_ID: "wall_switch_with_2_controls_button",
|
|
CONF_SUBTYPE: 1,
|
|
CONF_TYPE: "initial_press",
|
|
CONF_UNIQUE_ID: "c658d3d8-a013-4b81-8ac6-78b248537e70",
|
|
}
|
|
|
|
|
|
async def test_humanify_hue_events(
|
|
hass: HomeAssistant, mock_bridge_v2, device_registry: dr.DeviceRegistry
|
|
) -> None:
|
|
"""Test hue events when the devices are present in the registry."""
|
|
await setup_platform(hass, mock_bridge_v2, "sensor")
|
|
hass.config.components.add("recorder")
|
|
assert await async_setup_component(hass, "logbook", {})
|
|
await hass.async_block_till_done()
|
|
entry: ConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
|
|
|
|
v1_device = device_registry.async_get_or_create(
|
|
identifiers={(DOMAIN, "v1")}, name="Remote 1", config_entry_id=entry.entry_id
|
|
)
|
|
v2_device = device_registry.async_get_or_create(
|
|
identifiers={(DOMAIN, "v2")}, name="Remote 2", config_entry_id=entry.entry_id
|
|
)
|
|
|
|
(v1_event, v2_event) = mock_humanify(
|
|
hass,
|
|
[
|
|
MockRow(
|
|
ATTR_HUE_EVENT,
|
|
{**SAMPLE_V1_EVENT, CONF_DEVICE_ID: v1_device.id},
|
|
),
|
|
MockRow(
|
|
ATTR_HUE_EVENT,
|
|
{**SAMPLE_V2_EVENT, CONF_DEVICE_ID: v2_device.id},
|
|
),
|
|
],
|
|
)
|
|
|
|
assert v1_event["name"] == "Remote 1"
|
|
assert v1_event["domain"] == DOMAIN
|
|
assert v1_event["message"] == "Event 18"
|
|
|
|
assert v2_event["name"] == "Remote 2"
|
|
assert v2_event["domain"] == DOMAIN
|
|
assert v2_event["message"] == "first button pressed initially"
|
|
|
|
|
|
async def test_humanify_hue_events_devices_removed(
|
|
hass: HomeAssistant, mock_bridge_v2
|
|
) -> None:
|
|
"""Test hue events when the devices have been removed from the registry."""
|
|
await setup_platform(hass, mock_bridge_v2, "sensor")
|
|
hass.config.components.add("recorder")
|
|
assert await async_setup_component(hass, "logbook", {})
|
|
await hass.async_block_till_done()
|
|
|
|
(v1_event, v2_event) = mock_humanify(
|
|
hass,
|
|
[
|
|
MockRow(
|
|
ATTR_HUE_EVENT,
|
|
SAMPLE_V1_EVENT,
|
|
),
|
|
MockRow(
|
|
ATTR_HUE_EVENT,
|
|
SAMPLE_V2_EVENT,
|
|
),
|
|
],
|
|
)
|
|
|
|
assert v1_event["name"] == "hue_tap"
|
|
assert v1_event["domain"] == DOMAIN
|
|
assert v1_event["message"] == "Event 18"
|
|
|
|
assert v2_event["name"] == "wall_switch_with_2_controls_button"
|
|
assert v2_event["domain"] == DOMAIN
|
|
assert v2_event["message"] == "first button pressed initially"
|