From 0f2525d47601682cfe718242e252b368ef3068f0 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sun, 8 Sep 2024 22:13:32 +0200
Subject: [PATCH] Migrate anova to use runtime_data (#125536)

---
 homeassistant/components/anova/__init__.py | 15 +++++----------
 homeassistant/components/anova/models.py   |  4 ++++
 homeassistant/components/anova/sensor.py   |  8 +++-----
 tests/components/anova/test_init.py        |  2 +-
 4 files changed, 13 insertions(+), 16 deletions(-)

diff --git a/homeassistant/components/anova/__init__.py b/homeassistant/components/anova/__init__.py
index 7503de8ea10..02c468c1319 100644
--- a/homeassistant/components/anova/__init__.py
+++ b/homeassistant/components/anova/__init__.py
@@ -13,22 +13,20 @@ from anova_wifi import (
     WebsocketFailure,
 )
 
-from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady
 from homeassistant.helpers import aiohttp_client
 
-from .const import DOMAIN
 from .coordinator import AnovaCoordinator
-from .models import AnovaData
+from .models import AnovaConfigEntry, AnovaData
 
 PLATFORMS = [Platform.SENSOR]
 
 _LOGGER = logging.getLogger(__name__)
 
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: AnovaConfigEntry) -> bool:
     """Set up Anova from a config entry."""
     api = AnovaApi(
         aiohttp_client.async_get_clientsession(hass),
@@ -62,17 +60,14 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         assert api.websocket_handler is not None
     devices = list(api.websocket_handler.devices.values())
     coordinators = [AnovaCoordinator(hass, device) for device in devices]
-    hass.data.setdefault(DOMAIN, {})[entry.entry_id] = AnovaData(
-        api_jwt=api.jwt, coordinators=coordinators, api=api
-    )
+    entry.runtime_data = AnovaData(api_jwt=api.jwt, coordinators=coordinators, api=api)
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
     return True
 
 
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: AnovaConfigEntry) -> bool:
     """Unload a config entry."""
     if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
-        anova_data: AnovaData = hass.data[DOMAIN].pop(entry.entry_id)
         # Disconnect from WS
-        await anova_data.api.disconnect_websocket()
+        await entry.runtime_data.api.disconnect_websocket()
     return unload_ok
diff --git a/homeassistant/components/anova/models.py b/homeassistant/components/anova/models.py
index 8caf16eeae1..eef8180cf88 100644
--- a/homeassistant/components/anova/models.py
+++ b/homeassistant/components/anova/models.py
@@ -4,8 +4,12 @@ from dataclasses import dataclass
 
 from anova_wifi import AnovaApi
 
+from homeassistant.config_entries import ConfigEntry
+
 from .coordinator import AnovaCoordinator
 
+type AnovaConfigEntry = ConfigEntry[AnovaData]
+
 
 @dataclass
 class AnovaData:
diff --git a/homeassistant/components/anova/sensor.py b/homeassistant/components/anova/sensor.py
index e5fe9ededfd..aa572a0ee9b 100644
--- a/homeassistant/components/anova/sensor.py
+++ b/homeassistant/components/anova/sensor.py
@@ -7,7 +7,6 @@ from dataclasses import dataclass
 
 from anova_wifi import AnovaMode, AnovaState, APCUpdateSensor
 
-from homeassistant import config_entries
 from homeassistant.components.sensor import (
     SensorDeviceClass,
     SensorEntity,
@@ -19,10 +18,9 @@ from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.typing import StateType
 
-from .const import DOMAIN
 from .coordinator import AnovaCoordinator
 from .entity import AnovaDescriptionEntity
-from .models import AnovaData
+from .models import AnovaConfigEntry
 
 
 @dataclass(frozen=True, kw_only=True)
@@ -99,11 +97,11 @@ SENSOR_DESCRIPTIONS: list[AnovaSensorEntityDescription] = [
 
 async def async_setup_entry(
     hass: HomeAssistant,
-    entry: config_entries.ConfigEntry,
+    entry: AnovaConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
     """Set up Anova device."""
-    anova_data: AnovaData = hass.data[DOMAIN][entry.entry_id]
+    anova_data = entry.runtime_data
 
     for coordinator in anova_data.coordinators:
         setup_coordinator(coordinator, async_add_entities)
diff --git a/tests/components/anova/test_init.py b/tests/components/anova/test_init.py
index 5fc63fcaf93..66ea11fdaef 100644
--- a/tests/components/anova/test_init.py
+++ b/tests/components/anova/test_init.py
@@ -2,7 +2,7 @@
 
 from anova_wifi import AnovaApi
 
-from homeassistant.components.anova import DOMAIN
+from homeassistant.components.anova.const import DOMAIN
 from homeassistant.config_entries import ConfigEntryState
 from homeassistant.core import HomeAssistant