evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Erik Montnemery 2021-11-22 17:38:06 +01:00 • committed by GitHub
parent 67e13b35db
commit 49a27e12ad
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
31 changed files with 136 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/components/nzbget/test_sensor.py
View file

@ -45,7 +45,7 @@ async def test_sensors(hass, nzbget_api) -> None:
for (sensor_id, data) in sensors.items():
entity_entry = registry.async_get(f"sensor.nzbgettest_{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 == f"{entry.entry_id}_{data[0]}"
state = hass.states.get(f"sensor.nzbgettest_{sensor_id}")