@@ -98,7 +98,7 @@ async def test_new_dispatch_created(
9898 generator : DispatchGenerator ,
9999) -> None :
100100 """Test that a new dispatch is created."""
101- sample = generator .generate_dispatch (actor_env . microgrid_id )
101+ sample = generator .generate_dispatch ()
102102
103103 await _test_new_dispatch_created (actor_env , sample )
104104
@@ -134,7 +134,7 @@ async def _test_new_dispatch_created(
134134 Returns:
135135 The sample dispatch that was created
136136 """
137- await actor_env .client .create (** to_create_params (sample ))
137+ await actor_env .client .create (** to_create_params (actor_env . microgrid_id , sample ))
138138
139139 dispatch_event = await actor_env .updated_dispatches .receive ()
140140
@@ -154,7 +154,7 @@ async def test_existing_dispatch_updated(
154154 fake_time : time_machine .Coordinates ,
155155) -> None :
156156 """Test that an existing dispatch is updated."""
157- sample = generator .generate_dispatch (actor_env . microgrid_id )
157+ sample = generator .generate_dispatch ()
158158 sample = replace (
159159 sample ,
160160 active = False ,
@@ -167,6 +167,7 @@ async def test_existing_dispatch_updated(
167167 fake_time .shift (timedelta (seconds = 1 ))
168168
169169 await actor_env .client .update (
170+ microgrid_id = actor_env .microgrid_id ,
170171 dispatch_id = sample .id ,
171172 new_fields = {
172173 "active" : True ,
@@ -199,11 +200,13 @@ async def test_existing_dispatch_deleted(
199200 fake_time : time_machine .Coordinates ,
200201) -> None :
201202 """Test that an existing dispatch is deleted."""
202- sample = generator .generate_dispatch (actor_env . microgrid_id )
203+ sample = generator .generate_dispatch ()
203204
204205 sample = await _test_new_dispatch_created (actor_env , sample )
205206
206- await actor_env .client .delete (sample .id )
207+ await actor_env .client .delete (
208+ microgrid_id = actor_env .microgrid_id , dispatch_id = sample .id
209+ )
207210 fake_time .shift (timedelta (seconds = 10 ))
208211 await asyncio .sleep (10 )
209212
@@ -223,9 +226,9 @@ async def test_dispatch_schedule(
223226 fake_time : time_machine .Coordinates ,
224227) -> None :
225228 """Test that a random dispatch is scheduled correctly."""
226- sample = generator .generate_dispatch (actor_env . microgrid_id )
227- await actor_env .client .create (** to_create_params (sample ))
228- dispatch = Dispatch (actor_env .client .dispatches [0 ])
229+ sample = generator .generate_dispatch ()
230+ await actor_env .client .create (** to_create_params (actor_env . microgrid_id , sample ))
231+ dispatch = Dispatch (actor_env .client .dispatches ( actor_env . microgrid_id ) [0 ])
229232
230233 next_run = dispatch .next_run_after (_now ())
231234 assert next_run is not None
0 commit comments