Stop actors created by MockMicrogrid after each test

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

Author: shsms

tests/timeseries/mock_microgrid.py

@@ -6,6 +6,7 @@
 from __future__ import annotations
 
 import time
+import typing
 from typing import Iterator, Tuple
 
 from frequenz.api.microgrid import microgrid_pb2
@@ -67,6 +68,8 @@ def __init__(
         self.battery_ids: list[int] = []
         self.meter_ids: list[int] = [4]
 
+        self._actors: list[typing.Any] = []
+
     async def start(self) -> Tuple[Sender[ComponentMetricRequest], ChannelRegistry]:
         """Start the MockServer, and the data source and resampling actors.
 
@@ -248,20 +251,26 @@ async def _init_client_and_actors(
             resampling_actor_request_channel.new_receiver()
         )
 
-        DataSourcingActor(
-            request_receiver=data_source_request_receiver, registry=channel_registry
+        self._actors.append(
+            DataSourcingActor(
+                request_receiver=data_source_request_receiver, registry=channel_registry
+            )
         )
 
-        ComponentMetricsResamplingActor(
-            channel_registry=channel_registry,
-            data_sourcing_request_sender=data_source_request_sender,
-            resampling_request_receiver=resampling_actor_request_receiver,
-            config=ResamplerConfig(resampling_period_s=0.1),
+        self._actors.append(
+            ComponentMetricsResamplingActor(
+                channel_registry=channel_registry,
+                data_sourcing_request_sender=data_source_request_sender,
+                resampling_request_receiver=resampling_actor_request_receiver,
+                config=ResamplerConfig(resampling_period_s=0.1),
+            )
         )
 
         return (resampling_actor_request_sender, channel_registry)
 
     async def cleanup(self) -> None:
         """Clean up after a test."""
+        for actor in self._actors:
+            await actor._stop()  # pylint: disable=protected-access
         await self._server.graceful_shutdown()
         microgrid._microgrid._MICROGRID = None  # pylint: disable=protected-access