From 198f4a12f08e9f9b7bb3829703a66c140f4c95b3 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Sun, 3 Apr 2022 09:17:55 -0600 Subject: [PATCH] Exclude static select attributes from being recorded in the database (#69195) --- homeassistant/components/select/recorder.py | 12 ++++++ tests/components/select/test_recorder.py | 43 +++++++++++++++++++++ 2 files changed, 55 insertions(+) create mode 100644 homeassistant/components/select/recorder.py create mode 100644 tests/components/select/test_recorder.py diff --git a/homeassistant/components/select/recorder.py b/homeassistant/components/select/recorder.py new file mode 100644 index 00000000000..6660c8383d0 --- /dev/null +++ b/homeassistant/components/select/recorder.py @@ -0,0 +1,12 @@ +"""Integration platform for recorder.""" +from __future__ import annotations + +from homeassistant.core import HomeAssistant, callback + +from . import ATTR_OPTIONS + + +@callback +def exclude_attributes(hass: HomeAssistant) -> set[str]: + """Exclude static attributes from being recorded in the database.""" + return {ATTR_OPTIONS} diff --git a/tests/components/select/test_recorder.py b/tests/components/select/test_recorder.py new file mode 100644 index 00000000000..f6366045338 --- /dev/null +++ b/tests/components/select/test_recorder.py @@ -0,0 +1,43 @@ +"""The tests for select recorder.""" +from __future__ import annotations + +from datetime import timedelta + +from homeassistant.components import select +from homeassistant.components.recorder.models import StateAttributes, States +from homeassistant.components.recorder.util import session_scope +from homeassistant.components.select import ATTR_OPTIONS +from homeassistant.const import ATTR_FRIENDLY_NAME +from homeassistant.core import State +from homeassistant.setup import async_setup_component +from homeassistant.util import dt as dt_util + +from tests.common import async_fire_time_changed, async_init_recorder_component +from tests.components.recorder.common import async_wait_recording_done_without_instance + + +async def test_exclude_attributes(hass): + """Test select registered attributes to be excluded.""" + await async_init_recorder_component(hass) + await async_setup_component( + hass, select.DOMAIN, {select.DOMAIN: {"platform": "demo"}} + ) + await hass.async_block_till_done() + async_fire_time_changed(hass, dt_util.utcnow() + timedelta(minutes=5)) + await hass.async_block_till_done() + await async_wait_recording_done_without_instance(hass) + + def _fetch_states() -> list[State]: + with session_scope(hass=hass) as session: + native_states = [] + for db_state, db_state_attributes in session.query(States, StateAttributes): + state = db_state.to_native() + state.attributes = db_state_attributes.to_native() + native_states.append(state) + return native_states + + states: list[State] = await hass.async_add_executor_job(_fetch_states) + assert len(states) > 1 + for state in states: + assert ATTR_OPTIONS not in state.attributes + assert ATTR_FRIENDLY_NAME in state.attributes