diff options
Diffstat (limited to 'ceilometerclient/tests/unit/v2/test_shell.py')
| -rw-r--r-- | ceilometerclient/tests/unit/v2/test_shell.py | 1977 |
1 files changed, 0 insertions, 1977 deletions
diff --git a/ceilometerclient/tests/unit/v2/test_shell.py b/ceilometerclient/tests/unit/v2/test_shell.py deleted file mode 100644 index eca15e3..0000000 --- a/ceilometerclient/tests/unit/v2/test_shell.py +++ /dev/null @@ -1,1977 +0,0 @@ -# Copyright Ericsson AB 2014. All rights reserved -# -# 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 json -import re -import sys - -import mock -import six -from testtools import matchers - -from ceilometerclient import exc -from ceilometerclient import shell as base_shell -from ceilometerclient.tests.unit import test_shell -from ceilometerclient.tests.unit import utils -from ceilometerclient.v2 import alarms -from ceilometerclient.v2 import capabilities -from ceilometerclient.v2 import event_types -from ceilometerclient.v2 import events -from ceilometerclient.v2 import meters -from ceilometerclient.v2 import resources -from ceilometerclient.v2 import samples -from ceilometerclient.v2 import shell as ceilometer_shell -from ceilometerclient.v2 import statistics -from ceilometerclient.v2 import trait_descriptions -from ceilometerclient.v2 import traits -from keystoneauth1 import exceptions - - -class ShellAlarmStateCommandsTest(utils.BaseTestCase): - - ALARM_ID = 'foobar' - - def setUp(self): - super(ShellAlarmStateCommandsTest, self).setUp() - self.cc = mock.Mock() - self.cc.alarms = mock.Mock() - self.args = mock.Mock() - self.args.alarm_id = self.ALARM_ID - - def test_alarm_state_get(self): - ceilometer_shell.do_alarm_state_get(self.cc, self.args) - self.cc.alarms.get_state.assert_called_once_with(self.ALARM_ID) - self.assertFalse(self.cc.alarms.set_state.called) - - def test_alarm_state_set(self): - self.args.state = 'ok' - ceilometer_shell.do_alarm_state_set(self.cc, self.args) - self.cc.alarms.set_state.assert_called_once_with(self.ALARM_ID, 'ok') - self.assertFalse(self.cc.alarms.get_state.called) - - -class ShellAlarmHistoryCommandTest(utils.BaseTestCase): - - ALARM_ID = '768ff714-8cfb-4db9-9753-d484cb33a1cc' - FULL_DETAIL = ('{"alarm_actions": [], ' - '"user_id": "8185aa72421a4fd396d4122cba50e1b5", ' - '"name": "scombo", ' - '"timestamp": "2013-10-03T08:58:33.647912", ' - '"enabled": true, ' - '"state_timestamp": "2013-10-03T08:58:33.647912", ' - '"rule": {"operator": "or", "alarm_ids": ' - '["062cc907-3a9f-4867-ab3b-fa83212b39f7"]}, ' - '"alarm_id": "768ff714-8cfb-4db9-9753-d484cb33a1cc", ' - '"state": "insufficient data", ' - '"insufficient_data_actions": [], ' - '"repeat_actions": false, ' - '"ok_actions": [], ' - '"project_id": "57d04f24d0824b78b1ea9bcecedbda8f", ' - '"type": "combination", ' - '"severity": "low", ' - '"description": "Combined state of alarms ' - '062cc907-3a9f-4867-ab3b-fa83212b39f7"}') - ALARM_HISTORY = [{'on_behalf_of': '57d04f24d0824b78b1ea9bcecedbda8f', - 'user_id': '8185aa72421a4fd396d4122cba50e1b5', - 'event_id': 'c74a8611-6553-4764-a860-c15a6aabb5d0', - 'timestamp': '2013-10-03T08:59:28.326000', - 'detail': '{"state": "alarm"}', - 'alarm_id': '768ff714-8cfb-4db9-9753-d484cb33a1cc', - 'project_id': '57d04f24d0824b78b1ea9bcecedbda8f', - 'type': 'state transition'}, - {'on_behalf_of': '57d04f24d0824b78b1ea9bcecedbda8f', - 'user_id': '8185aa72421a4fd396d4122cba50e1b5', - 'event_id': 'c74a8611-6553-4764-a860-c15a6aabb5d0', - 'timestamp': '2013-10-03T08:59:28.326000', - 'detail': '{"description": "combination of one"}', - 'alarm_id': '768ff714-8cfb-4db9-9753-d484cb33a1cc', - 'project_id': '57d04f24d0824b78b1ea9bcecedbda8f', - 'type': 'rule change'}, - {'on_behalf_of': '57d04f24d0824b78b1ea9bcecedbda8f', - 'user_id': '8185aa72421a4fd396d4122cba50e1b5', - 'event_id': '4fd7df9e-190d-4471-8884-dc5a33d5d4bb', - 'timestamp': '2013-10-03T08:58:33.647000', - 'detail': FULL_DETAIL, - 'alarm_id': '768ff714-8cfb-4db9-9753-d484cb33a1cc', - 'project_id': '57d04f24d0824b78b1ea9bcecedbda8f', - 'type': 'creation'}] - TIMESTAMP_RE = (' +\| (\d{4})-(\d{2})-(\d{2})T' - '(\d{2})\:(\d{2})\:(\d{2})\.(\d{6}) \| +') - - def setUp(self): - super(ShellAlarmHistoryCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.alarms = mock.Mock() - self.args = mock.Mock() - self.args.alarm_id = self.ALARM_ID - - @mock.patch('sys.stdout', new=six.StringIO()) - def _do_test_alarm_history(self, raw_query=None, parsed_query=None): - self.args.query = raw_query - history = [alarms.AlarmChange(mock.Mock(), change) - for change in self.ALARM_HISTORY] - self.cc.alarms.get_history.return_value = history - - ceilometer_shell.do_alarm_history(self.cc, self.args) - self.cc.alarms.get_history.assert_called_once_with( - q=parsed_query, - alarm_id=self.ALARM_ID - ) - out = sys.stdout.getvalue() - required = [ - '.*creation%sname: scombo.*' % self.TIMESTAMP_RE, - '.*rule change%sdescription: combination of one.*' % - self.TIMESTAMP_RE, - '.*state transition%sstate: alarm.*' % self.TIMESTAMP_RE, - ] - for r in required: - self.assertThat(out, matchers.MatchesRegex(r, re.DOTALL)) - - def test_alarm_all_history(self): - self._do_test_alarm_history() - - def test_alarm_constrained_history(self): - parsed_query = [dict(field='timestamp', - value='2013-10-03T08:59:28', - op='gt', - type='')] - self._do_test_alarm_history(raw_query='timestamp>2013-10-03T08:59:28', - parsed_query=parsed_query) - - -class ShellAlarmCommandTest(utils.BaseTestCase): - - ALARM_ID = '768ff714-8cfb-4db9-9753-d484cb33a1cc' - ALARM = {"alarm_actions": ["log://"], - "ok_actions": [], - "description": "instance running hot", - "timestamp": "2013-11-20T10:38:42.206952", - "enabled": True, - "state_timestamp": "2013-11-19T17:20:44", - "threshold_rule": {"meter_name": "cpu_util", - "evaluation_periods": 3, - "period": 600, - "statistic": "avg", - "threshold": 99.0, - "query": [{"field": "resource_id", - "value": "INSTANCE_ID", - "op": "eq"}], - "comparison_operator": "gt"}, - "time_constraints": [{"name": "cons1", - "description": "desc1", - "start": "0 11 * * *", - "duration": 300, - "timezone": ""}, - {"name": "cons2", - "description": "desc2", - "start": "0 23 * * *", - "duration": 600, - "timezone": ""}], - "alarm_id": ALARM_ID, - "state": "insufficient data", - "severity": "low", - "insufficient_data_actions": [], - "repeat_actions": True, - "user_id": "528d9b68fa774689834b5c04b4564f8a", - "project_id": "ed9d4e2be2a748bc80108053cf4598f5", - "type": "threshold", - "name": "cpu_high"} - - THRESHOLD_ALARM_CLI_ARGS = [ - '--name', 'cpu_high', - '--description', 'instance running hot', - '--meter-name', 'cpu_util', - '--threshold', '70.0', - '--comparison-operator', 'gt', - '--statistic', 'avg', - '--period', '600', - '--evaluation-periods', '3', - '--alarm-action', 'log://', - '--alarm-action', 'http://example.com/alarm/state', - '--query', 'resource_id=INSTANCE_ID' - ] - - def setUp(self): - super(ShellAlarmCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.alarms = mock.Mock() - self.args = mock.Mock() - self.args.alarm_id = self.ALARM_ID - - @mock.patch('sys.stdout', new=six.StringIO()) - def _do_test_alarm_update_repeat_actions(self, method, repeat_actions): - self.args.threshold = 42.0 - if repeat_actions is not None: - self.args.repeat_actions = repeat_actions - alarm = [alarms.Alarm(mock.Mock(), self.ALARM)] - self.cc.alarms.get.return_value = alarm - self.cc.alarms.update.return_value = alarm[0] - - method(self.cc, self.args) - args, kwargs = self.cc.alarms.update.call_args - self.assertEqual(self.ALARM_ID, args[0]) - self.assertEqual(42.0, kwargs.get('threshold')) - if repeat_actions is not None: - self.assertEqual(repeat_actions, kwargs.get('repeat_actions')) - else: - self.assertNotIn('repeat_actions', kwargs) - - def test_alarm_update_repeat_actions_untouched(self): - method = ceilometer_shell.do_alarm_update - self._do_test_alarm_update_repeat_actions(method, None) - - def test_alarm_update_repeat_actions_set(self): - method = ceilometer_shell.do_alarm_update - self._do_test_alarm_update_repeat_actions(method, True) - - def test_alarm_update_repeat_actions_clear(self): - method = ceilometer_shell.do_alarm_update - self._do_test_alarm_update_repeat_actions(method, False) - - def test_alarm_combination_update_repeat_actions_untouched(self): - method = ceilometer_shell.do_alarm_combination_update - self._do_test_alarm_update_repeat_actions(method, None) - - def test_alarm_combination_update_repeat_actions_set(self): - method = ceilometer_shell.do_alarm_combination_update - self._do_test_alarm_update_repeat_actions(method, True) - - def test_alarm_combination_update_repeat_actions_clear(self): - method = ceilometer_shell.do_alarm_combination_update - self._do_test_alarm_update_repeat_actions(method, False) - - def test_alarm_threshold_update_repeat_actions_untouched(self): - method = ceilometer_shell.do_alarm_threshold_update - self._do_test_alarm_update_repeat_actions(method, None) - - def test_alarm_threshold_update_repeat_actions_set(self): - method = ceilometer_shell.do_alarm_threshold_update - self._do_test_alarm_update_repeat_actions(method, True) - - def test_alarm_threshold_update_repeat_actions_clear(self): - method = ceilometer_shell.do_alarm_threshold_update - self._do_test_alarm_update_repeat_actions(method, False) - - def test_alarm_event_upadte_repeat_action_untouched(self): - method = ceilometer_shell.do_alarm_event_update - self._do_test_alarm_update_repeat_actions(method, None) - - def test_alarm_event_upadte_repeat_action_set(self): - method = ceilometer_shell.do_alarm_event_update - self._do_test_alarm_update_repeat_actions(method, True) - - def test_alarm_event_upadte_repeat_action_clear(self): - method = ceilometer_shell.do_alarm_event_update - self._do_test_alarm_update_repeat_actions(method, False) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_alarm_threshold_create_args(self): - argv = ['alarm-threshold-create'] + self.THRESHOLD_ALARM_CLI_ARGS - self._test_alarm_threshold_action_args('create', argv) - - def test_alarm_threshold_update_args(self): - argv = ['alarm-threshold-update', 'x'] + self.THRESHOLD_ALARM_CLI_ARGS - self._test_alarm_threshold_action_args('update', argv) - - @mock.patch('sys.stdout', new=six.StringIO()) - def _test_alarm_threshold_action_args(self, action, argv): - shell = base_shell.CeilometerShell() - _, args = shell.parse_args(argv) - - alarm = alarms.Alarm(mock.Mock(), self.ALARM) - getattr(self.cc.alarms, action).return_value = alarm - - func = getattr(ceilometer_shell, 'do_alarm_threshold_' + action) - func(self.cc, args) - _, kwargs = getattr(self.cc.alarms, action).call_args - self._check_alarm_threshold_args(kwargs) - - def _check_alarm_threshold_args(self, kwargs): - self.assertEqual('cpu_high', kwargs.get('name')) - self.assertEqual('instance running hot', kwargs.get('description')) - actions = ['log://', 'http://example.com/alarm/state'] - self.assertEqual(actions, kwargs.get('alarm_actions')) - self.assertIn('threshold_rule', kwargs) - rule = kwargs['threshold_rule'] - self.assertEqual('cpu_util', rule.get('meter_name')) - self.assertEqual(70.0, rule.get('threshold')) - self.assertEqual('gt', rule.get('comparison_operator')) - self.assertEqual('avg', rule.get('statistic')) - self.assertEqual(600, rule.get('period')) - self.assertEqual(3, rule.get('evaluation_periods')) - query = dict(field='resource_id', type='', - value='INSTANCE_ID', op='eq') - self.assertEqual([query], rule['query']) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_alarm_create_time_constraints(self): - shell = base_shell.CeilometerShell() - argv = ['alarm-threshold-create', - '--name', 'cpu_high', - '--meter-name', 'cpu_util', - '--threshold', '70.0', - '--time-constraint', - 'name=cons1;start="0 11 * * *";duration=300', - '--time-constraint', - 'name=cons2;start="0 23 * * *";duration=600', - ] - _, args = shell.parse_args(argv) - - alarm = alarms.Alarm(mock.Mock(), self.ALARM) - self.cc.alarms.create.return_value = alarm - - ceilometer_shell.do_alarm_threshold_create(self.cc, args) - _, kwargs = self.cc.alarms.create.call_args - time_constraints = [dict(name='cons1', start='0 11 * * *', - duration='300'), - dict(name='cons2', start='0 23 * * *', - duration='600')] - self.assertEqual(time_constraints, kwargs['time_constraints']) - - -class ShellAlarmGnocchiCommandTest(test_shell.ShellTestBase): - - ALARM_ID = 'b69ecdb9-f19b-4fb5-950f-5eb53938b718' - TIME_CONSTRAINTS = [{ - u'duration': 300, - u'start': u'0 11 * * *', - u'description': u'desc1', - u'name': u'cons1', - u'timezone': u''}, { - u'duration': 600, - u'start': u'0 23 * * *', - u'name': u'cons2', - u'description': u'desc2', - u'timezone': u''}] - - ALARM1 = { - u'name': u'name_gnocchi_alarm', - u'description': u'description_gnocchi_alarm', - u'enabled': True, - u'ok_actions': [u'http://something/ok'], - u'alarm_actions': [u'http://something/alarm'], - u'timestamp': u'2015-12-21T03:10:32.305133', - u'state_timestamp': u'2015-12-21T03:10:32.305133', - u'gnocchi_resources_threshold_rule': { - u'evaluation_periods': 3, - u'metric': u'cpu_util', - u'resource_id': u'768ff714-8cfb-4db9-9753-d484cb33a1cc', - u'threshold': 70.0, - u'granularity': 60, - u'aggregation_method': u'count', - u'comparison_operator': u'le', - u'resource_type': u'instance', - }, - u'time_constraints': TIME_CONSTRAINTS, - u'alarm_id': ALARM_ID, - u'state': u'ok', - u'insufficient_data_actions': [u'http://something/insufficient'], - u'repeat_actions': True, - u'user_id': u'f28735621ee84f329144eb467c91fce6', - u'project_id': u'97fcad0402ce4f65ac3bd42a0c6a7e74', - u'type': u'gnocchi_resources_threshold', - u'severity': u'critical', - } - - ALARM2 = { - u'name': u'name_gnocchi_alarm', - u'description': u'description_gnocchi_alarm', - u'enabled': True, - u'ok_actions': [u'http://something/ok'], - u'alarm_actions': [u'http://something/alarm'], - u'timestamp': u'2015-12-21T03:10:32.305133', - u'state_timestamp': u'2015-12-21T03:10:32.305133', - u'gnocchi_aggregation_by_metrics_threshold_rule': { - u'evaluation_periods': 3, - u'metrics': [u'b3d9d8ab-05e8-439f-89ad-5e978dd2a5eb', - u'009d4faf-c275-46f0-8f2d-670b15bac2b0'], - u'threshold': 70.0, - u'granularity': 60, - u'aggregation_method': u'count', - u'comparison_operator': u'le', - }, - u'time_constraints': TIME_CONSTRAINTS, - u'alarm_id': ALARM_ID, - u'state': u'ok', - u'insufficient_data_actions': [u'http://something/insufficient'], - u'repeat_actions': True, - u'user_id': u'f28735621ee84f329144eb467c91fce6', - u'project_id': u'97fcad0402ce4f65ac3bd42a0c6a7e74', - u'type': u'gnocchi_aggregation_by_metrics_threshold', - u'severity': u'critical', - } - - ALARM3 = { - u'name': u'name_gnocchi_alarm', - u'description': u'description_gnocchi_alarm', - u'enabled': True, - u'ok_actions': [u'http://something/ok'], - u'alarm_actions': [u'http://something/alarm'], - u'timestamp': u'2015-12-21T03:10:32.305133', - u'state_timestamp': u'2015-12-21T03:10:32.305133', - u'gnocchi_aggregation_by_resources_threshold_rule': { - u'evaluation_periods': 3, - u'metric': u'cpu_util', - u'threshold': 70.0, - u'granularity': 60, - u'aggregation_method': u'count', - u'comparison_operator': u'le', - u'resource_type': u'instance', - u'query': u'{"=": {"server_group":"my_autoscaling_group"}}', - }, - u'time_constraints': TIME_CONSTRAINTS, - u'alarm_id': ALARM_ID, - u'state': u'ok', - u'insufficient_data_actions': [u'http://something/insufficient'], - u'repeat_actions': True, - u'user_id': u'f28735621ee84f329144eb467c91fce6', - u'project_id': u'97fcad0402ce4f65ac3bd42a0c6a7e74', - u'type': u'gnocchi_aggregation_by_resources_threshold', - u'severity': u'critical', - } - - COMMON_CLI_ARGS = [ - '--name', 'name_gnocchi_alarm', - '--description', 'description_gnocchi_alarm', - '--enabled', 'True', - '--state', 'ok', - '--severity', 'critical', - '--ok-action', 'http://something/ok', - '--alarm-action', 'http://something/alarm', - '--insufficient-data-action', 'http://something/insufficient', - '--repeat-actions', 'True', - '--comparison-operator', 'le', - '--aggregation-method', 'count', - '--threshold', '70', - '--evaluation-periods', '3', - '--granularity', '60', - '--time-constraint', - 'name=cons1;start="0 11 * * *";duration=300;description="desc1"', - '--time-constraint', - 'name=cons2;start="0 23 * * *";duration=600;description="desc2"', - '--user-id', 'f28735621ee84f329144eb467c91fce6', - '--project-id', '97fcad0402ce4f65ac3bd42a0c6a7e74', - ] - - GNOCCHI_RESOURCES_CLI_ARGS = COMMON_CLI_ARGS + [ - '--metric', 'cpu_util', - '--resource-type', 'instance', - '--resource-id', '768ff714-8cfb-4db9-9753-d484cb33a1cc', - ] - - GNOCCHI_AGGR_BY_METRICS_CLI_ARGS = COMMON_CLI_ARGS + [ - '-m', 'b3d9d8ab-05e8-439f-89ad-5e978dd2a5eb', - '-m', '009d4faf-c275-46f0-8f2d-670b15bac2b0', - ] - - GNOCCHI_AGGR_BY_RESOURCES_CLI_ARGS = COMMON_CLI_ARGS + [ - '--metric', 'cpu_util', - '--resource-type', 'instance', - '--query', '{"=": {"server_group":"my_autoscaling_group"}}' - ] - - def setUp(self): - super(ShellAlarmGnocchiCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.alarms = mock.Mock() - self.args = mock.Mock() - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_do_alarm_gnocchi_resources_threshold_create(self): - - alarm = alarms.Alarm(mock.Mock(), self.ALARM1) - self.cc.alarms.create.return_value = alarm - ceilometer_shell.do_alarm_gnocchi_resources_threshold_create(self.cc, - self.args) - self.assertEqual('''\ -+---------------------------+--------------------------------------+ -| Property | Value | -+---------------------------+--------------------------------------+ -| aggregation_method | count | -| alarm_actions | ["http://something/alarm"] | -| alarm_id | b69ecdb9-f19b-4fb5-950f-5eb53938b718 | -| comparison_operator | le | -| description | description_gnocchi_alarm | -| enabled | True | -| evaluation_periods | 3 | -| granularity | 60 | -| insufficient_data_actions | ["http://something/insufficient"] | -| metric | cpu_util | -| name | name_gnocchi_alarm | -| ok_actions | ["http://something/ok"] | -| project_id | 97fcad0402ce4f65ac3bd42a0c6a7e74 | -| repeat_actions | True | -| resource_id | 768ff714-8cfb-4db9-9753-d484cb33a1cc | -| resource_type | instance | -| severity | critical | -| state | ok | -| threshold | 70.0 | -| time_constraints | [{name: cons1, | -| | description: desc1, | -| | start: 0 11 * * *, | -| | duration: 300}, | -| | {name: cons2, | -| | description: desc2, | -| | start: 0 23 * * *, | -| | duration: 600}] | -| type | gnocchi_resources_threshold | -| user_id | f28735621ee84f329144eb467c91fce6 | -+---------------------------+--------------------------------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_do_alarm_gnocchi_aggr_by_metrics_threshold_create(self): - - alarm = alarms.Alarm(mock.Mock(), self.ALARM2) - self.cc.alarms.create.return_value = alarm - ceilometer_shell.\ - do_alarm_gnocchi_aggregation_by_metrics_threshold_create( - self.cc, self.args) - stdout = sys.stdout.getvalue() - self.assertIn("b69ecdb9-f19b-4fb5-950f-5eb53938b718", stdout) - self.assertIn("[\"http://something/alarm\"]", stdout) - self.assertIn("description_gnocchi_alarm", stdout) - self.assertIn("gnocchi_aggregation_by_metrics_threshold", stdout) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_do_alarm_gnocchi_aggr_by_resources_threshold_create(self): - - alarm = alarms.Alarm(mock.Mock(), self.ALARM3) - self.cc.alarms.create.return_value = alarm - ceilometer_shell.\ - do_alarm_gnocchi_aggregation_by_resources_threshold_create( - self.cc, self.args) - self.assertEqual('''\ -+---------------------------+------------------------------------------------+ -| Property | Value | -+---------------------------+------------------------------------------------+ -| aggregation_method | count | -| alarm_actions | ["http://something/alarm"] | -| alarm_id | b69ecdb9-f19b-4fb5-950f-5eb53938b718 | -| comparison_operator | le | -| description | description_gnocchi_alarm | -| enabled | True | -| evaluation_periods | 3 | -| granularity | 60 | -| insufficient_data_actions | ["http://something/insufficient"] | -| metric | cpu_util | -| name | name_gnocchi_alarm | -| ok_actions | ["http://something/ok"] | -| project_id | 97fcad0402ce4f65ac3bd42a0c6a7e74 | -| query | {"=": {"server_group":"my_autoscaling_group"}} | -| repeat_actions | True | -| resource_type | instance | -| severity | critical | -| state | ok | -| threshold | 70.0 | -| time_constraints | [{name: cons1, | -| | description: desc1, | -| | start: 0 11 * * *, | -| | duration: 300}, | -| | {name: cons2, | -| | description: desc2, | -| | start: 0 23 * * *, | -| | duration: 600}] | -| type | gnocchi_aggregation_by_resources_threshold | -| user_id | f28735621ee84f329144eb467c91fce6 | -+---------------------------+------------------------------------------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_do_alarm_gnocchi_resources_threshold_create_args(self): - argv = ['alarm-gnocchi-resources-threshold-create'] - argv.extend(self.GNOCCHI_RESOURCES_CLI_ARGS) - self._test_alarm_gnocchi_resources_arguments('create', argv) - - def test_do_alarm_gnocchi_resources_threshold_update_args(self): - argv = ['alarm-gnocchi-resources-threshold-update'] - argv.extend(self.GNOCCHI_RESOURCES_CLI_ARGS) - argv.append(self.ALARM_ID) - self._test_alarm_gnocchi_resources_arguments('update', argv) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_do_alarm_gnocchi_aggr_by_metrics_threshold_create_args(self): - argv = ['alarm-gnocchi-aggregation-by-metrics-threshold-create'] - argv.extend(self.GNOCCHI_AGGR_BY_METRICS_CLI_ARGS) - self._test_alarm_gnocchi_aggr_by_metrics_arguments('create', argv) - - def test_do_alarm_gnocchi_aggr_by_metrics_threshold_update_args(self): - argv = ['alarm-gnocchi-aggregation-by-metrics-threshold-update'] - argv.extend(self.GNOCCHI_AGGR_BY_METRICS_CLI_ARGS) - argv.append(self.ALARM_ID) - self._test_alarm_gnocchi_aggr_by_metrics_arguments('update', argv) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_do_alarm_gnocchi_aggr_by_resources_threshold_create_args(self): - argv = ['alarm-gnocchi-aggregation-by-resources-threshold-create'] - argv.extend(self.GNOCCHI_AGGR_BY_RESOURCES_CLI_ARGS) - self._test_alarm_gnocchi_aggr_by_resources_arguments('create', argv) - - def test_do_alarm_gnocchi_aggr_by_resources_threshold_update_args(self): - argv = ['alarm-gnocchi-aggregation-by-resources-threshold-update'] - argv.extend(self.GNOCCHI_AGGR_BY_RESOURCES_CLI_ARGS) - argv.append(self.ALARM_ID) - self._test_alarm_gnocchi_aggr_by_resources_arguments('update', argv) - - @mock.patch('sys.stdout', new=six.StringIO()) - def _test_common_alarm_gnocchi_arguments(self, kwargs): - self.assertEqual('97fcad0402ce4f65ac3bd42a0c6a7e74', - kwargs.get('project_id')) - self.assertEqual('f28735621ee84f329144eb467c91fce6', - kwargs.get('user_id')) - self.assertEqual('name_gnocchi_alarm', kwargs.get('name')) - self.assertEqual('description_gnocchi_alarm', - kwargs.get('description')) - self.assertEqual(['http://something/alarm'], - kwargs.get('alarm_actions')) - self.assertEqual(['http://something/ok'], kwargs.get('ok_actions')) - self.assertEqual(['http://something/insufficient'], - kwargs.get('insufficient_data_actions')) - self.assertEqual('critical', kwargs.get('severity')) - self.assertEqual('ok', kwargs.get('state')) - self.assertEqual(True, kwargs.get('enabled')) - self.assertEqual(True, kwargs.get('repeat_actions')) - time_constraints = [dict(name='cons1', start='0 11 * * *', - duration='300', description='desc1'), - dict(name='cons2', start='0 23 * * *', - duration='600', description='desc2')] - self.assertEqual(time_constraints, kwargs['time_constraints']) - - def _test_alarm_gnocchi_resources_arguments(self, action, argv): - self.make_env(test_shell.FAKE_V2_ENV) - with mock.patch.object(alarms.AlarmManager, action) as mocked: - with mock.patch('ceilometerclient.apiclient.' - 'client.HTTPClient.client_request') as request: - request.site_effect = exceptions.EndpointNotFound - base_shell.main(argv) - args, kwargs = mocked.call_args - self.assertEqual('gnocchi_resources_threshold', kwargs.get('type')) - self.assertIn('gnocchi_resources_threshold_rule', kwargs) - rule = kwargs['gnocchi_resources_threshold_rule'] - self.assertEqual('cpu_util', rule.get('metric')) - self.assertEqual(70.0, rule.get('threshold')) - self.assertEqual(60, rule.get('granularity')) - self.assertEqual('count', rule.get('aggregation_method')) - self.assertEqual('le', rule.get('comparison_operator')) - self.assertEqual(3, rule.get('evaluation_periods')) - self.assertEqual('768ff714-8cfb-4db9-9753-d484cb33a1cc', - rule.get('resource_id')) - self.assertEqual('instance', rule.get('resource_type')) - self._test_common_alarm_gnocchi_arguments(kwargs) - - def _test_alarm_gnocchi_aggr_by_metrics_arguments(self, action, argv): - self.make_env(test_shell.FAKE_V2_ENV) - with mock.patch.object(alarms.AlarmManager, action) as mocked: - with mock.patch('ceilometerclient.apiclient.' - 'client.HTTPClient.client_request') as request: - request.site_effect = exceptions.EndpointNotFound - base_shell.main(argv) - args, kwargs = mocked.call_args - self.assertEqual('gnocchi_aggregation_by_metrics_threshold', - kwargs.get('type')) - self.assertIn('gnocchi_aggregation_by_metrics_threshold_rule', kwargs) - rule = kwargs['gnocchi_aggregation_by_metrics_threshold_rule'] - self.assertEqual(['b3d9d8ab-05e8-439f-89ad-5e978dd2a5eb', - '009d4faf-c275-46f0-8f2d-670b15bac2b0'], - rule.get('metrics')) - self.assertEqual(70.0, rule.get('threshold')) - self.assertEqual(60, rule.get('granularity')) - self.assertEqual('count', rule.get('aggregation_method')) - self.assertEqual('le', rule.get('comparison_operator')) - self.assertEqual(3, rule.get('evaluation_periods')) - self._test_common_alarm_gnocchi_arguments(kwargs) - - def _test_alarm_gnocchi_aggr_by_resources_arguments(self, action, argv): - self.make_env(test_shell.FAKE_V2_ENV) - with mock.patch.object(alarms.AlarmManager, action) as mocked: - with mock.patch('ceilometerclient.apiclient.' - 'client.HTTPClient.client_request') as request: - request.site_effect = exceptions.EndpointNotFound - base_shell.main(argv) - args, kwargs = mocked.call_args - self.assertEqual('gnocchi_aggregation_by_resources_threshold', - kwargs.get('type')) - self.assertIn('gnocchi_aggregation_by_resources_threshold_rule', - kwargs) - rule = kwargs['gnocchi_aggregation_by_resources_threshold_rule'] - self.assertEqual('cpu_util', rule.get('metric')) - self.assertEqual(70.0, rule.get('threshold')) - self.assertEqual(60, rule.get('granularity')) - self.assertEqual('count', rule.get('aggregation_method')) - self.assertEqual('le', rule.get('comparison_operator')) - self.assertEqual(3, rule.get('evaluation_periods')) - self.assertEqual('instance', rule.get('resource_type')) - self.assertEqual('{"=": {"server_group":"my_autoscaling_group"}}', - rule.get('query')) - self._test_common_alarm_gnocchi_arguments(kwargs) - - -class ShellSampleListCommandTest(utils.BaseTestCase): - - METER = 'cpu_util' - SAMPLE_VALUES = ( - ("cpu_util", - "5dcf5537-3161-4e25-9235-407e1385bd35", - "2013-10-15T05:50:30", - "%", - 0.261666666667, - "gauge", - "86536501-b2c9-48f6-9c6a-7a5b14ba7482"), - ("cpu_util", - "87d197e9-9cf6-4c25-bc66-1b1f4cedb52f", - "2013-10-15T05:50:29", - "%", - 0.261666666667, - "gauge", - "fe2a91ec-602b-4b55-8cba-5302ce3b916e",), - ("cpu_util", - "5dcf5537-3161-4e25-9235-407e1385bd35", - "2013-10-15T05:40:30", - "%", - 0.251247920133, - "gauge", - "52768bcb-b4e9-4db9-a30c-738c758b6f43"), - ("cpu_util", - "87d197e9-9cf6-4c25-bc66-1b1f4cedb52f", - "2013-10-15T05:40:29", - "%", - 0.26, - "gauge", - "31ae614a-ac6b-4fb9-b106-4667bae03308"), - ) - - OLD_SAMPLES = [ - dict(counter_name=s[0], - resource_id=s[1], - timestamp=s[2], - counter_unit=s[3], - counter_volume=s[4], - counter_type=s[5]) - for s in SAMPLE_VALUES - ] - - SAMPLES = [ - dict(meter=s[0], - resource_id=s[1], - timestamp=s[2], - unit=s[3], - volume=s[4], - type=s[5], - id=s[6]) - for s in SAMPLE_VALUES - ] - - def setUp(self): - super(ShellSampleListCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.samples = mock.Mock() - self.cc.new_samples = mock.Mock() - self.args = mock.Mock() - self.args.query = None - self.args.limit = None - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_old_sample_list(self): - self.args.meter = self.METER - sample_list = [samples.OldSample(mock.Mock(), sample) - for sample in self.OLD_SAMPLES] - self.cc.samples.list.return_value = sample_list - - ceilometer_shell.do_sample_list(self.cc, self.args) - self.cc.samples.list.assert_called_once_with( - meter_name=self.METER, - q=None, - limit=None) - - self.assertEqual('''\ -+--------------------------------------+----------+-------+----------------\ -+------+---------------------+ -| Resource ID | Name | Type | Volume \ -| Unit | Timestamp | -+--------------------------------------+----------+-------+----------------\ -+------+---------------------+ -| 5dcf5537-3161-4e25-9235-407e1385bd35 | cpu_util | gauge | 0.261666666667 \ -| % | 2013-10-15T05:50:30 | -| 87d197e9-9cf6-4c25-bc66-1b1f4cedb52f | cpu_util | gauge | 0.261666666667 \ -| % | 2013-10-15T05:50:29 | -| 5dcf5537-3161-4e25-9235-407e1385bd35 | cpu_util | gauge | 0.251247920133 \ -| % | 2013-10-15T05:40:30 | -| 87d197e9-9cf6-4c25-bc66-1b1f4cedb52f | cpu_util | gauge | 0.26 \ -| % | 2013-10-15T05:40:29 | -+--------------------------------------+----------+-------+----------------\ -+------+---------------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_sample_list(self): - self.args.meter = None - sample_list = [samples.Sample(mock.Mock(), sample) - for sample in self.SAMPLES] - self.cc.new_samples.list.return_value = sample_list - - ceilometer_shell.do_sample_list(self.cc, self.args) - self.cc.new_samples.list.assert_called_once_with( - q=None, - limit=None) - - self.assertEqual('''\ -+--------------------------------------+--------------------------------------\ -+----------+-------+----------------+------+---------------------+ -| ID | Resource ID \ -| Name | Type | Volume | Unit | Timestamp | -+--------------------------------------+--------------------------------------\ -+----------+-------+----------------+------+---------------------+ -| 86536501-b2c9-48f6-9c6a-7a5b14ba7482 | 5dcf5537-3161-4e25-9235-407e1385bd35 \ -| cpu_util | gauge | 0.261666666667 | % | 2013-10-15T05:50:30 | -| fe2a91ec-602b-4b55-8cba-5302ce3b916e | 87d197e9-9cf6-4c25-bc66-1b1f4cedb52f \ -| cpu_util | gauge | 0.261666666667 | % | 2013-10-15T05:50:29 | -| 52768bcb-b4e9-4db9-a30c-738c758b6f43 | 5dcf5537-3161-4e25-9235-407e1385bd35 \ -| cpu_util | gauge | 0.251247920133 | % | 2013-10-15T05:40:30 | -| 31ae614a-ac6b-4fb9-b106-4667bae03308 | 87d197e9-9cf6-4c25-bc66-1b1f4cedb52f \ -| cpu_util | gauge | 0.26 | % | 2013-10-15T05:40:29 | -+--------------------------------------+--------------------------------------\ -+----------+-------+----------------+------+---------------------+ -''', sys.stdout.getvalue()) - - -class ShellSampleShowCommandTest(utils.BaseTestCase): - - SAMPLE = { - "user_id": None, - "resource_id": "9b651dfd-7d30-402b-972e-212b2c4bfb05", - "timestamp": "2014-11-03T13:37:46", - "meter": "image", - "volume": 1.0, - "source": "openstack", - "recorded_at": "2014-11-03T13:37:46.994458", - "project_id": "2cc3a7bb859b4bacbeab0aa9ca673033", - "type": "gauge", - "id": "98b5f258-635e-11e4-8bdd-0025647390c1", - "unit": "image", - "metadata": { - "name": "cirros-0.3.2-x86_64-uec", - } - } - - def setUp(self): - super(ShellSampleShowCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.new_samples = mock.Mock() - self.args = mock.Mock() - self.args.sample_id = "98b5f258-635e-11e4-8bdd-0025647390c1" - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_sample_show(self): - sample = samples.Sample(mock.Mock(), self.SAMPLE) - self.cc.new_samples.get.return_value = sample - - ceilometer_shell.do_sample_show(self.cc, self.args) - self.cc.new_samples.get.assert_called_once_with( - "98b5f258-635e-11e4-8bdd-0025647390c1") - - self.assertEqual('''\ -+-------------+--------------------------------------+ -| Property | Value | -+-------------+--------------------------------------+ -| id | 98b5f258-635e-11e4-8bdd-0025647390c1 | -| metadata | {"name": "cirros-0.3.2-x86_64-uec"} | -| meter | image | -| project_id | 2cc3a7bb859b4bacbeab0aa9ca673033 | -| recorded_at | 2014-11-03T13:37:46.994458 | -| resource_id | 9b651dfd-7d30-402b-972e-212b2c4bfb05 | -| source | openstack | -| timestamp | 2014-11-03T13:37:46 | -| type | gauge | -| unit | image | -| user_id | None | -| volume | 1.0 | -+-------------+--------------------------------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_sample_show_raises_command_err(self): - self.cc.new_samples.get.side_effect = exc.HTTPNotFound - - self.assertRaises(exc.CommandError, ceilometer_shell.do_sample_show, - self.cc, self.args) - - -class ShellSampleCreateCommandTest(utils.BaseTestCase): - - METER = 'instance' - METER_TYPE = 'gauge' - RESOURCE_ID = '0564c64c-3545-4e34-abfb-9d18e5f2f2f9' - SAMPLE_VOLUME = '1' - METER_UNIT = 'instance' - SAMPLE = [{ - u'counter_name': u'instance', - u'user_id': u'21b442b8101d407d8242b6610e0ed0eb', - u'resource_id': u'0564c64c-3545-4e34-abfb-9d18e5f2f2f9', - u'timestamp': u'2014-01-10T03: 05: 33.951170', - u'message_id': u'1247cbe6-79a4-11e3-a296-000c294c58e2', - u'source': u'384260c6987b451d8290e66e1f108082: openstack', - u'counter_unit': u'instance', - u'counter_volume': 1.0, - u'project_id': u'384260c6987b451d8290e66e1f108082', - u'counter_type': u'gauge', - u'resource_metadata': {u'display_name': u'test_name'} - }] - - def setUp(self): - super(ShellSampleCreateCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.samples = mock.Mock() - self.args = mock.Mock() - self.args.meter_name = self.METER - self.args.meter_type = self.METER_TYPE - self.args.meter_unit = self.METER_UNIT - self.args.resource_id = self.RESOURCE_ID - self.args.sample_volume = self.SAMPLE_VOLUME - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_sample_create(self): - ret_sample = [samples.OldSample(mock.Mock(), sample) - for sample in self.SAMPLE] - self.cc.samples.create.return_value = ret_sample - - ceilometer_shell.do_sample_create(self.cc, self.args) - - self.assertEqual('''\ -+-------------------+---------------------------------------------+ -| Property | Value | -+-------------------+---------------------------------------------+ -| message_id | 1247cbe6-79a4-11e3-a296-000c294c58e2 | -| name | instance | -| project_id | 384260c6987b451d8290e66e1f108082 | -| resource_id | 0564c64c-3545-4e34-abfb-9d18e5f2f2f9 | -| resource_metadata | {"display_name": "test_name"} | -| source | 384260c6987b451d8290e66e1f108082: openstack | -| timestamp | 2014-01-10T03: 05: 33.951170 | -| type | gauge | -| unit | instance | -| user_id | 21b442b8101d407d8242b6610e0ed0eb | -| volume | 1.0 | -+-------------------+---------------------------------------------+ -''', sys.stdout.getvalue()) - - def test_sample_create_with_invalid_resource_metadata(self): - self.args.resource_metadata = 'foo=bar' - with mock.patch('ceilometerclient.exc.CommandError') as e: - e.return_value = exc.BaseException() - self.assertRaises(exc.BaseException, - ceilometer_shell.do_sample_create, - self.cc, self.args) - e.assert_called_with('Invalid resource metadata, it should be a' - ' json string, like: \'{"foo":"bar"}\'') - - -class ShellSampleCreateListCommandTest(utils.BaseTestCase): - - SAMPLE = { - u'counter_name': u'image', - u'user_id': u'21b442b8101d407d8242b6610e0ed0eb', - u'resource_id': u'0564c64c-3545-4e34-abfb-9d18e5f2f2f9', - u'timestamp': u'2015-05-19T12:00:08.368574', - u'source': u'384260c6987b451d8290e66e1f108082: openstack', - u'counter_unit': u'image', - u'counter_volume': 1.0, - u'project_id': u'384260c6987b451d8290e66e1f108082', - u'resource_metadata': {}, - u'counter_type': u'cumulative' - } - - def setUp(self): - super(ShellSampleCreateListCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.samples = mock.Mock() - self.cc.samples.create_list = mock.Mock() - self.args = mock.Mock() - self.samples = [self.SAMPLE] * 5 - self.args.samples_list = json.dumps(self.samples) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_sample_create_list(self): - ret_samples = [samples.OldSample(mock.Mock(), - sample) for sample in self.samples] - self.cc.samples.create_list.return_value = ret_samples - ceilometer_shell.do_sample_create_list(self.cc, self.args) - self.cc.samples.create_list.assert_called_with(self.samples, - direct=mock.ANY) - self.assertEqual('''\ -+--------------------------------------+-------+------------+--------+-------\ -+----------------------------+ -| Resource ID | Name | Type | Volume | Unit \ -| Timestamp | -+--------------------------------------+-------+------------+--------+-------\ -+----------------------------+ -| 0564c64c-3545-4e34-abfb-9d18e5f2f2f9 | image | cumulative | 1.0 | image \ -| 2015-05-19T12:00:08.368574 | -| 0564c64c-3545-4e34-abfb-9d18e5f2f2f9 | image | cumulative | 1.0 | image \ -| 2015-05-19T12:00:08.368574 | -| 0564c64c-3545-4e34-abfb-9d18e5f2f2f9 | image | cumulative | 1.0 | image \ -| 2015-05-19T12:00:08.368574 | -| 0564c64c-3545-4e34-abfb-9d18e5f2f2f9 | image | cumulative | 1.0 | image \ -| 2015-05-19T12:00:08.368574 | -| 0564c64c-3545-4e34-abfb-9d18e5f2f2f9 | image | cumulative | 1.0 | image \ -| 2015-05-19T12:00:08.368574 | -+--------------------------------------+-------+------------+--------+-------\ -+----------------------------+ -''', sys.stdout.getvalue()) - - -class ShellQuerySamplesCommandTest(utils.BaseTestCase): - - SAMPLE = [{u'id': u'b55d1526-9929-11e3-a3f6-02163e5df1e6', - u'metadata': { - u'name1': u'value1', - u'name2': u'value2'}, - u'meter': 'instance', - u'project_id': u'35b17138-b364-4e6a-a131-8f3099c5be68', - u'resource_id': u'bd9431c1-8d69-4ad3-803a-8d4a6b89fd36', - u'source': u'openstack', - u'timestamp': u'2014-02-19T05:50:16.673604', - u'type': u'gauge', - u'unit': u'instance', - u'volume': 1, - u'user_id': 'efd87807-12d2-4b38-9c70-5f5c2ac427ff'}] - - QUERY = {"filter": {"and": [{"=": {"source": "openstack"}}, - {">": {"timestamp": "2014-02-19T05:50:16"}}]}, - "orderby": [{"timestamp": "desc"}, {"volume": "asc"}], - "limit": 10} - - def setUp(self): - super(ShellQuerySamplesCommandTest, self).setUp() - self.cc = mock.Mock() - self.args = mock.Mock() - self.args.filter = self.QUERY["filter"] - self.args.orderby = self.QUERY["orderby"] - self.args.limit = self.QUERY["limit"] - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_query(self): - ret_sample = [samples.Sample(mock.Mock(), sample) - for sample in self.SAMPLE] - self.cc.query_samples.query.return_value = ret_sample - - ceilometer_shell.do_query_samples(self.cc, self.args) - - self.assertEqual('''\ -+--------------------------------------+--------------------------------------\ -+----------+-------+--------+----------+----------------------------+ -| ID | Resource ID \ -| Meter | Type | Volume | Unit | Timestamp | -+--------------------------------------+--------------------------------------\ -+----------+-------+--------+----------+----------------------------+ -| b55d1526-9929-11e3-a3f6-02163e5df1e6 | bd9431c1-8d69-4ad3-803a-8d4a6b89fd36 \ -| instance | gauge | 1 | instance | 2014-02-19T05:50:16.673604 | -+--------------------------------------+--------------------------------------\ -+----------+-------+--------+----------+----------------------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_query_raises_command_error(self): - self.cc.query_samples.query.side_effect = exc.HTTPNotFound - - self.assertRaises(exc.CommandError, - ceilometer_shell.do_query_samples, - self.cc, self.args) - - -class ShellQueryAlarmsCommandTest(utils.BaseTestCase): - - ALARM = [{"alarm_actions": ["http://site:8000/alarm"], - "alarm_id": "768ff714-8cfb-4db9-9753-d484cb33a1cc", - "combination_rule": { - "alarm_ids": [ - "739e99cb-c2ec-4718-b900-332502355f38", - "153462d0-a9b8-4b5b-8175-9e4b05e9b856"], - "operator": "or"}, - "description": "An alarm", - "enabled": True, - "insufficient_data_actions": ["http://site:8000/nodata"], - "name": "SwiftObjectAlarm", - "ok_actions": ["http://site:8000/ok"], - "project_id": "c96c887c216949acbdfbd8b494863567", - "repeat_actions": False, - "state": "ok", - "severity": "critical", - "state_timestamp": "2014-02-20T10:37:15.589860", - "threshold_rule": None, - "timestamp": "2014-02-20T10:37:15.589856", - "time_constraints": [{"name": "test", "start": "0 23 * * *", - "duration": 10800}], - "type": "combination", - "user_id": "c96c887c216949acbdfbd8b494863567"}] - - QUERY = {"filter": {"and": [{"!=": {"state": "ok"}}, - {"=": {"type": "combination"}}]}, - "orderby": [{"state_timestamp": "desc"}], - "limit": 10} - - def setUp(self): - super(ShellQueryAlarmsCommandTest, self).setUp() - self.cc = mock.Mock() - self.args = mock.Mock() - self.args.filter = self.QUERY["filter"] - self.args.orderby = self.QUERY["orderby"] - self.args.limit = self.QUERY["limit"] - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_query(self): - ret_alarm = [alarms.Alarm(mock.Mock(), alarm) - for alarm in self.ALARM] - self.cc.query_alarms.query.return_value = ret_alarm - - ceilometer_shell.do_query_alarms(self.cc, self.args) - - self.assertEqual('''\ -+--------------------------------------+------------------+-------+----------+\ ----------+------------+-------------------------------------------------------\ ------------------------------------------------+-------------------------------\ --+ -| Alarm ID | Name | State | Severity \ -| Enabled | Continuous | Alarm condition \ - | Time constraints \ - | -+--------------------------------------+------------------+-------+----------+\ ----------+------------+-------------------------------------------------------\ ------------------------------------------------+--------------------------------+ -| 768ff714-8cfb-4db9-9753-d484cb33a1cc | SwiftObjectAlarm | ok | critical \ -| True | False | combinated states (OR) of \ -739e99cb-c2ec-4718-b900-332502355f38, 153462d0-a9b8-4b5b-8175-9e4b05e9b856 |\ - test at 0 23 * * * for 10800s | -+--------------------------------------+------------------+-------+----------+\ ----------+------------+-------------------------------------------------------\ ------------------------------------------------+------------------------------\ ---+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_time_constraints_compatibility(self): - # client should be backwards compatible - alarm_without_tc = dict(self.ALARM[0]) - del alarm_without_tc['time_constraints'] - - # NOTE(nsaje): Since we're accessing a nonexisting key in the resource, - # the resource is looking it up with the manager (which is a mock). - manager_mock = mock.Mock() - del manager_mock.get - ret_alarm = [alarms.Alarm(manager_mock, alarm_without_tc)] - self.cc.query_alarms.query.return_value = ret_alarm - - ceilometer_shell.do_query_alarms(self.cc, self.args) - - self.assertEqual('''\ -+--------------------------------------+------------------+-------+----------+\ ----------+------------+-------------------------------------------------------\ ------------------------------------------------+------------------+ -| Alarm ID | Name | State | Severity \ -| Enabled | Continuous | Alarm condition \ - | Time constraints | -+--------------------------------------+------------------+-------+----------+\ ----------+------------+-------------------------------------------------------\ ------------------------------------------------+------------------+ -| 768ff714-8cfb-4db9-9753-d484cb33a1cc | SwiftObjectAlarm | ok | critical \ -| True | False | combinated states (OR) of \ -739e99cb-c2ec-4718-b900-332502355f38, 153462d0-a9b8-4b5b-8175-9e4b05e9b856 \ -| None | -+--------------------------------------+------------------+-------+----------+\ ----------+------------+-------------------------------------------------------\ ------------------------------------------------+------------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_query_raises_command_err(self): - self.cc.query_alarms.query.side_effect = exc.HTTPNotFound - self.assertRaises(exc.CommandError, - ceilometer_shell.do_query_alarms, - self.cc, self.args) - - -class ShellQueryAlarmHistoryCommandTest(utils.BaseTestCase): - - ALARM_HISTORY = [{"alarm_id": "e8ff32f772a44a478182c3fe1f7cad6a", - "event_id": "c74a8611-6553-4764-a860-c15a6aabb5d0", - "detail": - "{\"threshold\": 42.0, \"evaluation_periods\": 4}", - "on_behalf_of": "92159030020611e3b26dde429e99ee8c", - "project_id": "b6f16144010811e387e4de429e99ee8c", - "timestamp": "2014-03-11T16:02:58.376261", - "type": "rule change", - "user_id": "3e5d11fda79448ac99ccefb20be187ca" - }] - - QUERY = {"filter": {"and": [{">": {"timestamp": "2014-03-11T16:02:58"}}, - {"=": {"type": "rule change"}}]}, - "orderby": [{"timestamp": "desc"}], - "limit": 10} - - def setUp(self): - super(ShellQueryAlarmHistoryCommandTest, self).setUp() - self.cc = mock.Mock() - self.args = mock.Mock() - self.args.filter = self.QUERY["filter"] - self.args.orderby = self.QUERY["orderby"] - self.args.limit = self.QUERY["limit"] - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_query(self): - ret_alarm_history = [alarms.AlarmChange(mock.Mock(), alarm_history) - for alarm_history in self.ALARM_HISTORY] - self.cc.query_alarm_history.query.return_value = ret_alarm_history - - ceilometer_shell.do_query_alarm_history(self.cc, self.args) - - self.assertEqual('''\ -+----------------------------------+--------------------------------------+-\ -------------+----------------------------------------------+----------------\ -------------+ -| Alarm ID | Event ID | \ -Type | Detail | Timestamp \ - | -+----------------------------------+--------------------------------------+-\ -------------+----------------------------------------------+----------------\ -------------+ -| e8ff32f772a44a478182c3fe1f7cad6a | c74a8611-6553-4764-a860-c15a6aabb5d0 | \ -rule change | {"threshold": 42.0, "evaluation_periods": 4} | 2014-03-11T16:0\ -2:58.376261 | -+----------------------------------+--------------------------------------+-\ -------------+----------------------------------------------+----------------\ -------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_query_raises_command_err(self): - self.cc.query_alarm_history.query.side_effect = exc.HTTPNotFound - self.assertRaises(exc.CommandError, - ceilometer_shell.do_query_alarm_history, - self.cc, self.args) - - -class ShellStatisticsTest(utils.BaseTestCase): - def setUp(self): - super(ShellStatisticsTest, self).setUp() - self.cc = mock.Mock() - self.displays = { - 'duration': 'Duration', - 'duration_end': 'Duration End', - 'duration_start': 'Duration Start', - 'period': 'Period', - 'period_end': 'Period End', - 'period_start': 'Period Start', - 'groupby': 'Group By', - 'avg': 'Avg', - 'count': 'Count', - 'max': 'Max', - 'min': 'Min', - 'sum': 'Sum', - 'stddev': 'Standard deviation', - 'cardinality': 'Cardinality' - } - self.args = mock.Mock() - self.args.meter_name = 'instance' - self.args.aggregate = [] - self.args.groupby = None - self.args.query = None - - def test_statistics_list_simple(self): - samples = [ - {u'count': 135, - u'duration_start': u'2013-02-04T10:51:42', - u'min': 1.0, - u'max': 1.0, - u'duration_end': - u'2013-02-05T15:46:09', - u'duration': 1734.0, - u'avg': 1.0, - u'sum': 135.0}, - ] - fields = [ - 'period', - 'period_start', - 'period_end', - 'max', - 'min', - 'avg', - 'sum', - 'count', - 'duration', - 'duration_start', - 'duration_end', - ] - statistics_ret = [ - statistics.Statistics(mock.Mock(), sample) for sample in samples - ] - self.cc.statistics.list.return_value = statistics_ret - with mock.patch('ceilometerclient.v2.shell.utils.print_list') as pmock: - ceilometer_shell.do_statistics(self.cc, self.args) - pmock.assert_called_with( - statistics_ret, - fields, - [self.displays[f] for f in fields] - ) - - def test_statistics_list_groupby(self): - samples = [ - {u'count': 135, - u'duration_start': u'2013-02-04T10:51:42', - u'min': 1.0, - u'max': 1.0, - u'duration_end': - u'2013-02-05T15:46:09', - u'duration': 1734.0, - u'avg': 1.0, - u'sum': 135.0, - u'groupby': {u'resource_id': u'foo'} - }, - {u'count': 12, - u'duration_start': u'2013-02-04T10:51:42', - u'min': 1.0, - u'max': 1.0, - u'duration_end': - u'2013-02-05T15:46:09', - u'duration': 1734.0, - u'avg': 1.0, - u'sum': 12.0, - u'groupby': {u'resource_id': u'bar'} - }, - ] - fields = [ - 'period', - 'period_start', - 'period_end', - 'groupby', - 'max', - 'min', - 'avg', - 'sum', - 'count', - 'duration', - 'duration_start', - 'duration_end', - ] - self.args.groupby = 'resource_id' - statistics_ret = [ - statistics.Statistics(mock.Mock(), sample) for sample in samples - ] - self.cc.statistics.list.return_value = statistics_ret - with mock.patch('ceilometerclient.v2.shell.utils.print_list') as pmock: - ceilometer_shell.do_statistics(self.cc, self.args) - pmock.assert_called_with( - statistics_ret, - fields, - [self.displays[f] for f in fields], - ) - - def test_statistics_list_aggregates(self): - samples = [ - {u'aggregate': {u'cardinality/resource_id': 4.0, u'count': 2.0}, - u'count': 2, - u'duration': 0.442451, - u'duration_end': u'2014-03-12T14:00:21.774154', - u'duration_start': u'2014-03-12T14:00:21.331703', - u'groupby': None, - u'period': 0, - u'period_end': u'2014-03-12T14:00:21.774154', - u'period_start': u'2014-03-12T14:00:21.331703', - u'unit': u'instance', - }, - ] - fields = [ - 'period', - 'period_start', - 'period_end', - 'count', - 'cardinality/resource_id', - 'duration', - 'duration_start', - 'duration_end', - ] - self.args.aggregate = ['count', 'cardinality<-resource_id'] - statistics_ret = [ - statistics.Statistics(mock.Mock(), sample) for sample in samples - ] - self.cc.statistics.list.return_value = statistics_ret - with mock.patch('ceilometerclient.v2.shell.utils.print_list') as pmock: - ceilometer_shell.do_statistics(self.cc, self.args) - pmock.assert_called_with( - statistics_ret, - fields, - [self.displays.get(f, f) for f in fields], - ) - - -class ShellEmptyIdTest(utils.BaseTestCase): - """Test empty field which will cause calling incorrect rest uri.""" - - def _test_entity_action_with_empty_values(self, entity, - *args, **kwargs): - positional = kwargs.pop('positional', False) - for value in ('', ' ', ' ', '\t'): - self._test_entity_action_with_empty_value(entity, value, - positional, *args) - - def _test_entity_action_with_empty_value(self, entity, value, - positional, *args): - new_args = [value] if positional else ['--%s' % entity, value] - argv = list(args) + new_args - shell = base_shell.CeilometerShell() - with mock.patch('ceilometerclient.exc.CommandError') as e: - e.return_value = exc.BaseException() - self.assertRaises(exc.BaseException, shell.parse_args, argv) - entity = entity.replace('-', '_') - e.assert_called_with('%s should not be empty' % entity) - - def _test_alarm_action_with_empty_ids(self, method, *args): - args = [method] + list(args) - self._test_entity_action_with_empty_values('alarm_id', - positional=True, *args) - - def test_alarm_show_with_empty_id(self): - self._test_alarm_action_with_empty_ids('alarm-show') - - def test_alarm_update_with_empty_id(self): - self._test_alarm_action_with_empty_ids('alarm-update') - - def test_alarm_threshold_update_with_empty_id(self): - self._test_alarm_action_with_empty_ids('alarm-threshold-update') - - def test_alarm_combination_update_with_empty_id(self): - self._test_alarm_action_with_empty_ids('alarm-combination-update') - - def test_alarm_gnocchi_resources_update_with_empty_id(self): - self._test_alarm_action_with_empty_ids( - 'alarm-gnocchi-resources-threshold-update') - - def test_alarm_gnocchi_aggr_by_resources_update_with_empty_id(self): - self._test_alarm_action_with_empty_ids( - 'alarm-gnocchi-aggregation-by-resources-threshold-update') - - def test_alarm_gnocchi_aggr_by_metrics_update_with_empty_id(self): - self._test_alarm_action_with_empty_ids( - 'alarm-gnocchi-aggregation-by-metrics-threshold-update') - - def test_alarm_delete_with_empty_id(self): - self._test_alarm_action_with_empty_ids('alarm-delete') - - def test_alarm_state_get_with_empty_id(self): - self._test_alarm_action_with_empty_ids('alarm-state-get') - - def test_alarm_state_set_with_empty_id(self): - args = ['alarm-state-set', '--state', 'ok'] - self._test_alarm_action_with_empty_ids(*args) - - def test_alarm_history_with_empty_id(self): - self._test_alarm_action_with_empty_ids('alarm-history') - - def test_event_show_with_empty_message_id(self): - args = ['event-show'] - self._test_entity_action_with_empty_values('message_id', *args) - - def test_resource_show_with_empty_id(self): - args = ['resource-show'] - self._test_entity_action_with_empty_values('resource_id', *args) - - def test_sample_list_with_empty_meter(self): - args = ['sample-list'] - self._test_entity_action_with_empty_values('meter', *args) - - def test_sample_create_with_empty_meter(self): - args = ['sample-create', '-r', 'x', '--meter-type', 'gauge', - '--meter-unit', 'B', '--sample-volume', '1'] - self._test_entity_action_with_empty_values('meter-name', *args) - - def test_statistics_with_empty_meter(self): - args = ['statistics'] - self._test_entity_action_with_empty_values('meter', *args) - - def test_trait_description_list_with_empty_event_type(self): - args = ['trait-description-list'] - self._test_entity_action_with_empty_values('event_type', *args) - - def test_trait_list_with_empty_event_type(self): - args = ['trait-list', '--trait_name', 'x'] - self._test_entity_action_with_empty_values('event_type', *args) - - def test_trait_list_with_empty_trait_name(self): - args = ['trait-list', '--event_type', 'x'] - self._test_entity_action_with_empty_values('trait_name', *args) - - -class ShellObsoletedArgsTest(utils.BaseTestCase): - """Test arguments that have been obsoleted.""" - - def _test_entity_obsoleted(self, entity, value, positional, *args): - new_args = [value] if positional else ['--%s' % entity, value] - argv = list(args) + new_args - shell = base_shell.CeilometerShell() - with mock.patch('sys.stdout', new_callable=six.StringIO) as stdout: - shell.parse_args(argv) - self.assertIn('obsolete', stdout.getvalue()) - - def test_obsolete_alarm_id(self): - for method in ['alarm-show', 'alarm-update', 'alarm-threshold-update', - 'alarm-combination-update', 'alarm-delete', - 'alarm-state-get', 'alarm-history']: - self._test_entity_obsoleted('alarm_id', 'abcde', False, method) - - -class ShellEventListCommandTest(utils.BaseTestCase): - - EVENTS = [ - { - "generated": "2015-01-12T04:03:25.741471", - "message_id": "fb2bef58-88af-4380-8698-e0f18fcf452d", - "event_type": "compute.instance.create.start", - "traits": [{ - "name": "state", - "type": "string", - "value": "building", - }], - }, - { - "generated": "2015-01-12T04:03:28.452495", - "message_id": "9b20509a-576b-4995-acfa-1a24ee5cf49f", - "event_type": "compute.instance.create.end", - "traits": [{ - "name": "state", - "type": "string", - "value": "active", - }], - }, - ] - - def setUp(self): - super(ShellEventListCommandTest, self).setUp() - self.cc = mock.Mock() - self.args = mock.Mock() - self.args.query = None - self.args.no_traits = None - self.args.limit = None - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_event_list(self): - ret_events = [events.Event(mock.Mock(), event) - for event in self.EVENTS] - self.cc.events.list.return_value = ret_events - ceilometer_shell.do_event_list(self.cc, self.args) - self.assertEqual('''\ -+--------------------------------------+-------------------------------+\ -----------------------------+-------------------------------+ -| Message ID | Event Type |\ - Generated | Traits | -+--------------------------------------+-------------------------------+\ -----------------------------+-------------------------------+ -| fb2bef58-88af-4380-8698-e0f18fcf452d | compute.instance.create.start |\ - 2015-01-12T04:03:25.741471 | +-------+--------+----------+ | -| | |\ - | | name | type | value | | -| | |\ - | +-------+--------+----------+ | -| | |\ - | | state | string | building | | -| | |\ - | +-------+--------+----------+ | -| 9b20509a-576b-4995-acfa-1a24ee5cf49f | compute.instance.create.end |\ - 2015-01-12T04:03:28.452495 | +-------+--------+--------+ | -| | |\ - | | name | type | value | | -| | |\ - | +-------+--------+--------+ | -| | |\ - | | state | string | active | | -| | |\ - | +-------+--------+--------+ | -+--------------------------------------+-------------------------------+\ -----------------------------+-------------------------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_event_list_no_traits(self): - self.args.no_traits = True - ret_events = [events.Event(mock.Mock(), event) - for event in self.EVENTS] - self.cc.events.list.return_value = ret_events - ceilometer_shell.do_event_list(self.cc, self.args) - self.assertEqual('''\ -+--------------------------------------+-------------------------------\ -+----------------------------+ -| Message ID | Event Type \ -| Generated | -+--------------------------------------+-------------------------------\ -+----------------------------+ -| fb2bef58-88af-4380-8698-e0f18fcf452d | compute.instance.create.start \ -| 2015-01-12T04:03:25.741471 | -| 9b20509a-576b-4995-acfa-1a24ee5cf49f | compute.instance.create.end \ -| 2015-01-12T04:03:28.452495 | -+--------------------------------------+-------------------------------\ -+----------------------------+ -''', sys.stdout.getvalue()) - - -class ShellShadowedArgsTest(test_shell.ShellTestBase): - - def _test_shadowed_args_alarm(self, command, args, method): - self.make_env(test_shell.FAKE_V2_ENV) - cli_args = [ - '--os-project-id', '0ba30185ddf44834914a0b859d244c56', - '--os-user-id', '85f59b3b17484ccb974c50596023bf8c', - '--debug', command, - '--project-id', 'the-project-id-i-want-to-set', - '--user-id', 'the-user-id-i-want-to-set', - '--name', 'project-id-test'] + args - with mock.patch.object(alarms.AlarmManager, method) as mocked: - with mock.patch('ceilometerclient.apiclient.' - 'client.HTTPClient.client_request') as request: - request.site_effect = exceptions.EndpointNotFound - base_shell.main(cli_args) - args, kwargs = mocked.call_args - self.assertEqual('the-project-id-i-want-to-set', - kwargs.get('project_id')) - self.assertEqual('the-user-id-i-want-to-set', - kwargs.get('user_id')) - - def test_shadowed_args_threshold_alarm(self): - cli_args = ['--meter-name', 'cpu', '--threshold', '90'] - self._test_shadowed_args_alarm('alarm-create', cli_args, 'create') - self._test_shadowed_args_alarm('alarm-threshold-create', - cli_args, 'create') - cli_args += ['--alarm_id', '437b7ed0-3733-4054-a877-e9a297b8be85'] - self._test_shadowed_args_alarm('alarm-update', cli_args, 'update') - self._test_shadowed_args_alarm('alarm-threshold-update', - cli_args, 'update') - - def test_shadowed_args_combination_alarm(self): - cli_args = ['--alarm_ids', 'fb16a05a-669d-414e-8bbe-93aa381df6a8', - '--alarm_ids', 'b189bcca-0a7b-49a9-a244-a927ac291881'] - self._test_shadowed_args_alarm('alarm-combination-create', - cli_args, 'create') - cli_args += ['--alarm_id', '437b7ed0-3733-4054-a877-e9a297b8be85'] - self._test_shadowed_args_alarm('alarm-combination-update', - cli_args, 'update') - - def test_shadowed_args_gnocchi_resources_threshold_alarm(self): - cli_args = [ - '--metric', 'cpu', - '--threshold', '80', - '--resource-type', 'instance', - '--resource-id', 'fb16a05a-669d-414e-8bbe-93aa381df6a8', - '--aggregation-method', 'last', - ] - self._test_shadowed_args_alarm('alarm-gnocchi-resources-' - 'threshold-create', - cli_args, 'create') - cli_args += ['--alarm_id', '437b7ed0-3733-4054-a877-e9a297b8be85'] - self._test_shadowed_args_alarm('alarm-gnocchi-resources-' - 'threshold-update', - cli_args, 'update') - - def test_shadowed_args_gnocchi_aggr_by_resources_threshold_alarm(self): - cli_args = [ - '--metric', 'cpu', - '--threshold', '80', - '--resource-type', 'instance', - '--aggregation-method', 'last', - '--query', '"server_group":"my_autoscaling_group"', - ] - self._test_shadowed_args_alarm('alarm-gnocchi-aggregation-' - 'by-resources-threshold-create', - cli_args, 'create') - cli_args += ['--alarm_id', '437b7ed0-3733-4054-a877-e9a297b8be85'] - self._test_shadowed_args_alarm('alarm-gnocchi-aggregation-' - 'by-resources-threshold-update', - cli_args, 'update') - - def test_shadowed_args_gnocchi_aggr_by_metrics_threshold_alarm(self): - cli_args = [ - '-m', 'b3d9d8ab-05e8-439f-89ad-5e978dd2a5eb', - '-m', '009d4faf-c275-46f0-8f2d-670b15bac2b0', - '--threshold', '80', - '--aggregation-method', 'last', - ] - self._test_shadowed_args_alarm('alarm-gnocchi-aggregation-' - 'by-metrics-threshold-create', - cli_args, 'create') - cli_args += ['--alarm_id', '437b7ed0-3733-4054-a877-e9a297b8be85'] - self._test_shadowed_args_alarm('alarm-gnocchi-aggregation-' - 'by-metrics-threshold-update', - cli_args, 'update') - - @mock.patch.object(samples.OldSampleManager, 'create') - def test_shadowed_args_sample_create(self, mocked): - self.make_env(test_shell.FAKE_V2_ENV) - cli_args = [ - '--os-project-id', '0ba30185ddf44834914a0b859d244c56', - '--os-user-id', '85f59b3b17484ccb974c50596023bf8c', - '--debug', 'sample-create', - '--project-id', 'the-project-id-i-want-to-set', - '--user-id', 'the-user-id-i-want-to-set', - '--resource-id', 'b666633d-9bb6-4e05-89c0-ee5a8752fb0b', - '--meter-name', 'cpu', - '--meter-type', 'cumulative', - '--meter-unit', 'ns', - '--sample-volume', '10086', - ] - with mock.patch('ceilometerclient.apiclient.client.' - 'HTTPClient.client_request') as client_request: - client_request.site_effect = exceptions.EndpointNotFound - base_shell.main(cli_args) - args, kwargs = mocked.call_args - self.assertEqual('the-project-id-i-want-to-set', - kwargs.get('project_id')) - self.assertEqual('the-user-id-i-want-to-set', - kwargs.get('user_id')) - - -class ShellCapabilityShowTest(utils.BaseTestCase): - - CAPABILITIES = { - "alarm_storage": { - "storage:production_ready": True - }, - "api": { - "alarms:query:complex": True, - "alarms:query:simple": True - }, - "event_storage": { - "storage:production_ready": True - }, - "storage": { - "storage:production_ready": True - }, - } - - def setUp(self): - super(ShellCapabilityShowTest, self).setUp() - self.cc = mock.Mock() - self.args = mock.Mock() - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_capability_show(self): - _cap = capabilities.Capabilities(mock.Mock, self.CAPABILITIES) - self.cc.capabilities.get.return_value = _cap - - ceilometer_shell.do_capabilities(self.cc, self.args) - self.assertEqual('''\ -+---------------+----------------------------------+ -| Property | Value | -+---------------+----------------------------------+ -| alarm_storage | "storage:production_ready": true | -| api | "alarms:query:complex": true, | -| | "alarms:query:simple": true | -| event_storage | "storage:production_ready": true | -| storage | "storage:production_ready": true | -+---------------+----------------------------------+ -''', sys.stdout.getvalue()) - - -class ShellMeterListCommandTest(utils.BaseTestCase): - - METER = { - "name": 'image', - "resource_id": "resource-id", - "meter": "image", - "project_id": "project", - "type": "gauge", - "unit": "image", - } - - def setUp(self): - super(ShellMeterListCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.meters.list = mock.Mock() - self.args = mock.MagicMock() - self.args.limit = None - self.args.unique = False - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_meter_list(self): - meter = meters.Meter(mock.Mock(), self.METER) - self.cc.meters.list.return_value = [meter] - - ceilometer_shell.do_meter_list(self.cc, self.args) - self.cc.meters.list.assert_called_once_with(q=[], limit=None, - unique=False) - - self.assertEqual('''\ -+-------+-------+-------+-------------+---------+------------+ -| Name | Type | Unit | Resource ID | User ID | Project ID | -+-------+-------+-------+-------------+---------+------------+ -| image | gauge | image | resource-id | | project | -+-------+-------+-------+-------------+---------+------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_unique_meter_list(self): - self.args.unique = True - meter = meters.Meter(mock.Mock(), self.METER) - self.cc.meters.list.return_value = [meter] - - ceilometer_shell.do_meter_list(self.cc, self.args) - self.cc.meters.list.assert_called_once_with(q=[], limit=None, - unique=True) - - self.assertEqual('''\ -+-------+-------+-------+-------------+---------+------------+ -| Name | Type | Unit | Resource ID | User ID | Project ID | -+-------+-------+-------+-------------+---------+------------+ -| image | gauge | image | resource-id | | project | -+-------+-------+-------+-------------+---------+------------+ -''', sys.stdout.getvalue()) - - -class ShellResourceListCommandTest(utils.BaseTestCase): - - RESOURCE = { - "source": "openstack", - "resource_id": "resource-id", - "project_id": "project", - "user_id": "user" - } - - def setUp(self): - super(ShellResourceListCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.resources.list = mock.Mock() - self.args = mock.MagicMock() - self.args.limit = None - self.args.meter_links = None - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_resource_list(self): - resource = resources.Resource(mock.Mock(), self.RESOURCE) - self.cc.resources.list.return_value = [resource] - ceilometer_shell.do_resource_list(self.cc, self.args) - self.cc.resources.list.assert_called_once_with(q=[], - limit=None) - - self.assertEqual('''\ -+-------------+-----------+---------+------------+ -| Resource ID | Source | User ID | Project ID | -+-------------+-----------+---------+------------+ -| resource-id | openstack | user | project | -+-------------+-----------+---------+------------+ -''', sys.stdout.getvalue()) - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_resource_list_with_links(self): - resource = resources.Resource(mock.Mock(), self.RESOURCE) - self.cc.resources.list.return_value = [resource] - ceilometer_shell.do_resource_list(self.cc, self.args) - self.cc.resources.list.assert_called_once_with(q=[], - limit=None) - self.assertEqual('''\ -+-------------+-----------+---------+------------+ -| Resource ID | Source | User ID | Project ID | -+-------------+-----------+---------+------------+ -| resource-id | openstack | user | project | -+-------------+-----------+---------+------------+ -''', sys.stdout.getvalue()) - - -class ShellEventTypeListCommandTest(utils.BaseTestCase): - - EVENT_TYPE = { - "event_type": "test_event" - } - - def setUp(self): - super(ShellEventTypeListCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.event_types.list = mock.Mock() - self.args = mock.Mock() - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_sample_show(self): - event_type = event_types.EventType(mock.Mock(), self.EVENT_TYPE) - self.cc.event_types.list.return_value = [event_type] - - ceilometer_shell.do_event_type_list(self.cc, self.args) - self.cc.event_types.list.assert_called_once_with() - - self.assertEqual('''\ -+------------+ -| Event Type | -+------------+ -| test_event | -+------------+ -''', sys.stdout.getvalue()) - - -class ShellTraitsListCommandTest(utils.BaseTestCase): - - TRAIT = { - "name": "test", - "value": "test", - "type": "string", - } - - def setUp(self): - super(ShellTraitsListCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.traits.list = mock.Mock() - self.args = mock.Mock() - self.args.event_type = "test" - self.args.trait_name = "test" - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_trait_list(self): - trait = traits.Trait(mock.Mock(), self.TRAIT) - self.cc.traits.list.return_value = [trait] - - ceilometer_shell.do_trait_list(self.cc, self.args) - self.cc.traits.list.assert_called_once_with(self.args.event_type, - self.args.trait_name) - - self.assertEqual('''\ -+------------+-------+-----------+ -| Trait Name | Value | Data Type | -+------------+-------+-----------+ -| test | test | string | -+------------+-------+-----------+ -''', sys.stdout.getvalue()) - - -class ShellTraitsDescriptionListCommandTest(utils.BaseTestCase): - - TRAIT_DESCRIPTION = { - "name": "test", - "type": "string", - } - - def setUp(self): - super(ShellTraitsDescriptionListCommandTest, self).setUp() - self.cc = mock.Mock() - self.cc.trait_descriptions.list = mock.Mock() - self.args = mock.Mock() - self.args.event_type = "test" - - @mock.patch('sys.stdout', new=six.StringIO()) - def test_traits_description_list(self): - trait_desc = trait_descriptions.TraitDescription( - mock.Mock(), self.TRAIT_DESCRIPTION) - self.cc.trait_descriptions.list.return_value = [trait_desc] - - ceilometer_shell.do_trait_description_list(self.cc, self.args) - self.cc.trait_descriptions.list.assert_called_once_with( - self.args.event_type) - - self.assertEqual('''\ -+------------+-----------+ -| Trait Name | Data Type | -+------------+-----------+ -| test | string | -+------------+-----------+ -''', sys.stdout.getvalue()) |