Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-25 05:26:47 +00:00
Code Activity

Allow removing entity registry items twice (#146519)

Browse Source
This commit is contained in:
Erik Montnemery
2025-06-11 12:42:09 +02:00
committed by GitHub
parent c49d95b230
commit 167e688139
2 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/helpers/test_entity_registry.py
View File

@@ -289,6 +289,24 @@ def test_get_or_create_suggested_object_id_conflict_existing(
assert entry.entity_id == "light.hue_1234_2"
def test_remove(entity_registry: er.EntityRegistry) -> None:
"""Test that we can remove an item."""
entry = entity_registry.async_get_or_create("light", "hue", "1234")
assert not entity_registry.deleted_entities
assert list(entity_registry.entities) == [entry.entity_id]
# Remove the item
entity_registry.async_remove(entry.entity_id)
assert list(entity_registry.deleted_entities) == [("light", "hue", "1234")]
assert not entity_registry.entities
# Remove the item again
entity_registry.async_remove(entry.entity_id)
assert list(entity_registry.deleted_entities) == [("light", "hue", "1234")]
assert not entity_registry.entities
def test_create_triggers_save(entity_registry: er.EntityRegistry) -> None:
"""Test that registering entry triggers a save."""
with patch.object(entity_registry, "async_schedule_save") as mock_schedule_save: