From bb0b01e4a9d378be6c46bbbb1b675980ad480955 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 24 May 2024 10:38:23 +0200 Subject: [PATCH] Add error message to snapshot_platform helper (#117974) --- tests/common.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/common.py b/tests/common.py index 33385a67d91..252e5309411 100644 --- a/tests/common.py +++ b/tests/common.py @@ -1757,5 +1757,6 @@ async def snapshot_platform( for entity_entry in entity_entries: assert entity_entry == snapshot(name=f"{entity_entry.entity_id}-entry") assert entity_entry.disabled_by is None, "Please enable all entities." - assert (state := hass.states.get(entity_entry.entity_id)) + state = hass.states.get(entity_entry.entity_id) + assert state, f"State not found for {entity_entry.entity_id}" assert state == snapshot(name=f"{entity_entry.entity_id}-state")