def test_schedule_in_invalid_time_throws_bad_arg(self):
async def foo():
pass
s = Scheduler()
with pytest.raises(BadArgument):
s.schedule_in(foo, time=-1)
async def get_task_valid_task():
s = Scheduler()
t_id = s.schedule_in(foo, time=1)
s.cancel(t_id)
assert len(s._scheduled_tasks) == 0
async def get_task_valid_task():
s = Scheduler()
t_id = s.schedule_in(foo, time=1)
assert s.get_task(t_id) is not None
async def valid_time_test():
s = Scheduler()
s.schedule_in(foo, time=1)
assert len(s._scheduled_tasks) == 1
def test_schedule_in_null_callback_throws_bad_arg(self):
s = Scheduler()
with pytest.raises(BadArgument):
s.schedule_in(None, time=datetime(1, 1, 1, 0, 0))
async def get_task_valid_task():
s = Scheduler()
s.schedule_in(foo, time=1)
with pytest.raises(KeyError):
s.cancel(1)