def test_timer_can_schedule_callback():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb = Callback()
timer.schedule(5, cb)
assert cb.call_count == 0
timer.service()
assert cb.call_count == 0
ts.value += 3
timer.service()
assert cb.call_count == 0
ts.value += 3
timer.service()
assert cb.call_count == 1
timer.service()
assert cb.call_count == 1
ts.value += 6
timer.service()
assert cb.call_count == 1
def test_timer_can_schedule_and_process_callback_twice():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb = Callback()
timer.schedule(3, cb)
timer.schedule(5, cb)
ts.value += 6
timer.service()
assert cb.call_count == 2
def test_timer_triggers_callback_on_time():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb = Callback()
timer.schedule(5, cb)
assert cb.call_count == 0
ts.value += 5
timer.service()
assert cb.call_count == 1
def test_timer_cancels_all_instances_of_callback():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb = Callback()
timer.schedule(5, cb)
timer.schedule(3, cb)
timer.cancel(cb)
ts.value += 6
timer.service()
assert cb.call_count == 0
def test_timer_can_schedule_and_simultaneously_process_different_callbacks():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb1 = Callback()
cb2 = Callback()
timer.schedule(5, cb1)
timer.schedule(3, cb2)
ts.value += 6
timer.service()
assert cb1.call_count == 1
assert cb2.call_count == 1
def test_timer_can_schedule_different_callbacks_on_same_time_twice():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb1 = Callback()
cb2 = Callback()
timer.schedule(5, cb1)
timer.schedule(5, cb2)
ts.value += 6
timer.service()
assert cb1.call_count == 1
assert cb2.call_count == 1
def test_timer_cancel_callback_doesnt_crash_for_nonexistant_callback():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb = Callback()
# This shouldn't crash
timer.cancel(cb)
# Make sure that callback which was scheduled later is still called
timer.schedule(5, cb)
ts.value += 6
timer.service()
assert cb.call_count == 1
# And this still shouldn't crash
timer.cancel(cb)
def test_timer_can_cancel_callback_without_touching_other_callbacks():
ts = MockTimestamp(0)
timer = QueueTimer(ts)
cb1 = Callback()
cb2 = Callback()
cb3 = Callback()
timer.schedule(5, cb1)
timer.schedule(3, cb2)
timer.schedule(4, cb3)
timer.cancel(cb2)
ts.value += 6
timer.service()
assert cb1.call_count == 1
assert cb2.call_count == 0
assert cb3.call_count == 1