diff --git a/homeassistant/components/airgradient/const.py b/homeassistant/components/airgradient/const.py index 23e45576f0408d..69c7ae79758965 100644 --- a/homeassistant/components/airgradient/const.py +++ b/homeassistant/components/airgradient/const.py @@ -71,6 +71,11 @@ class ModelCapabilities: } ) +OUTDOOR_CAPABILITIES = ModelCapabilities( + config=COMMON_LEGACY_CONFIG, + actions=frozenset({CO2_CALIBRATION}), +) + MODEL_CAPABILITIES: tuple[tuple[str, ModelCapabilities], ...] = ( ( "I-9PSL", @@ -86,13 +91,8 @@ class ModelCapabilities: actions=frozenset({CO2_CALIBRATION, LED_BAR_TEST}), ), ), - ( - "O-1", - ModelCapabilities( - config=COMMON_LEGACY_CONFIG, - actions=frozenset({CO2_CALIBRATION}), - ), - ), + ("O-1", OUTDOOR_CAPABILITIES), + ("0-1PS", OUTDOOR_CAPABILITIES), ( "DIY", ModelCapabilities( diff --git a/tests/components/airgradient/test_init.py b/tests/components/airgradient/test_init.py index 575792175d7ee4..3d80003dee85ac 100644 --- a/tests/components/airgradient/test_init.py +++ b/tests/components/airgradient/test_init.py @@ -8,7 +8,7 @@ import pytest from syrupy.assertion import SnapshotAssertion -from homeassistant.components.airgradient.const import DOMAIN +from homeassistant.components.airgradient.const import DOMAIN, get_model_capabilities from homeassistant.config_entries import ConfigEntryState from homeassistant.const import Platform from homeassistant.core import HomeAssistant @@ -19,6 +19,11 @@ from tests.common import MockConfigEntry, async_fire_time_changed +def test_outdoor_model_alias() -> None: + """Test the outdoor model firmware typo uses outdoor capabilities.""" + assert get_model_capabilities("0-1PS") == get_model_capabilities("O-1PPT") + + async def test_device_info( hass: HomeAssistant, snapshot: SnapshotAssertion,