diff --git a/homeassistant/components/wled/update.py b/homeassistant/components/wled/update.py index f0fc532b3b3..c0c099ead65 100644 --- a/homeassistant/components/wled/update.py +++ b/homeassistant/components/wled/update.py @@ -50,6 +50,11 @@ class WLEDUpdateEntity(WLEDEntity, UpdateEntity): return None return str(version) + @property + def entity_picture(self) -> str: + """Return the entity picture to use in the frontend, if any.""" + return "https://brands.home-assistant.io/wled/icon.png" + @property def latest_version(self) -> str | None: """Latest version available for install.""" diff --git a/tests/components/wled/test_update.py b/tests/components/wled/test_update.py index 452aeba6a80..e64018aba59 100644 --- a/tests/components/wled/test_update.py +++ b/tests/components/wled/test_update.py @@ -20,6 +20,7 @@ from homeassistant.components.update.const import ( from homeassistant.const import ( ATTR_DEVICE_CLASS, ATTR_ENTITY_ID, + ATTR_ENTITY_PICTURE, ATTR_ICON, ATTR_SUPPORTED_FEATURES, STATE_OFF, @@ -46,6 +47,10 @@ async def test_update_available( assert state assert state.attributes.get(ATTR_DEVICE_CLASS) == UpdateDeviceClass.FIRMWARE assert state.state == STATE_ON + assert ( + state.attributes[ATTR_ENTITY_PICTURE] + == "https://brands.home-assistant.io/wled/icon.png" + ) assert state.attributes[ATTR_INSTALLED_VERSION] == "0.8.5" assert state.attributes[ATTR_LATEST_VERSION] == "0.12.0" assert state.attributes[ATTR_RELEASE_SUMMARY] is None