Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

tests: airzone: fix python 3.10 tests 

Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
pull/92894/head
Álvaro Fernández Rojas 2023-05-11 11:37:06 +02:00
parent e503a1dd3a
commit 93f8bd1430
2 changed files with 25 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/airzone/select.py
View File

@ -110,9 +110,10 @@ class AirzoneBaseSelect(AirzoneEntity, SelectEntity):
@callback @callback
def _async_update_attrs(self) -> None: def _async_update_attrs(self) -> None:
"""Update select attributes.""" """Update select attributes."""
self._attr_current_option = str( value = self.get_airzone_value(self.entity_description.key)
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): class AirzoneZoneSelect(AirzoneZoneEntity, AirzoneBaseSelect):

42
tests/components/airzone/test_select.py
View File

@ -26,49 +26,49 @@ async def test_airzone_create_selects(hass: HomeAssistant) -> None:
await async_init_integration(hass) await async_init_integration(hass)
state = hass.states.get("select.despacho_cold_angle") 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") 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") 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") 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") 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") 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") 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") 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") 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") 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") 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") 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") 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") 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") 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: 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, SERVICE_SELECT_OPTION,
{ {
ATTR_ENTITY_ID: "select.dorm_1_sleep", ATTR_ENTITY_ID: "select.dorm_1_sleep",
ATTR_OPTION: str(SleepTimeout.SLEEP_30), ATTR_OPTION: str(SleepTimeout.SLEEP_30.value),
}, },
blocking=True, blocking=True,
) )
state = hass.states.get("select.dorm_1_sleep") 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: 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, SERVICE_SELECT_OPTION,
{ {
ATTR_ENTITY_ID: "select.dorm_1_cold_angle", ATTR_ENTITY_ID: "select.dorm_1_cold_angle",
ATTR_OPTION: str(GrilleAngle.DEG_50), ATTR_OPTION: str(GrilleAngle.DEG_50.value),
}, },
blocking=True, blocking=True,
) )
state = hass.states.get("select.dorm_1_cold_angle") 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 # Heat Angle
@ -169,10 +169,10 @@ async def test_airzone_select_grille_angle(hass: HomeAssistant) -> None:
SERVICE_SELECT_OPTION, SERVICE_SELECT_OPTION,
{ {
ATTR_ENTITY_ID: "select.dorm_1_heat_angle", ATTR_ENTITY_ID: "select.dorm_1_heat_angle",
ATTR_OPTION: str(GrilleAngle.DEG_45), ATTR_OPTION: str(GrilleAngle.DEG_45.value),
}, },
blocking=True, blocking=True,
) )
state = hass.states.get("select.dorm_1_heat_angle") state = hass.states.get("select.dorm_1_heat_angle")
assert state.state == str(GrilleAngle.DEG_45) assert state.state == str(GrilleAngle.DEG_45.value)