1 files changed, 93 insertions, 0 deletions
diff --git a/doctor_tests/alarm.py b/doctor_tests/alarm.py
new file mode 100644
index 00000000..3b1aaf3f
--- /dev/null
+++ b/doctor_tests/alarm.py
@@ -0,0 +1,93 @@
+##############################################################################
+# 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 doctor_tests.identity_auth import get_identity_auth
+from doctor_tests.identity_auth import get_session
+from doctor_tests.os_clients import aodh_client
+from doctor_tests.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.......')