Add Rabbit Air air quality sensor

homeassistant/components/rabbitair/__init__.py

@@ -8,7 +8,7 @@
 
 from .coordinator import RabbitAirConfigEntry, RabbitAirDataUpdateCoordinator
 
-PLATFORMS: list[Platform] = [Platform.FAN]
+PLATFORMS: list[Platform] = [Platform.FAN, Platform.SENSOR]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: RabbitAirConfigEntry) -> bool:

homeassistant/components/rabbitair/entity.py

@@ -32,7 +32,6 @@ def __init__(
     ) -> None:
         """Initialize the entity."""
         super().__init__(coordinator)
-        self._attr_name = entry.title
         self._attr_unique_id = entry.unique_id
         self._attr_device_info = DeviceInfo(
             identifiers={(DOMAIN, entry.data[CONF_MAC])},

homeassistant/components/rabbitair/fan.py

@@ -61,6 +61,7 @@ def __init__(
     ) -> None:
         """Initialize the entity."""
         super().__init__(coordinator, entry)
+        self._attr_name = entry.title
 
         if self._is_model(Model.MinusA2):
             self._attr_preset_modes = list(PRESET_MODES)

homeassistant/components/rabbitair/icons.json

@@ -12,6 +12,11 @@
           }
         }
       }
+    },
+    "sensor": {
+      "air_quality": {
+        "default": "mdi:air-filter"
+      }
     }
   }
 }

homeassistant/components/rabbitair/sensor.py

@@ -0,0 +1,63 @@
+"""Support for Rabbit Air sensors."""
+
+from rabbitair import Quality
+
+from homeassistant.components.sensor import (
+    SensorDeviceClass,
+    SensorEntity,
+    SensorEntityDescription,
+)
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.typing import StateType
+
+from .coordinator import RabbitAirConfigEntry, RabbitAirDataUpdateCoordinator
+from .entity import RabbitAirBaseEntity
+
+
+def _quality_value(quality: Quality | None) -> StateType:
+    """Return the air quality state."""
+    return None if quality is None else quality.name.lower()
+
+
+AIR_QUALITY_OPTIONS = [quality.name.lower() for quality in Quality]
+
+AIR_QUALITY_DESCRIPTION = SensorEntityDescription(
+    key="air_quality",
+    translation_key="air_quality",
+    device_class=SensorDeviceClass.ENUM,
+    options=AIR_QUALITY_OPTIONS,
+)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: RabbitAirConfigEntry,
+    async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+    """Set up Rabbit Air sensors."""
+    if entry.runtime_data.data.quality is not None:
+        async_add_entities([RabbitAirAirQualitySensor(entry.runtime_data, entry)])
+
+
+class RabbitAirAirQualitySensor(RabbitAirBaseEntity, SensorEntity):
+    """Rabbit Air air quality sensor."""
+
+    entity_description = AIR_QUALITY_DESCRIPTION
+    _attr_has_entity_name = True
+
+    def __init__(
+        self,
+        coordinator: RabbitAirDataUpdateCoordinator,
+        entry: RabbitAirConfigEntry,
+    ) -> None:
+        """Initialize the entity."""
+        super().__init__(coordinator, entry)
+        self._attr_unique_id = f"{entry.unique_id}_{self.entity_description.key}"
+        self._attr_native_value = _quality_value(coordinator.data.quality)
+
+    @callback
+    def _handle_coordinator_update(self) -> None:
+        """Handle updated data from the coordinator."""
+        self._attr_native_value = _quality_value(self.coordinator.data.quality)
+        super()._handle_coordinator_update()

homeassistant/components/rabbitair/strings.json

@@ -32,6 +32,18 @@
           }
         }
       }
+    },
+    "sensor": {
+      "air_quality": {
+        "name": "Air quality",
+        "state": {
+          "high": "[%key:common::state::high%]",
+          "highest": "Highest",
+          "low": "[%key:common::state::low%]",
+          "lowest": "Lowest",
+          "medium": "[%key:common::state::medium%]"
+        }
+      }
     }
   }
 }

tests/components/rabbitair/test_config_flow.py

@@ -5,7 +5,7 @@
 from unittest.mock import MagicMock, Mock, patch
 
 import pytest
-from rabbitair import Mode, Model, Speed
+from rabbitair import Mode, Model, Quality, Speed
 
 from homeassistant import config_entries
 from homeassistant.components.rabbitair.const import DOMAIN
@@ -62,6 +62,7 @@ def get_mock_state(
     main_firmware: str | None = TEST_HARDWARE,
     power: bool | None = True,
     mode: Mode | None = Mode.Auto,
+    quality: Quality | None = Quality.High,
     speed: Speed | None = Speed.Low,
     wifi_firmware: str | None = TEST_FIRMWARE,
 ) -> Mock:
@@ -71,6 +72,7 @@ def get_mock_state(
     mock_state.main_firmware = main_firmware
     mock_state.power = power
     mock_state.mode = mode
+    mock_state.quality = quality
     mock_state.speed = speed
     mock_state.wifi_firmware = wifi_firmware
     return mock_state

tests/components/rabbitair/test_sensor.py

@@ -0,0 +1,80 @@
+"""Test the Rabbit Air sensor platform."""
+
+from unittest.mock import patch
+
+import pytest
+from rabbitair import Quality
+
+from homeassistant.components.rabbitair.const import DOMAIN
+from homeassistant.const import CONF_ACCESS_TOKEN, CONF_HOST, CONF_MAC
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import entity_registry as er
+
+from .test_config_flow import (
+    TEST_HOST,
+    TEST_MAC,
+    TEST_TITLE,
+    TEST_TOKEN,
+    TEST_UNIQUE_ID,
+    get_mock_state,
+)
+
+from tests.common import MockConfigEntry
+
+
+@pytest.mark.usefixtures("mock_async_zeroconf")
+async def test_air_quality_sensor(
+    hass: HomeAssistant, entity_registry: er.EntityRegistry
+) -> None:
+    """Test the air quality sensor."""
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        data={
+            CONF_HOST: TEST_HOST,
+            CONF_ACCESS_TOKEN: TEST_TOKEN,
+            CONF_MAC: TEST_MAC,
+        },
+        title=TEST_TITLE,
+        unique_id=TEST_UNIQUE_ID,
+    )
+    entry.add_to_hass(hass)
+
+    with patch(
+        "rabbitair.UdpClient.get_state",
+        return_value=get_mock_state(quality=Quality.High),
+    ):
+        assert await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+    state = hass.states.get("sensor.rabbit_air_air_quality")
+    assert state
+    assert state.state == "high"
+
+    registry_entry = entity_registry.async_get("sensor.rabbit_air_air_quality")
+    assert registry_entry
+    assert registry_entry.unique_id == f"{TEST_UNIQUE_ID}_air_quality"
+
+
+@pytest.mark.usefixtures("mock_async_zeroconf")
+async def test_no_air_quality_sensor_when_quality_is_none(hass: HomeAssistant) -> None:
+    """Test the air quality sensor is not created when quality is unavailable."""
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        data={
+            CONF_HOST: TEST_HOST,
+            CONF_ACCESS_TOKEN: TEST_TOKEN,
+            CONF_MAC: TEST_MAC,
+        },
+        title=TEST_TITLE,
+        unique_id=TEST_UNIQUE_ID,
+    )
+    entry.add_to_hass(hass)
+
+    with patch(
+        "rabbitair.UdpClient.get_state",
+        return_value=get_mock_state(quality=None),
+    ):
+        assert await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+    assert hass.states.get("sensor.rabbit_air_air_quality") is None