82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
# -*- encoding: utf-8 -*-
|
|
#
|
|
# Copyright © 2013 Intel Corp
|
|
#
|
|
# Authors: Lianhao Lu <lianhao.lu@intel.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 fixtures
|
|
import mock
|
|
|
|
from ceilometer.central import manager
|
|
from ceilometer.hardware.inspector import base as inspector_base
|
|
from ceilometer.tests import base as test_base
|
|
|
|
|
|
class FakeInspector(inspector_base.Inspector):
|
|
CPU = inspector_base.CPUStats(cpu_1_min=0.99,
|
|
cpu_5_min=0.77,
|
|
cpu_15_min=0.55)
|
|
DISK = (inspector_base.Disk(device='/dev/sda1', path='/'),
|
|
inspector_base.DiskStats(size=1000, used=90))
|
|
MEMORY = inspector_base.MemoryStats(total=1000, used=90)
|
|
NET = (inspector_base.Interface(name='test.teest',
|
|
mac='001122334455',
|
|
ip='10.0.0.2'),
|
|
inspector_base.InterfaceStats(bandwidth=1000,
|
|
rx_bytes=90,
|
|
tx_bytes=80,
|
|
error=1))
|
|
|
|
def inspect_cpu(self, host):
|
|
yield self.CPU
|
|
|
|
def inspect_disk(self, host):
|
|
yield self.DISK
|
|
|
|
def inspect_memory(self, host):
|
|
yield self.MEMORY
|
|
|
|
def inspect_network(self, host):
|
|
yield self.NET
|
|
|
|
|
|
class TestPollsterBase(test_base.BaseTestCase):
|
|
def faux_get_inspector(url, namespace=None):
|
|
return FakeInspector()
|
|
|
|
def setUp(self):
|
|
super(TestPollsterBase, self).setUp()
|
|
self.host = ["test://test"]
|
|
self.useFixture(fixtures.MonkeyPatch(
|
|
'ceilometer.hardware.inspector.get_inspector',
|
|
self.faux_get_inspector))
|
|
|
|
@mock.patch('ceilometer.pipeline.setup_pipeline', mock.MagicMock())
|
|
def _check_get_samples(self, factory, name,
|
|
expected_value, expected_type):
|
|
mgr = manager.AgentManager()
|
|
pollster = factory()
|
|
cache = {}
|
|
samples = list(pollster.get_samples(mgr, cache, self.host))
|
|
self.assertTrue(samples)
|
|
self.assertIn(pollster.CACHE_KEY, cache)
|
|
self.assertIn(self.host[0], cache[pollster.CACHE_KEY])
|
|
|
|
self.assertEqual(set([name]),
|
|
set([s.name for s in samples]))
|
|
match = [s for s in samples if s.name == name]
|
|
self.assertEqual(expected_value, match[0].volume)
|
|
self.assertEqual(expected_type, match[0].type)
|