core/tests/components/hue/test_logbook.py

105 lines
3.3 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.helpers import device_registry
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, mock_bridge_v2):
"""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]
dev_reg = device_registry.async_get(hass)
v1_device = dev_reg.async_get_or_create(
identifiers={(DOMAIN, "v1")}, name="Remote 1", config_entry_id=entry.entry_id
)
v2_device = dev_reg.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, mock_bridge_v2):
"""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"