Improve scan interval for Airthings Corentium Home 2

homeassistant/components/airthings_ble/const.py

@@ -7,5 +7,6 @@
 VOLUME_PICOCURIE = "pCi/L"
 
 DEFAULT_SCAN_INTERVAL = 300
+RADON_SCAN_INTERVAL = 1800
 
 MAX_RETRIES_AFTER_STARTUP = 5

homeassistant/components/airthings_ble/coordinator.py

@@ -5,7 +5,11 @@
 from datetime import timedelta
 import logging
 
-from airthings_ble import AirthingsBluetoothDeviceData, AirthingsDevice
+from airthings_ble import (
+    AirthingsBluetoothDeviceData,
+    AirthingsDevice,
+    AirthingsDeviceType,
+)
 from bleak.backends.device import BLEDevice
 from bleak_retry_connector import close_stale_connections_by_address
 
@@ -16,7 +20,7 @@
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 from homeassistant.util.unit_system import METRIC_SYSTEM
 
-from .const import DEFAULT_SCAN_INTERVAL, DOMAIN
+from .const import DEFAULT_SCAN_INTERVAL, DOMAIN, RADON_SCAN_INTERVAL
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -65,4 +69,10 @@ async def _async_update_data(self) -> AirthingsDevice:
         except Exception as err:
             raise UpdateFailed(f"Unable to fetch data: {err}") from err
 
+        if (
+            self.update_interval == timedelta(seconds=DEFAULT_SCAN_INTERVAL)
+            and data.model == AirthingsDeviceType.CORENTIUM_HOME_2
+        ):
+            self.update_interval = timedelta(seconds=RADON_SCAN_INTERVAL)
+
         return data

tests/components/airthings_ble/__init__.py

@@ -265,6 +265,24 @@ def patch_airthings_device_update():
     address="cc:cc:cc:cc:cc:cc",
 )
 
+CORENTIUM_HOME_2_DEVICE_INFO = AirthingsDevice(
+    manufacturer="Airthings AS",
+    hw_version="REV X",
+    sw_version="R-SUB-1.3.4-master+0",
+    model=AirthingsDeviceType.CORENTIUM_HOME_2,
+    name="Airthings Corentium Home 2",
+    identifier="123456",
+    sensors={
+        "connectivity_mode": "Bluetooth",
+        "battery": 90,
+        "temperature": 20.0,
+        "humidity": 55.0,
+        "radon_1day_avg": 45,
+        "radon_1day_level": "low",
+    },
+    address="cc:cc:cc:cc:cc:cc",
+)
+
 TEMPERATURE_V1 = MockEntity(
     unique_id="Airthings Wave Plus 123456_temperature",
     name="Airthings Wave Plus 123456 Temperature",

tests/components/airthings_ble/test_coordinator.py

@@ -0,0 +1,55 @@
+"""Test the Airthings BLE coordinator."""
+
+from datetime import timedelta
+from unittest.mock import patch
+
+import pytest
+
+from homeassistant.components.airthings_ble.const import (
+    DEFAULT_SCAN_INTERVAL,
+    DOMAIN,
+    RADON_SCAN_INTERVAL,
+)
+from homeassistant.components.airthings_ble.coordinator import (
+    AirthingsBLEDataUpdateCoordinator,
+)
+from homeassistant.core import HomeAssistant
+
+from . import CORENTIUM_HOME_2_DEVICE_INFO, WAVE_DEVICE_INFO, WAVE_ENHANCE_DEVICE_INFO
+
+from tests.common import MockConfigEntry
+from tests.components.bluetooth import generate_ble_device
+
+
+@pytest.mark.parametrize(
+    ("device_info", "expected_interval"),
+    [
+        (CORENTIUM_HOME_2_DEVICE_INFO, RADON_SCAN_INTERVAL),
+        (WAVE_DEVICE_INFO, DEFAULT_SCAN_INTERVAL),
+        (WAVE_ENHANCE_DEVICE_INFO, DEFAULT_SCAN_INTERVAL),
+    ],
+)
+async def test_scan_interval_adjustment(
+    hass: HomeAssistant,
+    device_info,
+    expected_interval: int,
+) -> None:
+    """Test that scan interval is adjusted based on device type."""
+    coordinator = AirthingsBLEDataUpdateCoordinator(
+        hass=hass,
+        entry=MockConfigEntry(
+            domain=DOMAIN,
+            unique_id="cc:cc:cc:cc:cc:cc",
+        ),
+    )
+    coordinator.ble_device = generate_ble_device("cc:cc:cc:cc:cc:cc", device_info.name)
+
+    assert coordinator.update_interval == timedelta(seconds=DEFAULT_SCAN_INTERVAL)
+
+    with patch(
+        "homeassistant.components.airthings_ble.coordinator.AirthingsBluetoothDeviceData.update_device",
+        return_value=device_info,
+    ):
+        await coordinator.async_refresh()
+
+    assert coordinator.update_interval == timedelta(seconds=expected_interval)