jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix entity registry not saving name/icon (#31932)

Browse Source
This commit is contained in:
Paulus Schoutsen 2020-02-18 08:32:34 -08:00 committed by GitHub
parent 0ae86b022d
commit 7be3a4cd37
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
homeassistant/helpers/entity_registry.py
View File

@ -392,11 +392,14 @@ class EntityRegistry:
unique_id=entity["unique_id"], unique_id=entity["unique_id"],
platform=entity["platform"], platform=entity["platform"],
name=entity.get("name"), name=entity.get("name"),
icon=entity.get("icon"),
disabled_by=entity.get("disabled_by"), disabled_by=entity.get("disabled_by"),
capabilities=entity.get("capabilities") or {}, capabilities=entity.get("capabilities") or {},
supported_features=entity.get("supported_features", 0), supported_features=entity.get("supported_features", 0),
device_class=entity.get("device_class"), device_class=entity.get("device_class"),
unit_of_measurement=entity.get("unit_of_measurement"), unit_of_measurement=entity.get("unit_of_measurement"),
original_name=entity.get("original_name"),
original_icon=entity.get("original_icon"),
) )
self.entities = entities self.entities = entities
@ -419,11 +422,14 @@ class EntityRegistry:
"unique_id": entry.unique_id, "unique_id": entry.unique_id,
"platform": entry.platform, "platform": entry.platform,
"name": entry.name, "name": entry.name,
"icon": entry.icon,
"disabled_by": entry.disabled_by, "disabled_by": entry.disabled_by,
"capabilities": entry.capabilities, "capabilities": entry.capabilities,
"supported_features": entry.supported_features, "supported_features": entry.supported_features,
"device_class": entry.device_class, "device_class": entry.device_class,
"unit_of_measurement": entry.unit_of_measurement, "unit_of_measurement": entry.unit_of_measurement,
"original_name": entry.original_name,
"original_icon": entry.original_icon,
} }
for entry in self.entities.values() for entry in self.entities.values()
] ]

9
tests/helpers/test_entity_registry.py
View File

@ -159,6 +159,11 @@ async def test_loading_saving_data(hass, registry):
supported_features=5, supported_features=5,
device_class="mock-device-class", device_class="mock-device-class",
disabled_by=entity_registry.DISABLED_HASS, disabled_by=entity_registry.DISABLED_HASS,
original_name="Original Name",
original_icon="hass:original-icon",
)
orig_entry2 = registry.async_update_entity(
orig_entry2.entity_id, name="User Name", icon="hass:user-icon"
) )
assert len(registry.entities) == 2 assert len(registry.entities) == 2
@ -181,6 +186,10 @@ async def test_loading_saving_data(hass, registry):
assert new_entry2.capabilities == {"max": 100} assert new_entry2.capabilities == {"max": 100}
assert new_entry2.supported_features == 5 assert new_entry2.supported_features == 5
assert new_entry2.device_class == "mock-device-class" assert new_entry2.device_class == "mock-device-class"
assert new_entry2.name == "User Name"
assert new_entry2.icon == "hass:user-icon"
assert new_entry2.original_name == "Original Name"
assert new_entry2.original_icon == "hass:original-icon"
def test_generate_entity_considers_registered_entities(registry): def test_generate_entity_considers_registered_entities(registry):