diff options
Diffstat (limited to 'tests/alarm.py')
-rw-r--r-- | tests/alarm.py | 93 |
1 files changed, 0 insertions, 93 deletions
diff --git a/tests/alarm.py b/tests/alarm.py deleted file mode 100644 index 916f4405..00000000 --- a/tests/alarm.py +++ /dev/null @@ -1,93 +0,0 @@ -############################################################################## -# Copyright (c) 2017 ZTE Corporation and others. -# -# All rights reserved. This program and the accompanying materials -# are made available under the terms of the Apache License, Version 2.0 -# which accompanies this distribution, and is available at -# http://www.apache.org/licenses/LICENSE-2.0 -############################################################################## -from oslo_config import cfg - -from identity_auth import get_identity_auth -from identity_auth import get_session -from os_clients import aodh_client -from os_clients import nova_client - -OPTS = [ - cfg.StrOpt('alarm_basename', - default='doctor_alarm', - help='the base name of alarm', - required=True), -] - - -class Alarm(object): - - def __init__(self, conf, log): - self.conf = conf - self.log = log - self.auth = get_identity_auth(project=self.conf.doctor_project) - self.aodh = \ - aodh_client(conf.aodh_version, - get_session(auth=self.auth)) - self.nova = \ - nova_client(conf.nova_version, - get_session(auth=self.auth)) - self._init_alarm_name() - - def _init_alarm_name(self): - self.alarm_names = [] - for i in range(0, self.conf.instance_count): - alarm_name = '%s%d' % (self.conf.alarm_basename, i) - self.alarm_names.append(alarm_name) - - def create(self): - self.log.info('alarm create start......') - - alarms = {alarm['name']: alarm for alarm in self.aodh.alarm.list()} - servers = \ - {getattr(server, 'name'): server - for server in self.nova.servers.list()} - - for i in range(0, self.conf.instance_count): - alarm_name = self.alarm_names[i] - if alarm_name in alarms: - continue; - vm_name = '%s%d' % (self.conf.instance_basename, i) - vm_id = getattr(servers[vm_name], 'id') - alarm_request = dict( - name=alarm_name, - description=u'VM failure', - enabled=True, - alarm_actions=[u'http://%s:%d/failure' - % (self.conf.consumer.ip, - self.conf.consumer.port)], - repeat_actions=False, - severity=u'moderate', - type=u'event', - event_rule=dict( - event_type=u'compute.instance.update', - query=[ - dict(field=u'traits.instance_id', - type='', - op=u'eq', - value=vm_id), - dict(field=u'traits.state', - type='', - op=u'eq', - value=u'error')])) - self.aodh.alarm.create(alarm_request) - - self.log.info('alarm create end......') - - def delete(self): - self.log.info('alarm delete start.......') - - alarms = {alarm['name']: alarm for alarm in self.aodh.alarm.list()} - for alarm_name in self.alarm_names: - if alarm_name in alarms: - self.aodh.alarm.delete(alarms[alarm_name]['alarm_id']) - - del self.alarm_names[:] - - self.log.info('alarm delete end.......') |