Skip to content

Commit 4dc8b0f

Browse files
committed
Update code for new client-dispatch version
Using `replace` now as `Dispatch` is frozen in the new version. Signed-off-by: Mathias L. Baumann <[email protected]>
1 parent 8851560 commit 4dc8b0f

File tree

1 file changed

+21
-13
lines changed

1 file changed

+21
-13
lines changed

tests/test_frequenz_dispatch.py

Lines changed: 21 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)