WIP

Author: andrew-codechimp

custom_components/label_state/binary_sensor.py

@@ -6,16 +6,19 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
     CONF_NAME,
+    CONF_UNIQUE_ID,
     STATE_UNAVAILABLE,
     STATE_UNKNOWN,
 )
 from homeassistant.core import Event, EventStateChangedData, HomeAssistant, callback
 from homeassistant.helpers import entity_registry as er
 from homeassistant.helpers.entity_platform import (
     AddConfigEntryEntitiesCallback,
+    AddEntitiesCallback,
 )
 from homeassistant.helpers.entity_registry import EVENT_ENTITY_REGISTRY_UPDATED
 from homeassistant.helpers.event import async_track_state_change_event
+from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, StateType
 
 from .const import (
     ATTR_ENTITIES,
@@ -69,6 +72,37 @@ async def async_setup_entry(
     )
 
 
+async def async_setup_platform(
+    hass: HomeAssistant,
+    config: ConfigType,
+    async_add_entities: AddEntitiesCallback,
+    discovery_info: DiscoveryInfoType | None = None,
+) -> None:
+    """Set up the min/max/mean sensor."""
+    label: str = config[CONF_LABEL]
+    name: str | None = config.get(CONF_NAME)
+    state_type: str = config[CONF_STATE_TYPE]
+    state_to: str | None = config.get(CONF_STATE_TO)
+    state_lower_limit: float | None = config.get(CONF_STATE_LOWER_LIMIT)
+    state_upper_limit: float | None = config.get(CONF_STATE_UPPER_LIMIT)
+    unique_id = config.get(CONF_UNIQUE_ID)
+
+    async_add_entities(
+        [
+            LabelStateBinarySensor(
+                hass,
+                label,
+                name,
+                state_type,
+                state_to,
+                state_lower_limit,
+                state_upper_limit,
+                unique_id,
+            )
+        ]
+    )
+
+
 class LabelStateBinarySensor(BinarySensorEntity):
     """Representation of a Label State sensor."""
 

tests/test_binary_sensor.py

@@ -1,6 +1,14 @@
 """The test for the label_state binary sensor platform."""
 
 import pytest
+from homeassistant.const import (
+    CONF_ENTITY_ID,
+    CONF_NAME,
+    CONF_PLATFORM,
+    CONF_UNIQUE_ID,
+    EVENT_HOMEASSISTANT_STARTED,
+    STATE_ON,
+)
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
 from homeassistant.helpers import label_registry as lr
@@ -117,38 +125,37 @@ async def test_numeric_state_sensor(
     """Test the numeric state sensor."""
 
     test_label = label_registry.async_create(
-        "test",
+        "test_numeric_state",
     )
 
     sensor1_entity_entry = entity_registry.async_get_or_create(
         "sensor", "test_1", "unique", suggested_object_id="test_1"
     )
     sensor1_entity_entry.labels.add(test_label.label_id)
+    await hass.async_block_till_done()
     assert sensor1_entity_entry.entity_id == "sensor.test_1"
     assert test_label.label_id in sensor1_entity_entry.labels
 
     sensor2_entity_entry = entity_registry.async_get_or_create(
         "sensor", "test_2", "unique", suggested_object_id="test_2"
     )
     sensor2_entity_entry.labels.add(test_label.label_id)
+    await hass.async_block_till_done()
     assert sensor2_entity_entry.entity_id == "sensor.test_2"
+    assert test_label.label_id in sensor2_entity_entry.labels
 
-    config = MockConfigEntry(
-        domain="label_state",
-        data={},
-        options={
+    config = {
+        "binary_sensor": {
+            "platform": "label_state",
             "name": "test_numeric_state",
             "label": test_label.label_id,
             "state_type": "numeric_state",
             "state_lower_limit": state_lower_limit,
             "state_upper_limit": state_upper_limit,
-        },
-        title="test_numeric_state",
-    )
+        }
+    }
 
-    # config.add_to_hass(hass)
-    await setup_integration(hass, config)
-    # assert await hass.config_entries.async_setup(config.entry_id)
+    assert await async_setup_component(hass, "binary_sensor", config)
     await hass.async_block_till_done()
 
     hass.states.async_set(sensor1_entity_entry.entity_id, state_1)
@@ -157,6 +164,10 @@ async def test_numeric_state_sensor(
     hass.states.async_set(sensor2_entity_entry.entity_id, state_2)
     await hass.async_block_till_done()
 
+    state1 = hass.states.get(sensor1_entity_entry.entity_id)
+    assert state1 is not None
+    assert state1.state == state_1
+
     state = hass.states.get("binary_sensor.test_numeric_state")
 
     assert state is not None