mirror of
https://github.com/home-assistant/core.git
synced 2026-09-08 06:31:35 +01:00
111 lines
3.4 KiB
Python
111 lines
3.4 KiB
Python
"""The tests for assist_pipeline logbook."""
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import assist_pipeline, logbook
|
|
from homeassistant.const import ATTR_DEVICE_ID
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.components.logbook.common import MockRow, mock_humanify
|
|
|
|
|
|
async def test_recording_event(
|
|
hass: HomeAssistant, init_components, device_registry: dr.DeviceRegistry
|
|
) -> None:
|
|
"""Test recording event."""
|
|
hass.config.components.add("recorder")
|
|
assert await async_setup_component(hass, "logbook", {})
|
|
await hass.async_block_till_done()
|
|
|
|
entry = MockConfigEntry()
|
|
entry.add_to_hass(hass)
|
|
satellite_device = device_registry.async_get_or_create(
|
|
config_entry_id=entry.entry_id,
|
|
connections=set(),
|
|
identifiers={("demo", "satellite-1234")},
|
|
)
|
|
|
|
device_registry.async_update_device(satellite_device.id, name="My Satellite")
|
|
event = mock_humanify(
|
|
hass,
|
|
[
|
|
MockRow(
|
|
assist_pipeline.EVENT_RECORDING,
|
|
{ATTR_DEVICE_ID: satellite_device.id},
|
|
),
|
|
],
|
|
)[0]
|
|
|
|
assert event[logbook.LOGBOOK_ENTRY_NAME] == "My Satellite"
|
|
assert event[logbook.LOGBOOK_ENTRY_DOMAIN] == assist_pipeline.DOMAIN
|
|
assert (
|
|
event[logbook.LOGBOOK_ENTRY_MESSAGE] == "My Satellite captured an audio sample"
|
|
)
|
|
|
|
|
|
@pytest.mark.usefixtures("init_components")
|
|
async def test_recording_event_child_device(
|
|
hass: HomeAssistant, device_registry: dr.DeviceRegistry
|
|
) -> None:
|
|
"""Test recording event fired for a child device."""
|
|
hass.config.components.add("recorder")
|
|
assert await async_setup_component(hass, "logbook", {})
|
|
await hass.async_block_till_done()
|
|
|
|
entry = MockConfigEntry()
|
|
entry.add_to_hass(hass)
|
|
satellite_device = device_registry.async_get_or_create(
|
|
config_entry_id=entry.entry_id,
|
|
connections=set(),
|
|
identifiers={("demo", "satellite-1234")},
|
|
)
|
|
child_device = device_registry.async_get_or_create_child(
|
|
config_entry_id=entry.entry_id,
|
|
identifiers={("demo", "satellite-1234-child")},
|
|
parent_device_id=satellite_device.id,
|
|
name="My Child Satellite",
|
|
)
|
|
|
|
event = mock_humanify(
|
|
hass,
|
|
[
|
|
MockRow(
|
|
assist_pipeline.EVENT_RECORDING,
|
|
{ATTR_DEVICE_ID: child_device.id},
|
|
),
|
|
],
|
|
)[0]
|
|
|
|
assert event[logbook.LOGBOOK_ENTRY_NAME] == "My Child Satellite"
|
|
assert (
|
|
event[logbook.LOGBOOK_ENTRY_MESSAGE]
|
|
== "My Child Satellite captured an audio sample"
|
|
)
|
|
|
|
|
|
@pytest.mark.usefixtures("init_components")
|
|
async def test_recording_event_unknown_device(hass: HomeAssistant) -> None:
|
|
"""Test recording event for an unknown device does not raise."""
|
|
hass.config.components.add("recorder")
|
|
assert await async_setup_component(hass, "logbook", {})
|
|
await hass.async_block_till_done()
|
|
|
|
event = mock_humanify(
|
|
hass,
|
|
[
|
|
MockRow(
|
|
assist_pipeline.EVENT_RECORDING,
|
|
{ATTR_DEVICE_ID: "non-existing-device"},
|
|
),
|
|
],
|
|
)[0]
|
|
|
|
assert event[logbook.LOGBOOK_ENTRY_NAME] == "Unknown device"
|
|
assert (
|
|
event[logbook.LOGBOOK_ENTRY_MESSAGE]
|
|
== "Unknown device captured an audio sample"
|
|
)
|