From 4dc8aff3d564145d8d28492e2c3510e3f238cb45 Mon Sep 17 00:00:00 2001
From: Mike Fugate <mfugate@carnegielearning.com>
Date: Fri, 25 Mar 2022 16:47:28 -0400
Subject: [PATCH] Bump asyncsleepiq to 1.2.1 (#68680)

---
 .../components/sleepiq/manifest.json          |  2 +-
 homeassistant/components/sleepiq/select.py    | 25 +++----------------
 requirements_all.txt                          |  2 +-
 requirements_test_all.txt                     |  2 +-
 tests/components/sleepiq/test_select.py       |  6 ++---
 5 files changed, 8 insertions(+), 29 deletions(-)

diff --git a/homeassistant/components/sleepiq/manifest.json b/homeassistant/components/sleepiq/manifest.json
index bb386f1cf42..16881506cb8 100644
--- a/homeassistant/components/sleepiq/manifest.json
+++ b/homeassistant/components/sleepiq/manifest.json
@@ -3,7 +3,7 @@
   "name": "SleepIQ",
   "config_flow": true,
   "documentation": "https://www.home-assistant.io/integrations/sleepiq",
-  "requirements": ["asyncsleepiq==1.2.0"],
+  "requirements": ["asyncsleepiq==1.2.1"],
   "codeowners": ["@mfugate1", "@kbickar"],
   "dhcp": [
     {
diff --git a/homeassistant/components/sleepiq/select.py b/homeassistant/components/sleepiq/select.py
index 2bf305007ca..711f228d7f2 100644
--- a/homeassistant/components/sleepiq/select.py
+++ b/homeassistant/components/sleepiq/select.py
@@ -1,16 +1,7 @@
 """Support for SleepIQ foundation preset selection."""
 from __future__ import annotations
 
-from asyncsleepiq import (
-    FAVORITE,
-    FLAT,
-    READ,
-    SNORE,
-    WATCH_TV,
-    ZERO_G,
-    SleepIQBed,
-    SleepIQPreset,
-)
+from asyncsleepiq import BED_PRESETS, SleepIQBed, SleepIQPreset
 
 from homeassistant.components.select import SelectEntity
 from homeassistant.config_entries import ConfigEntry
@@ -22,16 +13,6 @@ from .const import DOMAIN
 from .coordinator import SleepIQData
 from .entity import SleepIQBedEntity
 
-FOUNDATION_PRESET_NAMES = {
-    "Not at preset": 0,
-    "Favorite": FAVORITE,
-    "Read": READ,
-    "Watch TV": WATCH_TV,
-    "Flat": FLAT,
-    "Zero G": ZERO_G,
-    "Snore": SNORE,
-}
-
 
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -50,7 +31,7 @@ async def async_setup_entry(
 class SleepIQSelectEntity(SleepIQBedEntity, SelectEntity):
     """Representation of a SleepIQ select entity."""
 
-    _attr_options = list(FOUNDATION_PRESET_NAMES.keys())
+    _attr_options = list(BED_PRESETS)
 
     def __init__(
         self, coordinator: DataUpdateCoordinator, bed: SleepIQBed, preset: SleepIQPreset
@@ -77,6 +58,6 @@ class SleepIQSelectEntity(SleepIQBedEntity, SelectEntity):
 
     async def async_select_option(self, option: str) -> None:
         """Change the current preset."""
-        await self.preset.set_preset(FOUNDATION_PRESET_NAMES[option])
+        await self.preset.set_preset(option)
         self._attr_current_option = option
         self.async_write_ha_state()
diff --git a/requirements_all.txt b/requirements_all.txt
index fc4522c6e5c..c0e2336941f 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -335,7 +335,7 @@ async-upnp-client==0.27.0
 asyncpysupla==0.0.5
 
 # homeassistant.components.sleepiq
-asyncsleepiq==1.2.0
+asyncsleepiq==1.2.1
 
 # homeassistant.components.aten_pe
 atenpdu==0.3.2
diff --git a/requirements_test_all.txt b/requirements_test_all.txt
index 5f493a5fe07..b78f5c70383 100644
--- a/requirements_test_all.txt
+++ b/requirements_test_all.txt
@@ -265,7 +265,7 @@ arcam-fmj==0.12.0
 async-upnp-client==0.27.0
 
 # homeassistant.components.sleepiq
-asyncsleepiq==1.2.0
+asyncsleepiq==1.2.1
 
 # homeassistant.components.aurora
 auroranoaa==0.0.2
diff --git a/tests/components/sleepiq/test_select.py b/tests/components/sleepiq/test_select.py
index 1b0c34d167c..855ca518f2d 100644
--- a/tests/components/sleepiq/test_select.py
+++ b/tests/components/sleepiq/test_select.py
@@ -1,8 +1,6 @@
 """Tests for the SleepIQ select platform."""
 from unittest.mock import MagicMock
 
-from asyncsleepiq import ZERO_G
-
 from homeassistant.components.select import DOMAIN, SERVICE_SELECT_OPTION
 from homeassistant.const import (
     ATTR_ENTITY_ID,
@@ -75,7 +73,7 @@ async def test_split_foundation_preset(
 
     mock_asyncsleepiq.beds[BED_ID].foundation.presets[0].set_preset.assert_called_once()
     mock_asyncsleepiq.beds[BED_ID].foundation.presets[0].set_preset.assert_called_with(
-        ZERO_G
+        "Zero G"
     )
 
 
@@ -116,4 +114,4 @@ async def test_single_foundation_preset(
     ].set_preset.assert_called_once()
     mock_asyncsleepiq_single_foundation.beds[BED_ID].foundation.presets[
         0
-    ].set_preset.assert_called_with(ZERO_G)
+    ].set_preset.assert_called_with("Zero G")