Add additional select for dmaker.airfresh.t2017 to xiaomi_miio (#67058)

tests/components/xiaomi_miio/test_select.py

@@ -0,0 +1,158 @@
+"""The tests for the xiaomi_miio select component."""
+
+from unittest.mock import MagicMock, patch
+
+from arrow import utcnow
+from miio.integrations.airpurifier.dmaker.airfresh_t2017 import (
+    DisplayOrientation,
+    PtcLevel,
+)
+import pytest
+
+from homeassistant.components.select import DOMAIN
+from homeassistant.components.select.const import (
+    ATTR_OPTION,
+    ATTR_OPTIONS,
+    SERVICE_SELECT_OPTION,
+)
+from homeassistant.components.xiaomi_miio import UPDATE_INTERVAL
+from homeassistant.components.xiaomi_miio.const import (
+    CONF_DEVICE,
+    CONF_FLOW_TYPE,
+    CONF_MAC,
+    DOMAIN as XIAOMI_DOMAIN,
+    MODEL_AIRFRESH_T2017,
+)
+from homeassistant.const import (
+    ATTR_ENTITY_ID,
+    CONF_HOST,
+    CONF_MODEL,
+    CONF_TOKEN,
+    Platform,
+)
+from homeassistant.core import HomeAssistant
+
+from tests.common import MockConfigEntry, async_fire_time_changed
+from tests.components.xiaomi_miio import TEST_MAC
+
+
+@pytest.fixture(autouse=True)
+async def setup_test(hass: HomeAssistant):
+    """Initialize test xiaomi_miio for select entity."""
+
+    mock_airfresh = MagicMock()
+    mock_airfresh.status().display_orientation = DisplayOrientation.Portrait
+    mock_airfresh.status().ptc_level = PtcLevel.Low
+
+    with patch(
+        "homeassistant.components.xiaomi_miio.get_platforms",
+        return_value=[
+            Platform.SELECT,
+        ],
+    ), patch("homeassistant.components.xiaomi_miio.AirFreshT2017") as mock_airfresh_cls:
+        mock_airfresh_cls.return_value = mock_airfresh
+        yield mock_airfresh
+
+
+async def test_select_params(hass: HomeAssistant) -> None:
+    """Test the initial parameters."""
+
+    entity_name = "test_airfresh_select"
+    entity_id = await setup_component(hass, entity_name)
+
+    select_entity = hass.states.get(entity_id + "_display_orientation")
+    assert select_entity
+    assert select_entity.state == "forward"
+    assert select_entity.attributes.get(ATTR_OPTIONS) == ["forward", "left", "right"]
+
+
+async def test_select_bad_attr(hass: HomeAssistant) -> None:
+    """Test selecting a different option with invalid option value."""
+
+    entity_name = "test_airfresh_select"
+    entity_id = await setup_component(hass, entity_name)
+
+    state = hass.states.get(entity_id + "_display_orientation")
+    assert state
+    assert state.state == "forward"
+
+    with pytest.raises(ValueError):
+        await hass.services.async_call(
+            "select",
+            SERVICE_SELECT_OPTION,
+            {ATTR_OPTION: "up", ATTR_ENTITY_ID: entity_id + "_display_orientation"},
+            blocking=True,
+        )
+    await hass.async_block_till_done()
+
+    state = hass.states.get(entity_id + "_display_orientation")
+    assert state
+    assert state.state == "forward"
+
+
+async def test_select_option(hass: HomeAssistant) -> None:
+    """Test selecting of a option."""
+
+    entity_name = "test_airfresh_select"
+    entity_id = await setup_component(hass, entity_name)
+
+    state = hass.states.get(entity_id + "_display_orientation")
+    assert state
+    assert state.state == "forward"
+
+    await hass.services.async_call(
+        "select",
+        SERVICE_SELECT_OPTION,
+        {ATTR_OPTION: "left", ATTR_ENTITY_ID: entity_id + "_display_orientation"},
+        blocking=True,
+    )
+    await hass.async_block_till_done()
+
+    state = hass.states.get(entity_id + "_display_orientation")
+    assert state
+    assert state.state == "left"
+
+
+async def test_select_coordinator_update(hass: HomeAssistant, setup_test) -> None:
+    """Test coordinator update of a option."""
+
+    entity_name = "test_airfresh_select"
+    entity_id = await setup_component(hass, entity_name)
+
+    state = hass.states.get(entity_id + "_display_orientation")
+    assert state
+    assert state.state == "forward"
+
+    # emulate someone change state from device maybe used app
+    setup_test.status().display_orientation = DisplayOrientation.LandscapeLeft
+
+    async_fire_time_changed(hass, utcnow() + UPDATE_INTERVAL)
+    await hass.async_block_till_done()
+
+    state = hass.states.get(entity_id + "_display_orientation")
+    assert state
+    assert state.state == "left"
+
+
+async def setup_component(hass, entity_name):
+    """Set up component."""
+    entity_id = f"{DOMAIN}.{entity_name}"
+
+    config_entry = MockConfigEntry(
+        domain=XIAOMI_DOMAIN,
+        unique_id="123456",
+        title=entity_name,
+        data={
+            CONF_FLOW_TYPE: CONF_DEVICE,
+            CONF_HOST: "0.0.0.0",
+            CONF_TOKEN: "12345678901234567890123456789012",
+            CONF_MODEL: MODEL_AIRFRESH_T2017,
+            CONF_MAC: TEST_MAC,
+        },
+    )
+
+    config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    return entity_id