Cover legacy data-retrieval error detection

Author: bouwew

plugwise/legacy/smile.py

@@ -25,7 +25,7 @@
     PlugwiseData,
     ThermoLoc,
 )
-from plugwise.exceptions import ConnectionFailedError, PlugwiseError
+from plugwise.exceptions import ConnectionFailedError, DataMissingError, PlugwiseError
 from plugwise.legacy.data import SmileLegacyData
 
 import aiohttp
@@ -120,18 +120,28 @@ async def async_update(self) -> PlugwiseData:
             )
             self.gw_data: GatewayData = {}
             self.gw_entities: dict[str, GwEntityData] = {}
-            await self.full_xml_update()
-            self.get_all_gateway_entities()
+            try:
+                await self.full_xml_update()
+                self.get_all_gateway_entities()
+                # Detect failed data-retrieval
+                _ = self.gw_entities[self.gateway_id]["location"]
+            except KeyError as err:
+                raise DataMissingError("No Plugwise data received") from err
         # Otherwise perform an incremental update
         else:
-            self._domain_objects = await self.request(DOMAIN_OBJECTS)
-            match self._target_smile:
-                case "smile_v2":
-                    self._modules = await self.request(MODULES)
-                case self._target_smile if self._target_smile in REQUIRE_APPLIANCES:
-                    self._appliances = await self.request(APPLIANCES)
-
-            self._update_gw_entities()
+            try:
+                self._domain_objects = await self.request(DOMAIN_OBJECTS)
+                match self._target_smile:
+                    case "smile_v2":
+                        self._modules = await self.request(MODULES)
+                    case self._target_smile if self._target_smile in REQUIRE_APPLIANCES:
+                        self._appliances = await self.request(APPLIANCES)
+
+                self._update_gw_entities()
+                # Detect failed data-retrieval
+                _ = self.gw_entities[self.gateway_id]["location"]
+            except KeyError as err:
+                raise DataMissingError("No Plugwise data received") from err
 
         self._previous_day_number = day_number
         return PlugwiseData(