def test_circuit_breaker_ramp_bogus_flow_rate(
self, mock_adjust, mock_get_flow, mock_describe_db_clusters):
my_alarm = {
"detail": {
"state": {
"value": "OK",
}
}
}
os.environ['STAGE'] = 'TEST'
mock_get_flow.return_value = 9000
with self.assertRaises(Exception) as context:
handler.circuit_breaker(my_alarm, self.context)
def test_circuit_breaker_ramp_10_to_10(
self, mock_adjust, mock_get_flow, mock_describe_db_clusters):
# Test where we ramp from zero
mock_get_flow.return_value = 10
my_alarm = {
"detail": {
"state": {
"value": "OK",
}
}
}
os.environ['STAGE'] = 'TEST'
mock_describe_db_clusters.return_value = DB['TEST']
handler.circuit_breaker(my_alarm, self.context)
mock_get_flow.assert_called_once()
mock_adjust.assert_not_called()
def test_circuit_breaker_ramp_5_to_0(
self, mock_adjust, mock_get_flow, mock_describe_db_clusters):
# Test where we ramp down from 5
mock_get_flow.return_value = 5
my_alarm = {
"detail": {
"state": {
"value": "ALARM",
}
}
}
os.environ['STAGE'] = 'TEST'
mock_describe_db_clusters.return_value = DB['TEST']
handler.circuit_breaker(my_alarm, self.context)
mock_get_flow.assert_called_once()
mock_adjust.assert_called_once_with(0)
def test_circuit_breaker_bogus_stage(
self, mock_adjust, mock_get_flow, mock_describe_db_clusters):
os.environ['STAGE'] = 'UNKNOWN'
with self.assertRaises(Exception) as context:
handler.circuit_breaker(self.initial_event, self.context)