Close status receivers when stopping a *pool

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/timeseries/battery_pool/_battery_pool_reference_store.py

@@ -90,9 +90,12 @@ def __init__(  # pylint: disable=too-many-arguments
         self._working_batteries: set[int] = set()
 
         self._update_battery_status_task: asyncio.Task[None] | None = None
+        self._batteries_status_receiver: Receiver[ComponentPoolStatus] = (
+            batteries_status_receiver
+        )
         if self._batteries:
             self._update_battery_status_task = asyncio.create_task(
-                self._update_battery_status(batteries_status_receiver)
+                self._update_battery_status(self._batteries_status_receiver)
             )
 
         self._min_update_interval: timedelta = min_update_interval
@@ -128,6 +131,7 @@ async def stop(self) -> None:
         if self._update_battery_status_task:
             tasks_to_stop.append(cancel_and_await(self._update_battery_status_task))
         await asyncio.gather(*tasks_to_stop)
+        self._batteries_status_receiver.close()
 
     def _get_all_batteries(self) -> frozenset[int]:
         """Get all batteries from the microgrid.

src/frequenz/sdk/timeseries/ev_charger_pool/_ev_charger_pool_reference_store.py

@@ -107,3 +107,4 @@ async def stop(self) -> None:
         """Stop all tasks and channels owned by the EVChargerPool."""
         await self.formula_pool.stop()
         await self.bounds_tracker.stop()
+        self.status_receiver.close()

src/frequenz/sdk/timeseries/pv_pool/_pv_pool_reference_store.py

@@ -108,3 +108,4 @@ async def stop(self) -> None:
         """Stop all tasks and channels owned by the PVInverterPool."""
         await self.formula_pool.stop()
         await self.bounds_tracker.stop()
+        self.status_receiver.close()