Emma Foley 53bd946ff7 Refactor directory structure for multiple plugins
- Add collectd_ceilometer.ceilometer dir
- Add collectd_ceilometer.common dir
- Move test files to collectd_ceilometer/tests/{ceilometer,common}
- Add base code to collectd_ceilometer/tests

Change-Id: I5bd0a8cfe8eb55407f3d6827c0bb4b509dae8c8c
2016-12-05 18:07:32 +00:00

98 lines
3.1 KiB
Python

# -*- coding: utf-8 -*-
# Copyright 2010-2011 OpenStack Foundation
# Copyright (c) 2015 Intel Corporation.
#
# 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.
"""Plugin tests"""
from __future__ import unicode_literals
from collectd_ceilometer.common.meters.base import Meter
from collectd_ceilometer.tests.base import TestCase
import mock
class Values(object):
"""Stub class to replace collectd.Values"""
def __init__(self, plugin="my_plugin", type="my_type"):
self.plugin = plugin
self.type = type
class CollectdMock(object):
"""Model for the collectd class to be mocked"""
def get_dataset(self, string):
pass
collectd_class = 'collectd_ceilometer.tests.test_meters_base.CollectdMock'
class MetersTest(TestCase):
"""Test the meters/base.py class"""
@mock.patch(collectd_class, spec=True)
def setUp(self, collectd):
super(MetersTest, self).setUp()
self._collectd = collectd
# need this as a parameter for sample_type()
self.vl = Values()
self.meter = Meter(self._collectd)
def test_sample_type_gauge(self):
# sample_type uses get_dataset()[0][1]
self._collectd.get_dataset.return_value = [('value', 'gauge', )]
actual = self.meter.sample_type(self.vl)
self._collectd.get_dataset.assert_called_once()
self.assertEqual("gauge", actual)
def test_sample_type_derive(self):
# sample_type uses get_dataset()[0][1]
self._collectd.get_dataset.return_value = [('value', 'derive', )]
actual = self.meter.sample_type(self.vl)
self._collectd.get_dataset.assert_called_once()
self.assertEqual("delta", actual)
def test_sample_type_absolute(self):
# sample_type uses get_dataset()[0][1]
self._collectd.get_dataset.return_value = [('value', 'absolute', )]
actual = self.meter.sample_type(self.vl)
self._collectd.get_dataset.assert_called_once()
self.assertEqual("cumulative", actual)
def test_sample_type_counter(self):
# sample_type uses get_dataset()[0][1]
self._collectd.get_dataset.return_value = [('value', 'counter', )]
actual = self.meter.sample_type(self.vl)
self._collectd.get_dataset.assert_called_once()
self.assertEqual("cumulative", actual)
@mock.patch('collectd_ceilometer.common.meters.base.LOGGER')
def test_sample_type_invalid(self, LOGGER):
self._collectd.get_dataset.side_effect = Exception("Boom!")
actual = self.meter.sample_type(self.vl)
self._collectd.get_dataset.assert_called_once()
LOGGER.warning.assert_called_once()
self.assertEqual("gauge", actual)