def _run_alarm_processor(self, alarm, sql_response, mock_log, mock_mysql): """Runs a mocked alarm processor reading from queue while running, returns (queue_message, log_message) """ # Since the log runs in another thread I can mock it directly, instead # change the methods to put to a queue mock_log.warn = self.trap.append mock_log.error = self.trap.append mock_log.exception = self.trap.append # Setup the sql response if sql_response is not None: mock_mysql.return_value = mock_mysql mock_mysql.cursor.return_value = mock_mysql mock_mysql.__iter__.return_value = sql_response self.conf_override(group='mysql', ssl=None, host='localhost', port='3306', user='******', db='dbname', passwd='mysql_passwd') self.conf_override(group='statsd', host='localhost', port=8125) processor = alarm_processor.AlarmProcessor() return processor.to_notification(alarm)
def _run_alarm_processor(self, alarm, sql_response, mock_log, mock_mysql): """Runs a mocked alarm processor reading from queue while running, returns (queue_message, log_message) """ # Since the log runs in another thread I can mock it directly, instead change the methods to put to a queue mock_log.warn = self.trap.append mock_log.error = self.trap.append mock_log.exception = self.trap.append # Setup the sql response if sql_response is not None: mock_mysql.return_value = mock_mysql mock_mysql.cursor.return_value = mock_mysql mock_mysql.__iter__.return_value = sql_response mysql_config = { 'mysql': { 'ssl': None, 'host': 'mysql_host', 'port': 'mysql_port', 'user': '******', 'db': 'dbname', 'passwd': 'mysql_passwd' } } processor = alarm_processor.AlarmProcessor(600, mysql_config) return processor.to_notification(alarm)
def __init__(self): self._statsd = get_statsd_client() self._consumer = consumer.KafkaConsumer( CONF.kafka.url, ','.join(CONF.zookeeper.url), CONF.zookeeper.notification_path, CONF.kafka.group, CONF.kafka.alarm_topic) self._producer = producer.KafkaProducer(CONF.kafka.url) self._alarms = ap.AlarmProcessor() self._notifier = np.NotificationProcessor()
def __init__(self): self._statsd = get_statsd_client() self._consumer = client_factory.get_kafka_consumer( CONF.kafka.url, CONF.kafka.group, CONF.kafka.alarm_topic, CONF.zookeeper.url, CONF.zookeeper.notification_path, CONF.kafka.legacy_kafka_client_enabled) self._producer = client_factory.get_kafka_producer( CONF.kafka.url, CONF.kafka.legacy_kafka_client_enabled) self._alarms = ap.AlarmProcessor() self._notifier = np.NotificationProcessor()