def test_eventloop_schedule_action_periodic(self):
scheduler = EventLoopScheduler(exit_if_empty=False)
gate = threading.Semaphore(0)
period = 0.05
counter = 3
def action(state):
nonlocal counter
if state:
counter -= 1
return state - 1
if counter == 0:
gate.release()
disp = scheduler.schedule_periodic(period, action, counter)
def dispose(scheduler, state):
disp.dispose()
gate.release()
gate.acquire()
assert counter == 0
assert scheduler._has_thread() is True
scheduler.schedule(dispose)
gate.acquire()
assert scheduler._has_thread() is True
sleep(period)
scheduler.dispose()
sleep(period)
assert scheduler._has_thread() is False
def test_eventloop_schedule_action_periodic(self):
scheduler = EventLoopScheduler(exit_if_empty=False)
gate = threading.Semaphore(0)
period = 0.05
counter = 3
def action(state):
nonlocal counter
if state:
counter -= 1
return state - 1
if counter == 0:
gate.release()
disp = scheduler.schedule_periodic(period, action, counter)
def dispose(scheduler, state):
disp.dispose()
gate.release()
gate.acquire()
assert counter == 0
assert scheduler._has_thread() is True
scheduler.schedule(dispose)
gate.acquire()
assert scheduler._has_thread() is True
sleep(period)
scheduler.dispose()
sleep(period)
assert scheduler._has_thread() is False
def test_eventloop_schedule_periodic_dispose_error(self):
scheduler = EventLoopScheduler(exit_if_empty=False)
scheduler.dispose()
ran = False
def action(scheduler, state):
nonlocal ran
ran = True
with pytest.raises(DisposedException):
scheduler.schedule_periodic(0.1, action)
assert ran is False
assert scheduler._has_thread() is False
def test_eventloop_schedule_periodic_dispose_error(self):
scheduler = EventLoopScheduler(exit_if_empty=False)
scheduler.dispose()
ran = False
def action(scheduler, state):
nonlocal ran
ran = True
with pytest.raises(DisposedException):
scheduler.schedule_periodic(0.1, action)
assert ran is False
assert scheduler._has_thread() is False
def test_eventloop_schedule_periodic_dispose(self):
scheduler = EventLoopScheduler(exit_if_empty=False)
scheduler.dispose()
ran = False
def action(scheduler, state):
nonlocal ran
ran = True
exc = None
try:
scheduler.schedule_periodic(0.1, scheduler.now, action)
except Exception as e:
exc = e
finally:
assert isinstance(exc, DisposedException)
assert ran is False
assert scheduler._has_thread() is False
