From 7e0ab6e2149df22455e58eb363eb291a985514de Mon Sep 17 00:00:00 2001 From: dongwenjuan Date: Tue, 16 May 2017 15:46:04 +0800 Subject: add create alarm JIRA: DOCTOR-93 Change-Id: I316608c9627ff38e55154257b798244de3f221c5 Signed-off-by: dongwenjuan --- tests/alarm.py | 95 ++++++++++++++++++++++++++++++++++++++++++++++ tests/config.py | 6 ++- tests/consumer/__init__.py | 26 +++++++++++++ tests/main.py | 6 +++ tests/os_clients.py | 6 +++ 5 files changed, 138 insertions(+), 1 deletion(-) create mode 100644 tests/alarm.py create mode 100644 tests/consumer/__init__.py (limited to 'tests') diff --git a/tests/alarm.py b/tests/alarm.py new file mode 100644 index 00000000..0582085e --- /dev/null +++ b/tests/alarm.py @@ -0,0 +1,95 @@ +############################################################################## +# 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(username=self.conf.doctor_user, + password=self.conf.doctor_passwd, + 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.......') diff --git a/tests/config.py b/tests/config.py index 99f335f2..94b85f7f 100644 --- a/tests/config.py +++ b/tests/config.py @@ -10,6 +10,8 @@ import itertools from oslo_config import cfg +import alarm +import consumer import image import instance import network @@ -19,12 +21,14 @@ import user def list_opts(): return [ + ('consumer', consumer.OPTS), ('DEFAULT', itertools.chain( os_clients.OPTS, image.OPTS, user.OPTS, network.OPTS, - instance.OPTS)) + instance.OPTS, + alarm.OPTS)) ] diff --git a/tests/consumer/__init__.py b/tests/consumer/__init__.py new file mode 100644 index 00000000..68cc5dc9 --- /dev/null +++ b/tests/consumer/__init__.py @@ -0,0 +1,26 @@ +############################################################################## +# 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 + + +OPTS = [ + cfg.StrOpt('type', + default='sample', + choices=['sample'], + help='the component of doctor consumer', + required=True), + cfg.StrOpt('ip', + default='127.0.0.1', + help='the ip of consumer', + required=True), + cfg.IntOpt('port', + default='12346', + help='the port of doctor consumer', + required=True), +] diff --git a/tests/main.py b/tests/main.py index 540a9244..45683d1e 100644 --- a/tests/main.py +++ b/tests/main.py @@ -10,6 +10,7 @@ import os from os.path import isfile, join import sys +from alarm import Alarm import config from image import Image from instance import Instance @@ -28,6 +29,7 @@ class DoctorTest(object): self.user = User(self.conf, LOG) self.network = Network(self.conf, LOG) self.instance = Instance(self.conf, LOG) + self.alarm = Alarm(self.conf, LOG) def setup(self): # prepare the cloud env @@ -44,6 +46,9 @@ class DoctorTest(object): self.instance.create() self.instance.wait_for_vm_launch() + # creating alarm... + self.alarm.create() + def run(self): """run doctor test""" try: @@ -62,6 +67,7 @@ class DoctorTest(object): self.cleanup() def cleanup(self): + self.alarm.delete() self.instance.delete() self.network.delete() self.image.delete() diff --git a/tests/os_clients.py b/tests/os_clients.py index 5f6d452e..2699930f 100644 --- a/tests/os_clients.py +++ b/tests/os_clients.py @@ -8,6 +8,7 @@ ############################################################################## from oslo_config import cfg +import aodhclient.client as aodhclient import glanceclient.client as glanceclient from keystoneclient.v2_0 import client as ks_client from neutronclient.v2_0 import client as neutronclient @@ -17,6 +18,7 @@ import novaclient.client as novaclient OPTS = [ cfg.StrOpt('glance_version', default='2', help='glance version'), cfg.StrOpt('nova_version', default='2.34', help='Nova version'), + cfg.StrOpt('aodh_version', default='2', help='aodh version'), ] @@ -36,3 +38,7 @@ def nova_client(version, session): def neutron_client(session): return neutronclient.Client(session=session) + + +def aodh_client(version, session): + return aodhclient.Client(version, session=session) -- cgit 1.2.3-korg