@@ -127,9 +127,12 @@ async def _test_new_dispatch_created(
127127 case Deleted (dispatch ) | Updated (dispatch ):
128128 assert False , "Expected a created event"
129129 case Created (dispatch ):
130- sample .update_time = dispatch .update_time
131- sample .create_time = dispatch .create_time
132- sample .id = dispatch .id
130+ sample = replace (
131+ sample ,
132+ update_time = dispatch .update_time ,
133+ create_time = dispatch .create_time ,
134+ id = dispatch .id ,
135+ )
133136 assert dispatch == sample
134137
135138 return sample
@@ -142,16 +145,19 @@ async def test_existing_dispatch_updated(
142145) -> None :
143146 """Test that an existing dispatch is updated."""
144147 sample = generator .generate_dispatch (actor_env .microgrid_id )
145- sample .active = False
146- sample .recurrence .frequency = Frequency .DAILY
148+ sample = replace (
149+ sample ,
150+ active = False ,
151+ recurrence = replace (sample .recurrence , frequency = Frequency .DAILY ),
152+ )
147153
148154 fake_time .shift (timedelta (seconds = 1 ))
149155
150- await _test_new_dispatch_created (actor_env , sample )
156+ sample = await _test_new_dispatch_created (actor_env , sample )
151157 fake_time .shift (timedelta (seconds = 1 ))
152158
153159 await actor_env .client .update (
154- sample .id ,
160+ dispatch_id = sample .id ,
155161 new_fields = {
156162 "active" : True ,
157163 "recurrence.frequency" : Frequency .UNSPECIFIED ,
@@ -164,11 +170,13 @@ async def test_existing_dispatch_updated(
164170 case Created (dispatch ) | Deleted (dispatch ):
165171 assert False , "Expected an updated event"
166172 case Updated (dispatch ):
167- sample .update_time = dispatch .update_time
168- sample .active = True
169- sample .recurrence = replace (
170- sample .recurrence , frequency = Frequency .UNSPECIFIED
173+ sample = replace (
174+ sample ,
175+ active = True ,
176+ recurrence = replace (sample .recurrence , frequency = Frequency .UNSPECIFIED ),
177+ update_time = dispatch .update_time ,
171178 )
179+
172180 assert dispatch == sample
173181
174182
@@ -180,7 +188,7 @@ async def test_existing_dispatch_deleted(
180188 """Test that an existing dispatch is deleted."""
181189 sample = generator .generate_dispatch (actor_env .microgrid_id )
182190
183- await _test_new_dispatch_created (actor_env , sample )
191+ sample = await _test_new_dispatch_created (actor_env , sample )
184192
185193 await actor_env .client .delete (sample .id )
186194 fake_time .shift (timedelta (seconds = 1 ))
@@ -190,7 +198,7 @@ async def test_existing_dispatch_deleted(
190198 case Created (dispatch ) | Updated (dispatch ):
191199 assert False , "Expected a deleted event"
192200 case Deleted (dispatch ):
193- sample . create_time = dispatch .create_time
201+ sample = replace ( sample , update_time = dispatch .update_time )
194202 assert dispatch == sample
195203
196204
0 commit comments