18
18
from frequenz .sdk .actor import Actor
19
19
from pytest import fixture
20
20
21
- from frequenz .dispatch import Dispatch , DispatchManagingActor , DispatchUpdate
21
+ from frequenz .dispatch import Dispatch , DispatchActorsService , DispatchUpdate
22
22
from frequenz .dispatch ._bg_service import DispatchScheduler
23
23
24
24
@@ -65,7 +65,7 @@ async def _run(self) -> None:
65
65
class TestEnv :
66
66
"""Test environment."""
67
67
68
- actors_service : DispatchManagingActor
68
+ actors_service : DispatchActorsService
69
69
running_status_sender : Sender [Dispatch ]
70
70
generator : DispatchGenerator = DispatchGenerator ()
71
71
@@ -90,7 +90,7 @@ async def test_env() -> AsyncIterator[TestEnv]:
90
90
"""Create a test environment."""
91
91
channel = Broadcast [Dispatch ](name = "dispatch ready test channel" )
92
92
93
- actors_service = DispatchManagingActor (
93
+ actors_service = DispatchActorsService (
94
94
actor_factory = MockActor ,
95
95
running_status_receiver = channel .new_receiver (),
96
96
)
@@ -135,6 +135,7 @@ async def test_simple_start_stop(
135
135
await asyncio .sleep (1 )
136
136
logging .info ("Sent dispatch" )
137
137
138
+ assert test_env .actor is not None
138
139
event = test_env .actor .initial_dispatch
139
140
assert event .options == {"test" : True }
140
141
assert event .components == dispatch .target
@@ -148,7 +149,7 @@ async def test_simple_start_stop(
148
149
fake_time .shift (duration )
149
150
await test_env .running_status_sender .send (Dispatch (dispatch ))
150
151
151
- # Give await actor.stop a chance to run in DispatchManagingActor
152
+ # Give await actor.stop a chance to run
152
153
await asyncio .sleep (1 )
153
154
154
155
assert test_env .actor is None
@@ -223,6 +224,7 @@ async def test_dry_run(test_env: TestEnv, fake_time: time_machine.Coordinates) -
223
224
fake_time .shift (timedelta (seconds = 1 ))
224
225
await asyncio .sleep (1 )
225
226
227
+ assert test_env .actor is not None
226
228
event = test_env .actor .initial_dispatch
227
229
228
230
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) -
235
237
fake_time .shift (dispatch .duration )
236
238
await test_env .running_status_sender .send (Dispatch (dispatch ))
237
239
238
- # Give await actor.stop a chance to run in DispatchManagingActor
240
+ # Give await actor.stop a chance to run
239
241
await asyncio .sleep (1 )
240
242
241
243
assert test_env .actor is None
0 commit comments