From 779b19ca46d4d763ec67cf3144e4a2978b36a4b9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=81lvaro=20Fern=C3=A1ndez=20Rojas?= Date: Mon, 6 Nov 2023 10:51:14 +0100 Subject: [PATCH] On Airzone cloud unload logout (#103487) --- homeassistant/components/airzone_cloud/__init__.py | 4 +++- tests/components/airzone_cloud/test_init.py | 3 +++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/airzone_cloud/__init__.py b/homeassistant/components/airzone_cloud/__init__.py index 38c764d4889..7e787ef4c69 100644 --- a/homeassistant/components/airzone_cloud/__init__.py +++ b/homeassistant/components/airzone_cloud/__init__.py @@ -46,7 +46,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Unload a config entry.""" + if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS): - hass.data[DOMAIN].pop(entry.entry_id) + coordinator: AirzoneUpdateCoordinator = hass.data[DOMAIN].pop(entry.entry_id) + await coordinator.airzone.logout() return unload_ok diff --git a/tests/components/airzone_cloud/test_init.py b/tests/components/airzone_cloud/test_init.py index 3a6497fdeba..f8a7a710e08 100644 --- a/tests/components/airzone_cloud/test_init.py +++ b/tests/components/airzone_cloud/test_init.py @@ -24,6 +24,9 @@ async def test_unload_entry(hass: HomeAssistant) -> None: with patch( "homeassistant.components.airzone_cloud.AirzoneCloudApi.login", return_value=None, + ), patch( + "homeassistant.components.airzone_cloud.AirzoneCloudApi.logout", + return_value=None, ), patch( "homeassistant.components.airzone_cloud.AirzoneCloudApi.list_installations", return_value=[],