1818from frequenz .sdk .actor import Actor
1919from pytest import fixture
2020
21- from frequenz .dispatch import Dispatch , DispatchManagingActor , DispatchUpdate
21+ from frequenz .dispatch import Dispatch , DispatchActorsService , DispatchUpdate
2222from frequenz .dispatch ._bg_service import DispatchScheduler
2323
2424
@@ -65,7 +65,7 @@ async def _run(self) -> None:
6565class TestEnv :
6666 """Test environment."""
6767
68- actors_service : DispatchManagingActor
68+ actors_service : DispatchActorsService
6969 running_status_sender : Sender [Dispatch ]
7070 generator : DispatchGenerator = DispatchGenerator ()
7171
@@ -90,7 +90,7 @@ async def test_env() -> AsyncIterator[TestEnv]:
9090 """Create a test environment."""
9191 channel = Broadcast [Dispatch ](name = "dispatch ready test channel" )
9292
93- actors_service = DispatchManagingActor (
93+ actors_service = DispatchActorsService (
9494 actor_factory = MockActor ,
9595 running_status_receiver = channel .new_receiver (),
9696 )
@@ -135,6 +135,7 @@ async def test_simple_start_stop(
135135 await asyncio .sleep (1 )
136136 logging .info ("Sent dispatch" )
137137
138+ assert test_env .actor is not None
138139 event = test_env .actor .initial_dispatch
139140 assert event .options == {"test" : True }
140141 assert event .components == dispatch .target
@@ -148,7 +149,7 @@ async def test_simple_start_stop(
148149 fake_time .shift (duration )
149150 await test_env .running_status_sender .send (Dispatch (dispatch ))
150151
151- # Give await actor.stop a chance to run in DispatchManagingActor
152+ # Give await actor.stop a chance to run
152153 await asyncio .sleep (1 )
153154
154155 assert test_env .actor is None
@@ -223,6 +224,7 @@ async def test_dry_run(test_env: TestEnv, fake_time: time_machine.Coordinates) -
223224 fake_time .shift (timedelta (seconds = 1 ))
224225 await asyncio .sleep (1 )
225226
227+ assert test_env .actor is not None
226228 event = test_env .actor .initial_dispatch
227229
228230 assert event .dry_run is dispatch .dry_run
@@ -235,7 +237,7 @@ async def test_dry_run(test_env: TestEnv, fake_time: time_machine.Coordinates) -
235237 fake_time .shift (dispatch .duration )
236238 await test_env .running_status_sender .send (Dispatch (dispatch ))
237239
238- # Give await actor.stop a chance to run in DispatchManagingActor
240+ # Give await actor.stop a chance to run
239241 await asyncio .sleep (1 )
240242
241243 assert test_env .actor is None
0 commit comments