Generator: Allow specifying a microgrid id

Signed-off-by: Mathias L. Baumann <[email protected]>

Author: Marenz

src/frequenz/client/dispatch/test/generator.py

@@ -69,9 +69,12 @@ def generate_recurrence_rule(self) -> RecurrenceRule:
             ],
         )
 
-    def generate_dispatch(self) -> Dispatch:
+    def generate_dispatch(self, microgrid_id: int | None = None) -> Dispatch:
         """Generate a random dispatch instance.
 
+        Args:
+            microgrid_id: The microgrid_id to set on the dispatch.
+
         Returns:
             a random dispatch instance
         """
@@ -84,7 +87,7 @@ def generate_dispatch(self) -> Dispatch:
             id=self._last_id,
             create_time=create_time,
             update_time=create_time + timedelta(seconds=self._rng.randint(0, 1000000)),
-            microgrid_id=self._rng.randint(0, 100),
+            microgrid_id=microgrid_id or self._rng.randint(0, 100),
             type=str(self._rng.randint(0, 100_000)),
             start_time=datetime.now().astimezone(timezone.utc)
             + timedelta(seconds=self._rng.randint(0, 1000000)),