tests: airzone: fix python 3.10 tests

Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
Noltari · May 11, 2023 · 93f8bd1 · 93f8bd1
1 parent e503a1d
commit 93f8bd1
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 24 deletions.
diff --git a/homeassistant/components/airzone/select.py b/homeassistant/components/airzone/select.py
@@ -110,9 +110,10 @@ def _handle_coordinator_update(self) -> None:
     @callback
     def _async_update_attrs(self) -> None:
         """Update select attributes."""
-        self._attr_current_option = str(
-            self.get_airzone_value(self.entity_description.key)
-        )
+        value = self.get_airzone_value(self.entity_description.key)
+        if value is not None:
+            value = str(value.value)
+        self._attr_current_option = value
 
 
 class AirzoneZoneSelect(AirzoneZoneEntity, AirzoneBaseSelect):

diff --git a/tests/components/airzone/test_select.py b/tests/components/airzone/test_select.py
@@ -26,49 +26,49 @@ async def test_airzone_create_selects(hass: HomeAssistant) -> None:
     await async_init_integration(hass)
 
     state = hass.states.get("select.despacho_cold_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.despacho_heat_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.despacho_sleep")
-    assert state.state == str(SleepTimeout.SLEEP_OFF)
+    assert state.state == str(SleepTimeout.SLEEP_OFF.value)
 
     state = hass.states.get("select.dorm_1_cold_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.dorm_1_heat_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.dorm_1_sleep")
-    assert state.state == str(SleepTimeout.SLEEP_OFF)
+    assert state.state == str(SleepTimeout.SLEEP_OFF.value)
 
     state = hass.states.get("select.dorm_2_cold_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.dorm_2_heat_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.dorm_2_sleep")
-    assert state.state == str(SleepTimeout.SLEEP_OFF)
+    assert state.state == str(SleepTimeout.SLEEP_OFF.value)
 
     state = hass.states.get("select.dorm_ppal_cold_angle")
-    assert state.state == str(GrilleAngle.DEG_45)
+    assert state.state == str(GrilleAngle.DEG_45.value)
 
     state = hass.states.get("select.dorm_ppal_heat_angle")
-    assert state.state == str(GrilleAngle.DEG_50)
+    assert state.state == str(GrilleAngle.DEG_50.value)
 
     state = hass.states.get("select.dorm_ppal_sleep")
-    assert state.state == str(SleepTimeout.SLEEP_30)
+    assert state.state == str(SleepTimeout.SLEEP_30.value)
 
     state = hass.states.get("select.salon_cold_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.salon_heat_angle")
-    assert state.state == str(GrilleAngle.DEG_90)
+    assert state.state == str(GrilleAngle.DEG_90.value)
 
     state = hass.states.get("select.salon_sleep")
-    assert state.state == str(SleepTimeout.SLEEP_OFF)
+    assert state.state == str(SleepTimeout.SLEEP_OFF.value)
 
 
 async def test_airzone_select_sleep(hass: HomeAssistant) -> None:
@@ -106,13 +106,13 @@ async def test_airzone_select_sleep(hass: HomeAssistant) -> None:
             SERVICE_SELECT_OPTION,
             {
                 ATTR_ENTITY_ID: "select.dorm_1_sleep",
-                ATTR_OPTION: str(SleepTimeout.SLEEP_30),
+                ATTR_OPTION: str(SleepTimeout.SLEEP_30.value),
             },
             blocking=True,
         )
 
     state = hass.states.get("select.dorm_1_sleep")
-    assert state.state == str(SleepTimeout.SLEEP_30)
+    assert state.state == str(SleepTimeout.SLEEP_30.value)
 
 
 async def test_airzone_select_grille_angle(hass: HomeAssistant) -> None:
@@ -141,13 +141,13 @@ async def test_airzone_select_grille_angle(hass: HomeAssistant) -> None:
             SERVICE_SELECT_OPTION,
             {
                 ATTR_ENTITY_ID: "select.dorm_1_cold_angle",
-                ATTR_OPTION: str(GrilleAngle.DEG_50),
+                ATTR_OPTION: str(GrilleAngle.DEG_50.value),
             },
             blocking=True,
         )
 
     state = hass.states.get("select.dorm_1_cold_angle")
-    assert state.state == str(GrilleAngle.DEG_50)
+    assert state.state == str(GrilleAngle.DEG_50.value)
 
     # Heat Angle
 
@@ -169,10 +169,10 @@ async def test_airzone_select_grille_angle(hass: HomeAssistant) -> None:
             SERVICE_SELECT_OPTION,
             {
                 ATTR_ENTITY_ID: "select.dorm_1_heat_angle",
-                ATTR_OPTION: str(GrilleAngle.DEG_45),
+                ATTR_OPTION: str(GrilleAngle.DEG_45.value),
             },
             blocking=True,
         )
 
     state = hass.states.get("select.dorm_1_heat_angle")
-    assert state.state == str(GrilleAngle.DEG_45)
+    assert state.state == str(GrilleAngle.DEG_45.value)