mirror of
https://github.com/home-assistant/core.git
synced 2025-07-17 02:07:09 +00:00
Add binary sensor platform to WLED (#59798)
This commit is contained in:
parent
65d1f8183f
commit
d95c615f86
@ -1,6 +1,7 @@
|
|||||||
"""Support for WLED."""
|
"""Support for WLED."""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from homeassistant.components.binary_sensor import DOMAIN as BINARY_SENSOR_DOMAIN
|
||||||
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN
|
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN
|
||||||
from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN
|
from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN
|
||||||
from homeassistant.components.number import DOMAIN as NUMBER_DOMAIN
|
from homeassistant.components.number import DOMAIN as NUMBER_DOMAIN
|
||||||
@ -14,6 +15,7 @@ from .const import DOMAIN
|
|||||||
from .coordinator import WLEDDataUpdateCoordinator
|
from .coordinator import WLEDDataUpdateCoordinator
|
||||||
|
|
||||||
PLATFORMS = (
|
PLATFORMS = (
|
||||||
|
BINARY_SENSOR_DOMAIN,
|
||||||
BUTTON_DOMAIN,
|
BUTTON_DOMAIN,
|
||||||
LIGHT_DOMAIN,
|
LIGHT_DOMAIN,
|
||||||
SELECT_DOMAIN,
|
SELECT_DOMAIN,
|
||||||
|
58
homeassistant/components/wled/binary_sensor.py
Normal file
58
homeassistant/components/wled/binary_sensor.py
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
"""Support for WLED binary sensor."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from homeassistant.components.binary_sensor import (
|
||||||
|
DEVICE_CLASS_UPDATE,
|
||||||
|
BinarySensorEntity,
|
||||||
|
)
|
||||||
|
from homeassistant.config_entries import ConfigEntry
|
||||||
|
from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
|
from .const import DOMAIN
|
||||||
|
from .coordinator import WLEDDataUpdateCoordinator
|
||||||
|
from .models import WLEDEntity
|
||||||
|
|
||||||
|
|
||||||
|
async def async_setup_entry(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
entry: ConfigEntry,
|
||||||
|
async_add_entities: AddEntitiesCallback,
|
||||||
|
) -> None:
|
||||||
|
"""Set up a WLED binary sensor based on a config entry."""
|
||||||
|
coordinator: WLEDDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
|
||||||
|
async_add_entities(
|
||||||
|
[
|
||||||
|
WLEDUpdateBinarySensor(coordinator),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class WLEDUpdateBinarySensor(WLEDEntity, BinarySensorEntity):
|
||||||
|
"""Defines a WLED firmware binary sensor."""
|
||||||
|
|
||||||
|
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
|
||||||
|
_attr_device_class = DEVICE_CLASS_UPDATE
|
||||||
|
|
||||||
|
def __init__(self, coordinator: WLEDDataUpdateCoordinator) -> None:
|
||||||
|
"""Initialize the button entity."""
|
||||||
|
super().__init__(coordinator=coordinator)
|
||||||
|
self._attr_name = f"{coordinator.data.info.name} Firmware"
|
||||||
|
self._attr_unique_id = f"{coordinator.data.info.mac_address}_update"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_on(self) -> bool:
|
||||||
|
"""Return the state of the sensor."""
|
||||||
|
current = self.coordinator.data.info.version
|
||||||
|
beta = self.coordinator.data.info.version_latest_beta
|
||||||
|
stable = self.coordinator.data.info.version_latest_stable
|
||||||
|
|
||||||
|
return current is not None and (
|
||||||
|
(stable is not None and stable > current)
|
||||||
|
or (
|
||||||
|
beta is not None
|
||||||
|
and (current.alpha or current.beta or current.release_candidate)
|
||||||
|
and beta > current
|
||||||
|
)
|
||||||
|
)
|
54
tests/components/wled/test_binary_sensor.py
Normal file
54
tests/components/wled/test_binary_sensor.py
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
"""Tests for the WLED binary sensor platform."""
|
||||||
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from homeassistant.components.binary_sensor import DEVICE_CLASS_UPDATE
|
||||||
|
from homeassistant.const import (
|
||||||
|
ATTR_DEVICE_CLASS,
|
||||||
|
ATTR_ICON,
|
||||||
|
ENTITY_CATEGORY_DIAGNOSTIC,
|
||||||
|
STATE_OFF,
|
||||||
|
STATE_ON,
|
||||||
|
)
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.helpers import entity_registry as er
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry
|
||||||
|
|
||||||
|
|
||||||
|
async def test_update_available(
|
||||||
|
hass: HomeAssistant, init_integration: MockConfigEntry, mock_wled: MagicMock
|
||||||
|
) -> None:
|
||||||
|
"""Test the firmware update binary sensor."""
|
||||||
|
entity_registry = er.async_get(hass)
|
||||||
|
|
||||||
|
state = hass.states.get("binary_sensor.wled_rgb_light_firmware")
|
||||||
|
assert state
|
||||||
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_UPDATE
|
||||||
|
assert state.state == STATE_ON
|
||||||
|
assert ATTR_ICON not in state.attributes
|
||||||
|
|
||||||
|
entry = entity_registry.async_get("binary_sensor.wled_rgb_light_firmware")
|
||||||
|
assert entry
|
||||||
|
assert entry.unique_id == "aabbccddeeff_update"
|
||||||
|
assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("mock_wled", ["wled/rgb_websocket.json"], indirect=True)
|
||||||
|
async def test_no_update_available(
|
||||||
|
hass: HomeAssistant, init_integration: MockConfigEntry, mock_wled: MagicMock
|
||||||
|
) -> None:
|
||||||
|
"""Test the update binary sensor. There is no update available."""
|
||||||
|
entity_registry = er.async_get(hass)
|
||||||
|
|
||||||
|
state = hass.states.get("binary_sensor.wled_websocket_firmware")
|
||||||
|
assert state
|
||||||
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_UPDATE
|
||||||
|
assert state.state == STATE_OFF
|
||||||
|
assert ATTR_ICON not in state.attributes
|
||||||
|
|
||||||
|
entry = entity_registry.async_get("binary_sensor.wled_websocket_firmware")
|
||||||
|
assert entry
|
||||||
|
assert entry.unique_id == "aabbccddeeff_update"
|
||||||
|
assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC
|
Loading…
x
Reference in New Issue
Block a user