def test_to_unknown(self): self._set_all_alarms('ok') with mock.patch('ceilometerclient.client.get_client', return_value=self.api_client): broken = exc.CommunicationError(message='broken') self.api_client.alarms.get.side_effect = [ broken, self._get_alarm('ok'), self._get_alarm('insufficient data'), self._get_alarm('ok'), ] self._evaluate_all_alarms() expected = [mock.call(alarm.alarm_id, state='insufficient data') for alarm in self.alarms] update_calls = self.api_client.alarms.set_state.call_args_list self.assertEqual(expected, update_calls) reasons = ['Alarms %s are in unknown state' % self.alarms[0].rule['alarm_ids'][0], 'Alarms %s are in unknown state' % self.alarms[1].rule['alarm_ids'][0]] reason_datas = [ self._reason_data([self.alarms[0].rule['alarm_ids'][0]]), self._reason_data([self.alarms[1].rule['alarm_ids'][0]])] expected = [mock.call(alarm, 'ok', reason, reason_data) for alarm, reason, reason_data in zip(self.alarms, reasons, reason_datas)] self.assertEqual(expected, self.notifier.notify.call_args_list)
def test_to_unknown(self): self._set_all_alarms('ok') with mock.patch('ceilometerclient.client.get_client', return_value=self.api_client): broken = exc.CommunicationError(message='broken') self.api_client.alarms.get.side_effect = [ broken, self._get_alarm('ok'), self._get_alarm('insufficient data'), self._get_alarm('ok'), ] self._evaluate_all_alarms() expected = [ mock.call(alarm.alarm_id, state='insufficient data') for alarm in self.alarms ] update_calls = self.api_client.alarms.set_state.call_args_list self.assertEqual(update_calls, expected) reasons = [ '1 alarms in' ' 9cfc3e51-2ff1-4b1d-ac01-c1bd4c6d0d1e,' '1d441595-d069-4e05-95ab-8693ba6a8302' ' are in unknown state', '1 alarms in' ' b82734f4-9d06-48f3-8a86-fa59a0c99dc8,' '15a700e5-2fe8-4b3d-8c55-9e92831f6a2b' ' are in unknown state' ] expected = [ mock.call(alarm, 'ok', reason) for alarm, reason in zip(self.alarms, reasons) ] self.assertEqual(self.notifier.notify.call_args_list, expected)
def test_to_unknown(self): self._set_all_alarms('ok') broken = exc.CommunicationError(message='broken') self.storage_conn.get_alarms.side_effect = [ broken, self._get_alarms('ok'), self._get_alarms('insufficient data'), self._get_alarms('ok'), ] self._evaluate_all_alarms() expected = [mock.call(alarm) for alarm in self.alarms] update_calls = self.storage_conn.update_alarm.call_args_list self.assertEqual(expected, update_calls) reasons = [ 'Alarms %s are in unknown state' % self.alarms[0].rule['alarm_ids'][0], 'Alarms %s are in unknown state' % self.alarms[1].rule['alarm_ids'][0] ] reason_datas = [ self._reason_data([self.alarms[0].rule['alarm_ids'][0]]), self._reason_data([self.alarms[1].rule['alarm_ids'][0]]) ] expected = [ mock.call(alarm, 'ok', reason, reason_data) for alarm, reason, reason_data in zip( self.alarms, reasons, reason_datas) ] self.assertEqual(expected, self.notifier.notify.call_args_list)
def _http_request(self, url, method, **kwargs): """Send an http request with the specified characteristics. Wrapper around httplib.HTTP(S)Connection.request to handle tasks such as setting headers and error handling. """ # Copy the kwargs so we can reuse the original in case of redirects kwargs['headers'] = copy.deepcopy(kwargs.get('headers', {})) kwargs['headers'].setdefault('User-Agent', USER_AGENT) auth_token = self.auth_token() if auth_token: kwargs['headers'].setdefault('X-Auth-Token', auth_token) self.log_curl_request(method, url, kwargs) conn = self.get_connection() try: if self.proxy_url: conn_url = (self.endpoint.rstrip('/') + self._make_connection_url(url)) else: conn_url = self._make_connection_url(url) conn.request(method, conn_url, **kwargs) resp = conn.getresponse() except socket.gaierror as e: message = ("Error finding address for %(url)s: %(e)s" % dict(url=url, e=e)) raise exc.InvalidEndpoint(message=message) except (socket.error, socket.timeout) as e: endpoint = self.endpoint message = ("Error communicating with %(endpoint)s %(e)s" % dict(endpoint=endpoint, e=e)) raise exc.CommunicationError(message=message) body_iter = ResponseBodyIterator(resp) # Read body into string if it isn't obviously image data if resp.getheader('content-type', None) != 'application/octet-stream': body_str = ''.join([chunk for chunk in body_iter]) self.log_http_response(resp, body_str) body_iter = six.StringIO(body_str) else: self.log_http_response(resp) if 400 <= resp.status < 600: LOG.warn("Request returned failure status.") raise exc.from_response(resp, ''.join(body_iter)) elif resp.status in (301, 302, 305): # Redirected. Reissue the request to the new location. return self._http_request(resp['location'], method, **kwargs) elif resp.status == 300: raise exc.from_response(resp) return resp, body_iter
def test_retry_transient_api_failure(self): broken = exc.CommunicationError(message='broken') avgs = [self._get_stat('avg', self.alarms[0].rule['threshold'] - v) for v in moves.xrange(5)] maxs = [self._get_stat('max', self.alarms[1].rule['threshold'] + v) for v in moves.xrange(1, 5)] self.api_client.statistics.list.side_effect = [broken, broken, avgs, maxs] self._evaluate_all_alarms() self._assert_all_alarms('insufficient data') self._evaluate_all_alarms() self._assert_all_alarms('ok')
def test_simple_insufficient(self): self._set_all_alarms('ok') broken = exc.CommunicationError(message='broken') self.storage_conn.get_alarms.side_effect = broken self._evaluate_all_alarms() self._assert_all_alarms('insufficient data') expected = [mock.call(alarm) for alarm in self.alarms] update_calls = self.storage_conn.update_alarm.call_args_list self.assertEqual(expected, update_calls) expected = [ mock.call(alarm, 'ok', ('Alarms %s are in unknown state' % (",".join(alarm.rule['alarm_ids']))), self._reason_data(alarm.rule['alarm_ids'])) for alarm in self.alarms ] self.assertEqual(expected, self.notifier.notify.call_args_list)
def test_retry_transient_api_failure(self): broken = exc.CommunicationError(message='broken') self.storage_conn.get_alarms.side_effect = [ broken, broken, broken, broken, self._get_alarms('ok'), self._get_alarms('ok'), self._get_alarms('ok'), self._get_alarms('ok'), ] self._evaluate_all_alarms() self._assert_all_alarms('insufficient data') self._evaluate_all_alarms() self._assert_all_alarms('ok')
def test_retry_transient_api_failure(self): with mock.patch('ceilometerclient.client.get_client', return_value=self.api_client): broken = exc.CommunicationError(message='broken') self.api_client.alarms.get.side_effect = [ broken, broken, broken, broken, self._get_alarm('ok'), self._get_alarm('ok'), self._get_alarm('ok'), self._get_alarm('ok'), ] self._evaluate_all_alarms() self._assert_all_alarms('insufficient data') self._evaluate_all_alarms() self._assert_all_alarms('ok')
def test_retry_transient_api_failure(self): with mock.patch('ceilometerclient.client.get_client', return_value=self.api_client): broken = exc.CommunicationError(message='broken') avgs = [ self._get_stat('avg', self.alarms[0].threshold - v) for v in xrange(5) ] maxs = [ self._get_stat('max', self.alarms[1].threshold + v) for v in xrange(1, 4) ] self.api_client.statistics.list.side_effect = [ broken, broken, avgs, maxs ] self.evaluator.evaluate() self._assert_all_alarms('insufficient data') self.evaluator.evaluate() self._assert_all_alarms('ok')
def test_simple_insufficient(self): self._set_all_alarms('ok') with mock.patch('ceilometerclient.client.get_client', return_value=self.api_client): broken = exc.CommunicationError(message='broken') self.api_client.alarms.get.side_effect = broken self._evaluate_all_alarms() self._assert_all_alarms('insufficient data') expected = [mock.call(alarm.alarm_id, state='insufficient data') for alarm in self.alarms] update_calls = self.api_client.alarms.set_state.call_args_list self.assertEqual(expected, update_calls) expected = [mock.call( alarm, 'ok', ('Alarms %s are in unknown state' % (",".join(alarm.rule['alarm_ids']))), self._reason_data(alarm.rule['alarm_ids'])) for alarm in self.alarms] self.assertEqual(expected, self.notifier.notify.call_args_list)