diff options
author | Alex Grönholm <alex.gronholm@nextday.fi> | 2021-09-06 00:10:34 +0300 |
---|---|---|
committer | Alex Grönholm <alex.gronholm@nextday.fi> | 2021-09-06 01:39:07 +0300 |
commit | da94af0571e6c40553b4463b5fba1d94f598b136 (patch) | |
tree | 9becae5076b7cfa75224c686a6efc1a2d364f301 /tests | |
parent | 148b29270eb8fa0974f29be4d85a0ee03b848d1a (diff) | |
download | apscheduler-da94af0571e6c40553b4463b5fba1d94f598b136.tar.gz |
Added missing TaskUpdated event
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_datastores.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tests/test_datastores.py b/tests/test_datastores.py index b4342d5..5ce7bcd 100644 --- a/tests/test_datastores.py +++ b/tests/test_datastores.py @@ -10,7 +10,8 @@ from freezegun.api import FrozenDateTimeFactory from apscheduler.abc import AsyncDataStore, Job, Schedule from apscheduler.enums import CoalescePolicy, ConflictPolicy, JobOutcome -from apscheduler.events import Event, ScheduleAdded, ScheduleRemoved, ScheduleUpdated, TaskAdded +from apscheduler.events import ( + Event, ScheduleAdded, ScheduleRemoved, ScheduleUpdated, TaskAdded, TaskUpdated) from apscheduler.structures import JobResult, Task from apscheduler.triggers.date import DateTrigger @@ -56,7 +57,8 @@ class TestAsyncStores: self, datastore_cm: AsyncContextManager[AsyncDataStore]) -> None: import math - async with datastore_cm as store, capture_events(store, 3, {TaskAdded}) as events: + event_types = {TaskAdded, TaskUpdated} + async with datastore_cm as store, capture_events(store, 3, event_types) as events: await store.add_task(Task(id='test_task', func=print)) await store.add_task(Task(id='test_task2', func=math.ceil)) await store.add_task(Task(id='test_task', func=repr)) @@ -69,12 +71,15 @@ class TestAsyncStores: assert tasks[1].func is math.ceil received_event = events.pop(0) + assert isinstance(received_event, TaskAdded) assert received_event.task_id == 'test_task' received_event = events.pop(0) + assert isinstance(received_event, TaskAdded) assert received_event.task_id == 'test_task2' received_event = events.pop(0) + assert isinstance(received_event, TaskUpdated) assert received_event.task_id == 'test_task' assert not events |