def test_cancelling_timer_calls_engine_remove_timer(self):
engine = Engine()
engine._remove_timer = MagicMock()
timer = _Timer(engine, None, None)
timer.cancel()
engine._remove_timer.assert_called_once_with(timer)
def test_comparing_two_timers_compares_end(self):
timer1 = _Timer(None, None, None, end=1)
timer2 = _Timer(None, None, None, end=2)
self.assertTrue(timer2 > timer1)
self.assertTrue(timer1 < timer2)
def test_calling_timer_calls_cancel(self):
timer = _Timer(None, None, None)
timer.cancel = MagicMock()
timer()
timer.cancel.assert_called_once_with()
def test_cancelling_timer_calls_engine_remove_timer(self):
engine = Engine()
engine._remove_timer = MagicMock()
timer = _Timer(engine, None, None)
timer.cancel()
engine._remove_timer.assert_called_once_with(timer)
def test_comparing_two_timers_compares_end(self):
timer1 = _Timer(None, None, None, end=1)
timer2 = _Timer(None, None, None, end=2)
self.assertTrue(timer2 > timer1)
self.assertTrue(timer1 < timer2)
def test_calling_timer_calls_cancel(self):
timer = _Timer(None, None, None)
timer.cancel = MagicMock()
timer()
timer.cancel.assert_called_once_with()
