def test_raising(self, mock_alarm_manager):
alarm = TooLongAlarm(0.1)
alarm.trigger("a")
sleep(0.3)
mock_alarm = mock_alarm_manager.get_alarm.return_value
mock_alarm.set.assert_called_once_with()
alarm.quit()
def test_cancellation(self, mock_alarm_manager):
alarm = TooLongAlarm(0.1)
alarm.trigger("c")
alarm.cancel()
sleep(0.3)
mock_alarm = mock_alarm_manager.get_alarm.return_value
mock_alarm.clear.assert_called_once_with()
alarm.quit()
def test_not_triggered_early(self, mock_alarm_manager):
mock_alarm = mock_alarm_manager.get_alarm.return_value
alarm = TooLongAlarm(0.1)
alarm.trigger("b")
self.assertEqual([], mock_alarm.set.call_args_list)
sleep(0.3)
mock_alarm.set.assert_called_once_with()
alarm.quit()
def test_correct_alarm(self, mock_alarm_manager):
alarm = TooLongAlarm(0.1)
mock_get_alarm = mock_alarm_manager.get_alarm
mock_get_alarm.assert_called_once_with('cluster-manager',
TOO_LONG_CLUSTERING)
alarm.quit()
