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_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