From 214d14b06b7dfef7ffdf5099b5705d39ae764353 Mon Sep 17 00:00:00 2001
From: Luke Lashley <conway220@gmail.com>
Date: Sun, 16 Mar 2025 11:57:21 -0400
Subject: [PATCH] Add binary sensor to Snoo (#140729)

* Add binary sensor

* Update homeassistant/components/snoo/binary_sensor.py

Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>

---------

Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
---
 homeassistant/components/snoo/__init__.py     |  2 +-
 .../components/snoo/binary_sensor.py          | 70 +++++++++++++++++++
 homeassistant/components/snoo/strings.json    |  9 +++
 tests/components/snoo/test_binary_sensor.py   | 30 ++++++++
 4 files changed, 110 insertions(+), 1 deletion(-)
 create mode 100644 homeassistant/components/snoo/binary_sensor.py
 create mode 100644 tests/components/snoo/test_binary_sensor.py

diff --git a/homeassistant/components/snoo/__init__.py b/homeassistant/components/snoo/__init__.py
index ca561a52a3f..23b5d5201db 100644
--- a/homeassistant/components/snoo/__init__.py
+++ b/homeassistant/components/snoo/__init__.py
@@ -17,7 +17,7 @@ from .coordinator import SnooConfigEntry, SnooCoordinator
 
 _LOGGER = logging.getLogger(__name__)
 
-PLATFORMS: list[Platform] = [Platform.SELECT, Platform.SENSOR]
+PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SELECT, Platform.SENSOR]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: SnooConfigEntry) -> bool:
diff --git a/homeassistant/components/snoo/binary_sensor.py b/homeassistant/components/snoo/binary_sensor.py
new file mode 100644
index 00000000000..3c91db5b86d
--- /dev/null
+++ b/homeassistant/components/snoo/binary_sensor.py
@@ -0,0 +1,70 @@
+"""Support for Snoo Binary Sensors."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from python_snoo.containers import SnooData
+
+from homeassistant.components.binary_sensor import (
+    BinarySensorDeviceClass,
+    BinarySensorEntity,
+    BinarySensorEntityDescription,
+    EntityCategory,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import SnooConfigEntry
+from .entity import SnooDescriptionEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SnooBinarySensorEntityDescription(BinarySensorEntityDescription):
+    """Describes a Snoo Binary Sensor."""
+
+    value_fn: Callable[[SnooData], bool]
+
+
+BINARY_SENSOR_DESCRIPTIONS: list[SnooBinarySensorEntityDescription] = [
+    SnooBinarySensorEntityDescription(
+        key="left_clip",
+        translation_key="left_clip",
+        value_fn=lambda data: data.left_safety_clip,
+        device_class=BinarySensorDeviceClass.CONNECTIVITY,
+        entity_category=EntityCategory.DIAGNOSTIC,
+    ),
+    SnooBinarySensorEntityDescription(
+        key="right_clip",
+        translation_key="right_clip",
+        value_fn=lambda data: data.left_safety_clip,
+        device_class=BinarySensorDeviceClass.CONNECTIVITY,
+        entity_category=EntityCategory.DIAGNOSTIC,
+    ),
+]
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: SnooConfigEntry,
+    async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+    """Set up Snoo device."""
+    coordinators = entry.runtime_data
+    async_add_entities(
+        SnooBinarySensor(coordinator, description)
+        for coordinator in coordinators.values()
+        for description in BINARY_SENSOR_DESCRIPTIONS
+    )
+
+
+class SnooBinarySensor(SnooDescriptionEntity, BinarySensorEntity):
+    """A Binary sensor using Snoo coordinator."""
+
+    entity_description: SnooBinarySensorEntityDescription
+
+    @property
+    def is_on(self) -> bool:
+        """Return true if the binary sensor is on."""
+        return self.entity_description.value_fn(self.coordinator.data)
diff --git a/homeassistant/components/snoo/strings.json b/homeassistant/components/snoo/strings.json
index 47e59603a14..8211480f771 100644
--- a/homeassistant/components/snoo/strings.json
+++ b/homeassistant/components/snoo/strings.json
@@ -27,6 +27,15 @@
     }
   },
   "entity": {
+    "binary_sensor": {
+      "left_clip": {
+        "name": "Left safety clip"
+      },
+      "right_clip": {
+        "name": "Right safety clip"
+      }
+    },
+
     "sensor": {
       "state": {
         "name": "State",
diff --git a/tests/components/snoo/test_binary_sensor.py b/tests/components/snoo/test_binary_sensor.py
new file mode 100644
index 00000000000..77b2e36c1fe
--- /dev/null
+++ b/tests/components/snoo/test_binary_sensor.py
@@ -0,0 +1,30 @@
+"""Test Snoo Binary Sensors."""
+
+from unittest.mock import AsyncMock
+
+from homeassistant.const import STATE_ON, STATE_UNAVAILABLE
+from homeassistant.core import HomeAssistant
+
+from . import async_init_integration, find_update_callback
+from .const import MOCK_SNOO_DATA
+
+
+async def test_binary_sensors(hass: HomeAssistant, bypass_api: AsyncMock) -> None:
+    """Test binary sensors and check test values are correctly set."""
+    await async_init_integration(hass)
+    assert len(hass.states.async_all("binary_sensor")) == 2
+    assert (
+        hass.states.get("binary_sensor.test_snoo_left_safety_clip").state
+        == STATE_UNAVAILABLE
+    )
+    assert (
+        hass.states.get("binary_sensor.test_snoo_right_safety_clip").state
+        == STATE_UNAVAILABLE
+    )
+    find_update_callback(bypass_api, "random_num")(MOCK_SNOO_DATA)
+    await hass.async_block_till_done()
+    assert len(hass.states.async_all("binary_sensor")) == 2
+    assert hass.states.get("binary_sensor.test_snoo_left_safety_clip").state == STATE_ON
+    assert (
+        hass.states.get("binary_sensor.test_snoo_right_safety_clip").state == STATE_ON
+    )