def test_function(self): timer_task = mock.MagicMock() timer = ResettableTimer(10, timer_task) timer.start() # wait for it timer.join() self.assertEqual(1, timer_task.call_count)
def test_reset_callback(self): timer_task = mock.MagicMock() on_reset_cb = mock.MagicMock() timer = ResettableTimer(10, timer_task, on_reset=on_reset_cb) timer.start() timer.reset() # wait for it timer.join() self.assertEqual(1, timer_task.call_count) self.assertEqual(1, on_reset_cb.call_count)