From 9876dd804e4f0b0034eac17c9099c4e76f68b38f Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Mon, 20 Feb 2023 07:09:56 -0600
Subject: [PATCH] Exclude event_id and event_score from being recorded in
 unifiprotect (#88387)

---
 .../components/unifiprotect/recorder.py       | 12 +++
 .../components/unifiprotect/test_recorder.py  | 91 +++++++++++++++++++
 2 files changed, 103 insertions(+)
 create mode 100644 homeassistant/components/unifiprotect/recorder.py
 create mode 100644 tests/components/unifiprotect/test_recorder.py

diff --git a/homeassistant/components/unifiprotect/recorder.py b/homeassistant/components/unifiprotect/recorder.py
new file mode 100644
index 00000000000..6603a0543f8
--- /dev/null
+++ b/homeassistant/components/unifiprotect/recorder.py
@@ -0,0 +1,12 @@
+"""Integration platform for recorder."""
+from __future__ import annotations
+
+from homeassistant.core import HomeAssistant, callback
+
+from .const import ATTR_EVENT_ID, ATTR_EVENT_SCORE
+
+
+@callback
+def exclude_attributes(hass: HomeAssistant) -> set[str]:
+    """Exclude event_id and event_score from being recorded in the database."""
+    return {ATTR_EVENT_ID, ATTR_EVENT_SCORE}
diff --git a/tests/components/unifiprotect/test_recorder.py b/tests/components/unifiprotect/test_recorder.py
new file mode 100644
index 00000000000..628f5022afd
--- /dev/null
+++ b/tests/components/unifiprotect/test_recorder.py
@@ -0,0 +1,91 @@
+"""The tests for unifiprotect recorder."""
+from __future__ import annotations
+
+from datetime import datetime, timedelta
+from unittest.mock import Mock
+
+from pyunifiprotect.data import Camera, Event, EventType
+
+from homeassistant.components.recorder import Recorder
+from homeassistant.components.recorder.db_schema import StateAttributes, States
+from homeassistant.components.recorder.util import session_scope
+from homeassistant.components.unifiprotect.binary_sensor import EVENT_SENSORS
+from homeassistant.components.unifiprotect.const import (
+    ATTR_EVENT_ID,
+    ATTR_EVENT_SCORE,
+    DEFAULT_ATTRIBUTION,
+)
+from homeassistant.const import ATTR_ATTRIBUTION, ATTR_FRIENDLY_NAME, STATE_ON, Platform
+from homeassistant.core import HomeAssistant, State
+
+from .utils import MockUFPFixture, ids_from_device_description, init_entry
+
+from tests.components.recorder.common import async_wait_recording_done
+
+
+async def test_exclude_attributes(
+    recorder_mock: Recorder,
+    hass: HomeAssistant,
+    ufp: MockUFPFixture,
+    doorbell: Camera,
+    unadopted_camera: Camera,
+    fixed_now: datetime,
+) -> None:
+    """Test binary_sensor has event_id and event_score excluded from recording."""
+
+    await init_entry(hass, ufp, [doorbell, unadopted_camera])
+
+    _, entity_id = ids_from_device_description(
+        Platform.BINARY_SENSOR, doorbell, EVENT_SENSORS[1]
+    )
+
+    event = Event(
+        id="test_event_id",
+        type=EventType.MOTION,
+        start=fixed_now - timedelta(seconds=1),
+        end=None,
+        score=100,
+        smart_detect_types=[],
+        smart_detect_event_ids=[],
+        camera_id=doorbell.id,
+    )
+
+    new_camera = doorbell.copy()
+    new_camera.is_motion_detected = True
+    new_camera.last_motion_event_id = event.id
+
+    mock_msg = Mock()
+    mock_msg.changed_data = {}
+    mock_msg.new_obj = new_camera
+
+    ufp.api.bootstrap.cameras = {new_camera.id: new_camera}
+    ufp.api.bootstrap.events = {event.id: event}
+    ufp.ws_msg(mock_msg)
+    await hass.async_block_till_done()
+
+    state = hass.states.get(entity_id)
+    assert state
+    assert state.state == STATE_ON
+    assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
+    assert state.attributes[ATTR_EVENT_SCORE] == 100
+    await async_wait_recording_done(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
+            ).outerjoin(
+                StateAttributes, States.attributes_id == StateAttributes.attributes_id
+            ):
+                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_EVENT_SCORE not in state.attributes
+        assert ATTR_EVENT_ID not in state.attributes
+        assert ATTR_FRIENDLY_NAME in state.attributes