Add support to entity registry for overriding device_class (#59985)

This commit is contained in:
Erik Montnemery
2021-11-22 17:38:06 +01:00
committed by GitHub
parent 67e13b35db
commit 49a27e12ad
31 changed files with 136 additions and 69 deletions

View File

@@ -78,7 +78,7 @@ async def test_sensors_pro(hass, canary) -> None:
for (sensor_id, data) in sensors.items():
entity_entry = registry.async_get(f"sensor.{sensor_id}")
assert entity_entry
assert entity_entry.device_class == data[3]
assert entity_entry.original_device_class == data[3]
assert entity_entry.unique_id == data[0]
assert entity_entry.original_icon == data[4]
@@ -197,7 +197,7 @@ async def test_sensors_flex(hass, canary) -> None:
for (sensor_id, data) in sensors.items():
entity_entry = registry.async_get(f"sensor.{sensor_id}")
assert entity_entry
assert entity_entry.device_class == data[3]
assert entity_entry.original_device_class == data[3]
assert entity_entry.unique_id == data[0]
assert entity_entry.original_icon == data[4]