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)