Skip to content

Commit aef7863

Browse files
committed
Fix tests
Signed-off-by: Mathias L. Baumann <[email protected]>
1 parent fcd9c80 commit aef7863

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

tests/test_frequenz_dispatch.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -473,12 +473,13 @@ async def test_dispatch_new_but_finished(
473473
new_dispatch,
474474
active=True,
475475
duration=timedelta(seconds=10),
476-
start_time=_now() + timedelta(seconds=5),
476+
start_time=_now() + timedelta(seconds=500),
477477
recurrence=RecurrenceRule(),
478478
type="TEST_TYPE",
479479
)
480480

481481
new_dispatch = await _test_new_dispatch_created(test_env, new_dispatch)
482+
assert new_dispatch.started is False
482483

483484
# Advance time to when the new dispatch should still not start
484485
fake_time.shift(timedelta(seconds=100))

tests/test_mananging_actor.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -344,10 +344,15 @@ def __init__(self, *, server_url: str, key: str):
344344
sender = channel.new_sender()
345345

346346
async def new_mock_receiver(
347-
_: Dispatcher, dispatch_type: str, *, merge_strategy: MergeStrategy | None
347+
_: Dispatcher,
348+
dispatch_type: str,
349+
*,
350+
merge_strategy: MergeStrategy | None,
351+
initial_fetch_timeout: timedelta = timedelta(seconds=10),
348352
) -> Receiver[Dispatch]:
349353
assert dispatch_type == "MANAGE_TEST"
350354
assert merge_strategy is strategy
355+
assert initial_fetch_timeout == timedelta(seconds=10)
351356
return channel.new_receiver()
352357

353358
with patch(
@@ -363,6 +368,7 @@ async def new_mock_receiver(
363368
# pylint: disable=protected-access
364369
assert "MANAGE_TEST" in dispatcher._actor_dispatchers
365370
actor_manager = dispatcher._actor_dispatchers["MANAGE_TEST"]
371+
# pylint: disable=comparison-with-callable
366372
assert actor_manager._actor_factory == MockActor.create
367373

368374
dispatch = Dispatch(

0 commit comments

Comments
 (0)