90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
#
|
|
# Copyright 2016 IBM Corp.
|
|
#
|
|
# 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 copy
|
|
|
|
from oslo_context import context
|
|
from oslo_utils import timeutils
|
|
from oslotest import base
|
|
|
|
from ceilometer import sample
|
|
from ceilometer.transformer import conversions
|
|
|
|
|
|
class AggregatorTransformerTestCase(base.BaseTestCase):
|
|
SAMPLE = sample.Sample(
|
|
name='cpu',
|
|
type=sample.TYPE_CUMULATIVE,
|
|
unit='ns',
|
|
volume='1234567',
|
|
user_id='56c5692032f34041900342503fecab30',
|
|
project_id='ac9494df2d9d4e709bac378cceabaf23',
|
|
resource_id='1ca738a1-c49c-4401-8346-5c60ebdb03f4',
|
|
timestamp="2015-10-29 14:12:15.485877+00:00",
|
|
resource_metadata={}
|
|
)
|
|
|
|
def setUp(self):
|
|
super(AggregatorTransformerTestCase, self).setUp()
|
|
self._sample_offset = 0
|
|
|
|
def test_init_input_validation(self):
|
|
aggregator = conversions.AggregatorTransformer("2", "15", None,
|
|
None, None)
|
|
self.assertEqual(2, aggregator.size)
|
|
self.assertEqual(15, aggregator.retention_time)
|
|
|
|
def test_init_no_size_or_rention_time(self):
|
|
aggregator = conversions.AggregatorTransformer()
|
|
self.assertEqual(1, aggregator.size)
|
|
self.assertEqual(None, aggregator.retention_time)
|
|
|
|
def test_init_size_zero(self):
|
|
aggregator = conversions.AggregatorTransformer(size="0")
|
|
self.assertEqual(1, aggregator.size)
|
|
self.assertEqual(None, aggregator.retention_time)
|
|
|
|
def test_init_input_validation_size_invalid(self):
|
|
self.assertRaises(ValueError, conversions.AggregatorTransformer,
|
|
"abc", "15", None, None, None)
|
|
|
|
def test_init_input_validation_retention_time_invalid(self):
|
|
self.assertRaises(ValueError, conversions.AggregatorTransformer,
|
|
"2", "abc", None, None, None)
|
|
|
|
def test_size_unbounded(self):
|
|
aggregator = conversions.AggregatorTransformer(size="0",
|
|
retention_time="300")
|
|
self._insert_sample_data(aggregator)
|
|
|
|
samples = aggregator.flush(context.get_admin_context())
|
|
|
|
self.assertEqual([], samples)
|
|
|
|
def test_size_bounded(self):
|
|
aggregator = conversions.AggregatorTransformer(size="100")
|
|
self._insert_sample_data(aggregator)
|
|
|
|
samples = aggregator.flush(context.get_admin_context())
|
|
|
|
self.assertEqual(100, len(samples))
|
|
|
|
def _insert_sample_data(self, aggregator):
|
|
for _ in range(100):
|
|
sample = copy.copy(self.SAMPLE)
|
|
sample.resource_id = sample.resource_id + str(self._sample_offset)
|
|
sample.timestamp = timeutils.isotime()
|
|
aggregator.handle_sample(context.get_admin_context(), sample)
|
|
self._sample_offset += 1
|