96d16ad490
I don't know what I was thinking with this typo: s/ordination/coordination/ Change-Id: Iad5d159372b335a27cd7fe4b13c136617c0b0d10
241 lines
8.9 KiB
Python
241 lines
8.9 KiB
Python
#
|
|
# Copyright 2013 eNovance <licensing@enovance.com>
|
|
#
|
|
# Authors: Mehdi Abaakouk <mehdi.abaakouk@enovance.com>
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import uuid
|
|
|
|
from ceilometerclient.v2 import alarms
|
|
import eventlet
|
|
|
|
from ceilometer.alarm import rpc as rpc_alarm
|
|
from ceilometer.alarm.storage import models
|
|
from ceilometer import messaging
|
|
from ceilometer.openstack.common.fixture import config
|
|
from ceilometer.openstack.common import timeutils
|
|
from ceilometer.tests import base as tests_base
|
|
|
|
|
|
class FakeNotifier(object):
|
|
def __init__(self, transport):
|
|
self.rpc = messaging.get_rpc_server(
|
|
transport, "alarm_notifier", self)
|
|
self.notified = []
|
|
|
|
def start(self, expected_length):
|
|
self.expected_length = expected_length
|
|
self.rpc.start()
|
|
|
|
def notify_alarm(self, context, data):
|
|
self.notified.append(data)
|
|
if len(self.notified) == self.expected_length:
|
|
self.rpc.stop()
|
|
|
|
|
|
class TestRPCAlarmNotifier(tests_base.BaseTestCase):
|
|
def setUp(self):
|
|
super(TestRPCAlarmNotifier, self).setUp()
|
|
self.CONF = self.useFixture(config.Config()).conf
|
|
self.setup_messaging(self.CONF)
|
|
|
|
self.notifier_server = FakeNotifier(self.transport)
|
|
self.notifier = rpc_alarm.RPCAlarmNotifier()
|
|
self.alarms = [
|
|
alarms.Alarm(None, info={
|
|
'name': 'instance_running_hot',
|
|
'meter_name': 'cpu_util',
|
|
'comparison_operator': 'gt',
|
|
'threshold': 80.0,
|
|
'evaluation_periods': 5,
|
|
'statistic': 'avg',
|
|
'state': 'ok',
|
|
'ok_actions': ['http://host:8080/path'],
|
|
'user_id': 'foobar',
|
|
'project_id': 'snafu',
|
|
'period': 60,
|
|
'alarm_id': str(uuid.uuid4()),
|
|
'matching_metadata':{'resource_id':
|
|
'my_instance'}
|
|
}),
|
|
alarms.Alarm(None, info={
|
|
'name': 'group_running_idle',
|
|
'meter_name': 'cpu_util',
|
|
'comparison_operator': 'le',
|
|
'threshold': 10.0,
|
|
'statistic': 'max',
|
|
'evaluation_periods': 4,
|
|
'state': 'insufficient data',
|
|
'insufficient_data_actions': ['http://other_host/path'],
|
|
'user_id': 'foobar',
|
|
'project_id': 'snafu',
|
|
'period': 300,
|
|
'alarm_id': str(uuid.uuid4()),
|
|
'matching_metadata':{'metadata.user_metadata.AS':
|
|
'my_group'}
|
|
}),
|
|
]
|
|
|
|
def test_rpc_target(self):
|
|
topic = self.notifier.client.target.topic
|
|
self.assertEqual('alarm_notifier', topic)
|
|
|
|
def test_notify_alarm(self):
|
|
self.notifier_server.start(2)
|
|
|
|
previous = ['alarm', 'ok']
|
|
for i, a in enumerate(self.alarms):
|
|
self.notifier.notify(a, previous[i], "what? %d" % i,
|
|
{'fire': '%d' % i})
|
|
|
|
self.notifier_server.rpc.wait()
|
|
|
|
self.assertEqual(2, len(self.notifier_server.notified))
|
|
for i, a in enumerate(self.alarms):
|
|
actions = getattr(a, models.Alarm.ALARM_ACTIONS_MAP[a.state])
|
|
self.assertEqual(self.alarms[i].alarm_id,
|
|
self.notifier_server.notified[i]["alarm_id"])
|
|
self.assertEqual(actions,
|
|
self.notifier_server.notified[i]["actions"])
|
|
self.assertEqual(previous[i],
|
|
self.notifier_server.notified[i]["previous"])
|
|
self.assertEqual(self.alarms[i].state,
|
|
self.notifier_server.notified[i]["current"])
|
|
self.assertEqual("what? %d" % i,
|
|
self.notifier_server.notified[i]["reason"])
|
|
self.assertEqual({'fire': '%d' % i},
|
|
self.notifier_server.notified[i]["reason_data"])
|
|
|
|
def test_notify_non_string_reason(self):
|
|
self.notifier_server.start(1)
|
|
self.notifier.notify(self.alarms[0], 'ok', 42, {})
|
|
self.notifier_server.rpc.wait()
|
|
reason = self.notifier_server.notified[0]['reason']
|
|
self.assertIsInstance(reason, basestring)
|
|
|
|
def test_notify_no_actions(self):
|
|
alarm = alarms.Alarm(None, info={
|
|
'name': 'instance_running_hot',
|
|
'meter_name': 'cpu_util',
|
|
'comparison_operator': 'gt',
|
|
'threshold': 80.0,
|
|
'evaluation_periods': 5,
|
|
'statistic': 'avg',
|
|
'state': 'ok',
|
|
'user_id': 'foobar',
|
|
'project_id': 'snafu',
|
|
'period': 60,
|
|
'ok_actions': [],
|
|
'alarm_id': str(uuid.uuid4()),
|
|
'matching_metadata': {'resource_id':
|
|
'my_instance'}
|
|
})
|
|
self.notifier.notify(alarm, 'alarm', "what?", {})
|
|
self.assertEqual(0, len(self.notifier_server.notified))
|
|
|
|
|
|
class FakeCoordinator(object):
|
|
def __init__(self, transport):
|
|
self.rpc = messaging.get_rpc_server(
|
|
transport, "alarm_partition_coordination", self)
|
|
self.notified = []
|
|
|
|
def presence(self, context, data):
|
|
self._record('presence', data)
|
|
|
|
def allocate(self, context, data):
|
|
self._record('allocate', data)
|
|
|
|
def assign(self, context, data):
|
|
self._record('assign', data)
|
|
|
|
def _record(self, method, data):
|
|
self.notified.append((method, data))
|
|
self.rpc.stop()
|
|
|
|
|
|
class TestRPCAlarmPartitionCoordination(tests_base.BaseTestCase):
|
|
def setUp(self):
|
|
super(TestRPCAlarmPartitionCoordination, self).setUp()
|
|
self.CONF = self.useFixture(config.Config()).conf
|
|
self.setup_messaging(self.CONF)
|
|
|
|
self.coordinator_server = FakeCoordinator(self.transport)
|
|
self.coordinator_server.rpc.start()
|
|
eventlet.sleep() # must be sure that fanout queue is created
|
|
|
|
self.coordination = rpc_alarm.RPCAlarmPartitionCoordination()
|
|
self.alarms = [
|
|
alarms.Alarm(None, info={
|
|
'name': 'instance_running_hot',
|
|
'meter_name': 'cpu_util',
|
|
'comparison_operator': 'gt',
|
|
'threshold': 80.0,
|
|
'evaluation_periods': 5,
|
|
'statistic': 'avg',
|
|
'state': 'ok',
|
|
'ok_actions': ['http://host:8080/path'],
|
|
'user_id': 'foobar',
|
|
'project_id': 'snafu',
|
|
'period': 60,
|
|
'alarm_id': str(uuid.uuid4()),
|
|
'matching_metadata':{'resource_id':
|
|
'my_instance'}
|
|
}),
|
|
alarms.Alarm(None, info={
|
|
'name': 'group_running_idle',
|
|
'meter_name': 'cpu_util',
|
|
'comparison_operator': 'le',
|
|
'threshold': 10.0,
|
|
'statistic': 'max',
|
|
'evaluation_periods': 4,
|
|
'state': 'insufficient data',
|
|
'insufficient_data_actions': ['http://other_host/path'],
|
|
'user_id': 'foobar',
|
|
'project_id': 'snafu',
|
|
'period': 300,
|
|
'alarm_id': str(uuid.uuid4()),
|
|
'matching_metadata':{'metadata.user_metadata.AS':
|
|
'my_group'}
|
|
}),
|
|
]
|
|
|
|
def test_coordination_presence(self):
|
|
id = str(uuid.uuid4())
|
|
priority = float(timeutils.utcnow().strftime('%s.%f'))
|
|
self.coordination.presence(id, priority)
|
|
self.coordinator_server.rpc.wait()
|
|
method, args = self.coordinator_server.notified[0]
|
|
self.assertEqual(id, args['uuid'])
|
|
self.assertEqual(priority, args['priority'])
|
|
self.assertEqual('presence', method)
|
|
|
|
def test_coordination_assign(self):
|
|
id = str(uuid.uuid4())
|
|
self.coordination.assign(id, self.alarms)
|
|
self.coordinator_server.rpc.wait()
|
|
method, args = self.coordinator_server.notified[0]
|
|
self.assertEqual(id, args['uuid'])
|
|
self.assertEqual(2, len(args['alarms']))
|
|
self.assertEqual('assign', method)
|
|
|
|
def test_coordination_allocate(self):
|
|
id = str(uuid.uuid4())
|
|
self.coordination.allocate(id, self.alarms)
|
|
self.coordinator_server.rpc.wait()
|
|
method, args = self.coordinator_server.notified[0]
|
|
self.assertEqual(id, args['uuid'])
|
|
self.assertEqual(2, len(args['alarms']))
|
|
self.assertEqual('allocate', method)
|