128 lines
4.1 KiB
Python
128 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2014 Objectif Libre
|
|
#
|
|
# 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 voluptuous import error as verror
|
|
|
|
from cloudkitty import collector
|
|
from cloudkitty import tests
|
|
|
|
|
|
class MetricConfigValidationTest(tests.TestCase):
|
|
|
|
base_data = {
|
|
'metrics': {
|
|
'metric_one': {
|
|
'groupby': ['one'],
|
|
'metadata': ['one'],
|
|
'unit': 'u',
|
|
}
|
|
}
|
|
}
|
|
|
|
base_output = {
|
|
'metric_one': {
|
|
'groupby': ['one'],
|
|
'metadata': ['one'],
|
|
'unit': 'u',
|
|
'factor': 1,
|
|
'offset': 0,
|
|
'mutate': 'NONE',
|
|
}
|
|
}
|
|
|
|
def test_base_minimal_config(self):
|
|
data = copy.deepcopy(self.base_data)
|
|
expected_output = copy.deepcopy(self.base_output)
|
|
expected_output['metric_one']['groupby'].append('project_id')
|
|
|
|
self.assertEqual(
|
|
collector.BaseCollector.check_configuration(data),
|
|
expected_output,
|
|
)
|
|
|
|
def test_gnocchi_minimal_config_no_extra_args(self):
|
|
data = copy.deepcopy(self.base_data)
|
|
|
|
self.assertRaises(
|
|
verror.MultipleInvalid,
|
|
collector.gnocchi.GnocchiCollector.check_configuration,
|
|
data,
|
|
)
|
|
|
|
def test_gnocchi_minimal_config_minimal_extra_args(self):
|
|
data = copy.deepcopy(self.base_data)
|
|
data['metrics']['metric_one']['extra_args'] = {'resource_type': 'res'}
|
|
expected_output = copy.deepcopy(self.base_output)
|
|
expected_output['metric_one']['groupby'] += ['project_id', 'id']
|
|
expected_output['metric_one']['extra_args'] = {
|
|
'aggregation_method': 'max',
|
|
'resource_type': 'res',
|
|
'resource_key': 'id',
|
|
}
|
|
|
|
self.assertEqual(
|
|
collector.gnocchi.GnocchiCollector.check_configuration(data),
|
|
expected_output,
|
|
)
|
|
|
|
def test_monasca_minimal_config_no_extra_args(self):
|
|
data = copy.deepcopy(self.base_data)
|
|
|
|
self.assertRaises(
|
|
verror.MultipleInvalid,
|
|
collector.monasca.MonascaCollector.check_configuration,
|
|
data,
|
|
)
|
|
|
|
def test_monasca_minimal_config_minimal_extra_args(self):
|
|
data = copy.deepcopy(self.base_data)
|
|
data['metrics']['metric_one']['extra_args'] = {}
|
|
expected_output = copy.deepcopy(self.base_output)
|
|
expected_output['metric_one']['groupby'].extend(
|
|
['project_id', 'resource_id'])
|
|
expected_output['metric_one']['extra_args'] = {
|
|
'aggregation_method': 'max',
|
|
'resource_key': 'resource_id',
|
|
}
|
|
|
|
self.assertEqual(
|
|
collector.monasca.MonascaCollector.check_configuration(data),
|
|
expected_output,
|
|
)
|
|
|
|
def test_prometheus_minimal_config_empty_extra_args(self):
|
|
data = copy.deepcopy(self.base_data)
|
|
data['extra_args'] = {}
|
|
|
|
self.assertRaises(
|
|
verror.MultipleInvalid,
|
|
collector.prometheus.PrometheusCollector.check_configuration,
|
|
data,
|
|
)
|
|
|
|
def test_prometheus_minimal_config_minimal_extra_args(self):
|
|
data = copy.deepcopy(self.base_data)
|
|
data['metrics']['metric_one']['extra_args'] = {'query': 'query'}
|
|
expected_output = copy.deepcopy(self.base_output)
|
|
expected_output['metric_one']['groupby'].append('project_id')
|
|
expected_output['metric_one']['extra_args'] = {'query': 'query'}
|
|
|
|
self.assertEqual(
|
|
collector.prometheus.PrometheusCollector.check_configuration(data),
|
|
expected_output,
|
|
)
|