@@ -55,7 +55,7 @@ def _now() -> datetime:
5555
5656
5757@dataclass (frozen = True ) 
58- class  TestEnv :
58+ class  _TestEnv :
5959    """Test environment for the service.""" 
6060
6161    service : DispatchScheduler 
@@ -71,7 +71,7 @@ class TestEnv:
7171
7272
7373@fixture  
74- async  def  test_env () ->  AsyncIterator [TestEnv ]:
74+ async  def  test_env () ->  AsyncIterator [_TestEnv ]:
7575    """Return an actor test environment.""" 
7676    microgrid_id  =  randint (1 , 100 )
7777    client  =  FakeClient ()
@@ -83,7 +83,7 @@ async def test_env() -> AsyncIterator[TestEnv]:
8383
8484    service .start ()
8585    try :
86-         yield  TestEnv (
86+         yield  _TestEnv (
8787            service = service ,
8888            lifecycle_events = service .new_lifecycle_events_receiver ("TEST_TYPE" ),
8989            running_state_change = await  service .new_running_state_event_receiver (
@@ -103,7 +103,7 @@ def generator() -> DispatchGenerator:
103103
104104
105105async  def  test_new_dispatch_created (
106-     test_env : TestEnv ,
106+     test_env : _TestEnv ,
107107    generator : DispatchGenerator ,
108108) ->  None :
109109    """Test that a new dispatch is created.""" 
@@ -131,7 +131,7 @@ def update_dispatch(sample: BaseDispatch, dispatch: BaseDispatch) -> BaseDispatc
131131
132132
133133async  def  _test_new_dispatch_created (
134-     test_env : TestEnv ,
134+     test_env : _TestEnv ,
135135    sample : BaseDispatch ,
136136) ->  Dispatch :
137137    """Test that a new dispatch is created. 
@@ -159,7 +159,7 @@ async def _test_new_dispatch_created(
159159
160160
161161async  def  test_existing_dispatch_updated (
162-     test_env : TestEnv ,
162+     test_env : _TestEnv ,
163163    generator : DispatchGenerator ,
164164    fake_time : time_machine .Coordinates ,
165165) ->  None :
@@ -197,7 +197,7 @@ async def test_existing_dispatch_updated(
197197
198198
199199async  def  test_existing_dispatch_deleted (
200-     test_env : TestEnv ,
200+     test_env : _TestEnv ,
201201    generator : DispatchGenerator ,
202202    fake_time : time_machine .Coordinates ,
203203) ->  None :
@@ -220,7 +220,7 @@ async def test_existing_dispatch_deleted(
220220
221221
222222async  def  test_dispatch_inf_duration_deleted (
223-     test_env : TestEnv ,
223+     test_env : _TestEnv ,
224224    generator : DispatchGenerator ,
225225    fake_time : time_machine .Coordinates ,
226226) ->  None :
@@ -255,7 +255,7 @@ async def test_dispatch_inf_duration_deleted(
255255
256256
257257async  def  test_dispatch_inf_duration_updated_stopped_started (
258-     test_env : TestEnv ,
258+     test_env : _TestEnv ,
259259    generator : DispatchGenerator ,
260260    fake_time : time_machine .Coordinates ,
261261) ->  None :
@@ -304,7 +304,7 @@ async def test_dispatch_inf_duration_updated_stopped_started(
304304
305305
306306async  def  test_dispatch_inf_duration_updated_to_finite_and_stops (
307-     test_env : TestEnv ,
307+     test_env : _TestEnv ,
308308    generator : DispatchGenerator ,
309309    fake_time : time_machine .Coordinates ,
310310) ->  None :
@@ -347,7 +347,7 @@ async def test_dispatch_inf_duration_updated_to_finite_and_stops(
347347
348348
349349async  def  test_dispatch_schedule (
350-     test_env : TestEnv ,
350+     test_env : _TestEnv ,
351351    generator : DispatchGenerator ,
352352    fake_time : time_machine .Coordinates ,
353353) ->  None :
@@ -385,7 +385,7 @@ async def test_dispatch_schedule(
385385
386386
387387async  def  test_dispatch_inf_duration_updated_to_finite_and_continues (
388-     test_env : TestEnv ,
388+     test_env : _TestEnv ,
389389    generator : DispatchGenerator ,
390390    fake_time : time_machine .Coordinates ,
391391) ->  None :
@@ -432,7 +432,7 @@ async def test_dispatch_inf_duration_updated_to_finite_and_continues(
432432
433433
434434async  def  test_dispatch_new_but_finished (
435-     test_env : TestEnv ,
435+     test_env : _TestEnv ,
436436    generator : DispatchGenerator ,
437437    fake_time : time_machine .Coordinates ,
438438) ->  None :
@@ -490,7 +490,7 @@ async def test_dispatch_new_but_finished(
490490
491491
492492async  def  test_notification_on_actor_start (
493-     test_env : TestEnv ,
493+     test_env : _TestEnv ,
494494    generator : DispatchGenerator ,
495495    fake_time : time_machine .Coordinates ,
496496) ->  None :
0 commit comments