Add Switch entity to SleepIQ (#66966)

Co-authored-by: J. Nick Koston <nick@koston.org>

tests/components/sleepiq/test_switch.py

@@ -0,0 +1,69 @@
+"""The tests for SleepIQ switch platform."""
+from homeassistant.components.sleepiq.coordinator import LONGER_UPDATE_INTERVAL
+from homeassistant.components.switch import DOMAIN
+from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON
+from homeassistant.helpers import entity_registry as er
+from homeassistant.util.dt import utcnow
+
+from tests.common import async_fire_time_changed
+from tests.components.sleepiq.conftest import (
+    BED_ID,
+    BED_NAME,
+    BED_NAME_LOWER,
+    setup_platform,
+)
+
+
+async def test_setup(hass, mock_asyncsleepiq):
+    """Test for successfully setting up the SleepIQ platform."""
+    entry = await setup_platform(hass, DOMAIN)
+    entity_registry = er.async_get(hass)
+
+    assert len(entity_registry.entities) == 1
+
+    entry = entity_registry.async_get(f"switch.sleepnumber_{BED_NAME_LOWER}_pause_mode")
+    assert entry
+    assert entry.original_name == f"SleepNumber {BED_NAME} Pause Mode"
+    assert entry.unique_id == f"{BED_ID}-pause-mode"
+
+
+async def test_switch_set_states(hass, mock_asyncsleepiq):
+    """Test button press."""
+    await setup_platform(hass, DOMAIN)
+
+    await hass.services.async_call(
+        DOMAIN,
+        "turn_off",
+        {ATTR_ENTITY_ID: f"switch.sleepnumber_{BED_NAME_LOWER}_pause_mode"},
+        blocking=True,
+    )
+    await hass.async_block_till_done()
+    mock_asyncsleepiq.beds[BED_ID].set_pause_mode.assert_called_with(False)
+
+    await hass.services.async_call(
+        DOMAIN,
+        "turn_on",
+        {ATTR_ENTITY_ID: f"switch.sleepnumber_{BED_NAME_LOWER}_pause_mode"},
+        blocking=True,
+    )
+    await hass.async_block_till_done()
+    mock_asyncsleepiq.beds[BED_ID].set_pause_mode.assert_called_with(True)
+
+
+async def test_switch_get_states(hass, mock_asyncsleepiq):
+    """Test button press."""
+    await setup_platform(hass, DOMAIN)
+
+    assert (
+        hass.states.get(f"switch.sleepnumber_{BED_NAME_LOWER}_pause_mode").state
+        == STATE_OFF
+    )
+    mock_asyncsleepiq.beds[BED_ID].paused = True
+
+    async_fire_time_changed(hass, utcnow() + LONGER_UPDATE_INTERVAL)
+    await hass.async_block_till_done()
+
+    assert (
+        hass.states.get(f"switch.sleepnumber_{BED_NAME_LOWER}_pause_mode").state
+        == STATE_ON
+    )