a10773e413
Add a dictionary as a data cache for the pollsters, creating it outside of the loop where they are invoked and passing it to them. Future changesets will include changes to the pollsters to put data in the cache. blueprint one-meter-per-plugin Change-Id: Ie65526dfe65a8880ad8683b62fae62f8e7f9e69b Signed-off-by: Doug Hellmann <doug.hellmann@dreamhost.com>
110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Author: François Rossigneux <francois.rossigneux@inria.fr>
|
|
#
|
|
# 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 datetime
|
|
import mock
|
|
|
|
from ceilometer.tests import base
|
|
from ceilometer.energy import kwapi
|
|
from ceilometer.central import manager
|
|
from ceilometer.openstack.common import context
|
|
|
|
from keystoneclient import exceptions
|
|
|
|
PROBE_DICT = {
|
|
"probes": {
|
|
"A": {
|
|
"timestamp": 1357730232.68754,
|
|
"w": 107.3,
|
|
"kwh": 0.001058255421506034
|
|
},
|
|
"B": {
|
|
"timestamp": 1357730232.048158,
|
|
"w": 15.0,
|
|
"kwh": 0.029019045026169896
|
|
},
|
|
"C": {
|
|
"timestamp": 1357730232.223375,
|
|
"w": 95.0,
|
|
"kwh": 0.17361822634312918
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
class TestManager(manager.AgentManager):
|
|
|
|
def __init__(self):
|
|
super(TestManager, self).__init__()
|
|
self.keystone = None
|
|
|
|
|
|
class TestKwapiPollster(base.TestCase):
|
|
|
|
@staticmethod
|
|
def fake_kwapi_iter_probes(self, ksclient):
|
|
probes = PROBE_DICT['probes']
|
|
for key, value in probes.iteritems():
|
|
probe_dict = value
|
|
probe_dict['id'] = key
|
|
yield probe_dict
|
|
|
|
@staticmethod
|
|
def fake_kwapi_get_kwapi_client(self, ksclient):
|
|
raise exceptions.EndpointNotFound("fake keystone exception")
|
|
|
|
@mock.patch('ceilometer.pipeline.setup_pipeline', mock.MagicMock())
|
|
def setUp(self):
|
|
super(TestKwapiPollster, self).setUp()
|
|
self.context = context.get_admin_context()
|
|
self.manager = TestManager()
|
|
|
|
def test_kwapi_endpoint_not_exist(self):
|
|
self.stubs.Set(kwapi._Base, 'get_kwapi_client',
|
|
self.fake_kwapi_get_kwapi_client)
|
|
|
|
counters = list(kwapi.KwapiPollster().get_counters(self.manager, {}))
|
|
self.assertEqual(len(counters), 0)
|
|
|
|
def test_kwapi_counter(self):
|
|
self.stubs.Set(kwapi._Base, 'iter_probes', self.fake_kwapi_iter_probes)
|
|
|
|
counters = list(kwapi.KwapiPollster().get_counters(self.manager, {}))
|
|
self.assertEqual(len(counters), 6)
|
|
energy_counters = [counter for counter in counters
|
|
if counter.name == "energy"]
|
|
power_counters = [counter for counter in counters
|
|
if counter.name == "power"]
|
|
for probe in PROBE_DICT['probes'].values():
|
|
self.assert_(
|
|
any(map(lambda counter: counter.timestamp ==
|
|
datetime.datetime.fromtimestamp(
|
|
probe['timestamp']).isoformat(),
|
|
counters)))
|
|
self.assert_(
|
|
any(map(lambda counter: counter.volume == probe['kwh'],
|
|
energy_counters)))
|
|
self.assert_(
|
|
any(map(lambda counter: counter.volume == probe['w'],
|
|
power_counters)))
|
|
|
|
def test_kwapi_counter_list(self):
|
|
self.stubs.Set(kwapi._Base, 'iter_probes', self.fake_kwapi_iter_probes)
|
|
|
|
counters = list(kwapi.KwapiPollster().get_counters(self.manager, {}))
|
|
self.assertEqual(set([c.name for c in counters]),
|
|
set(kwapi.KwapiPollster().get_counter_names()))
|