Fix test_env in test that re-created it

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

Author: Marenz

tests/test_frequenz_dispatch.py

@@ -433,7 +433,7 @@ async def test_dispatch_new_but_finished(
     generator: DispatchGenerator,
     fake_time: time_machine.Coordinates,
 ) -> None:
-    """Test that a dispatch that is already finished is not started."""
+    """Test that a finished dispatch is not started at startup."""
     # Generate a dispatch that is already finished
     finished_dispatch = generator.generate_dispatch()
     finished_dispatch = replace(
@@ -444,10 +444,25 @@ async def test_dispatch_new_but_finished(
         recurrence=RecurrenceRule(),
         type="TEST_TYPE",
     )
-    # Create an old dispatch
+    # Inject an old dispatch
     test_env.client.set_dispatches(test_env.microgrid_id, [finished_dispatch])
     await test_env.service.stop()
     test_env.service.start()
+    test_env = replace(
+        test_env,
+        lifecycle_events=test_env.service.new_lifecycle_events_receiver("TEST_TYPE"),
+        running_state_change=(
+            await test_env.service.new_running_state_event_receiver(
+                "TEST_TYPE", unify_running_intervals=None
+            )
+        ),
+    )
+
+    fake_time.shift(timedelta(seconds=1))
+    # Process the lifecycle event caused by the old dispatch at startup
+    await test_env.lifecycle_events.receive()
+
+    await asyncio.sleep(1)
 
     # Create another dispatch the normal way
     new_dispatch = generator.generate_dispatch()
@@ -459,8 +474,7 @@ async def test_dispatch_new_but_finished(
         recurrence=RecurrenceRule(),
         type="TEST_TYPE",
     )
-    # Consume one lifecycle_updates event
-    await test_env.lifecycle_events.receive()
+
     new_dispatch = await _test_new_dispatch_created(test_env, new_dispatch)
 
     # Advance time to when the new dispatch should still not start