From f4d3810c3e978382bf80e59d5d9a5dcd2485f56c Mon Sep 17 00:00:00 2001 From: Akihiro Motoki Date: Mon, 6 Mar 2017 14:20:20 +0900 Subject: command list: Move network meter appropriately 'network meter' commands are not placed alphabetically in the command list. It was due to the incorrect filename. Trivial Fix Change-Id: I7a76d3133915883cf41be8c7430def284d292d6e --- openstackclient/network/v2/meter.py | 190 ------------- openstackclient/network/v2/network_meter.py | 190 +++++++++++++ .../tests/functional/network/v2/test_meter.py | 102 ------- .../functional/network/v2/test_network_meter.py | 102 +++++++ .../tests/unit/network/v2/test_meter.py | 304 --------------------- .../tests/unit/network/v2/test_network_meter.py | 304 +++++++++++++++++++++ 6 files changed, 596 insertions(+), 596 deletions(-) delete mode 100644 openstackclient/network/v2/meter.py create mode 100644 openstackclient/network/v2/network_meter.py delete mode 100644 openstackclient/tests/functional/network/v2/test_meter.py create mode 100644 openstackclient/tests/functional/network/v2/test_network_meter.py delete mode 100644 openstackclient/tests/unit/network/v2/test_meter.py create mode 100644 openstackclient/tests/unit/network/v2/test_network_meter.py (limited to 'openstackclient') diff --git a/openstackclient/network/v2/meter.py b/openstackclient/network/v2/meter.py deleted file mode 100644 index df0e1da1..00000000 --- a/openstackclient/network/v2/meter.py +++ /dev/null @@ -1,190 +0,0 @@ -# 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. -# - -"""Metering Label Implementations""" - -import logging - -from osc_lib.command import command -from osc_lib import exceptions -from osc_lib import utils - -from openstackclient.i18n import _ -from openstackclient.identity import common as identity_common -from openstackclient.network import sdk_utils - -LOG = logging.getLogger(__name__) - - -def _get_columns(item): - column_map = { - 'is_shared': 'shared', - 'tenant_id': 'project_id', - } - return sdk_utils.get_osc_show_columns_for_sdk_resource(item, column_map) - - -def _get_attrs(client_manager, parsed_args): - attrs = {} - - if parsed_args.description is not None: - attrs['description'] = parsed_args.description - if parsed_args.project is not None and 'project' in parsed_args: - identity_client = client_manager.identity - project_id = identity_common.find_project( - identity_client, - parsed_args.project, - parsed_args.project_domain, - ).id - attrs['tenant_id'] = project_id - if parsed_args.share: - attrs['shared'] = True - if parsed_args.no_share: - attrs['shared'] = False - if parsed_args.name is not None: - attrs['name'] = parsed_args.name - - return attrs - - -# TODO(ankur-gupta-f): Use the SDK resource mapped attribute names once the -# OSC minimum requirements include SDK 1.0. -class CreateMeter(command.ShowOne): - _description = _("Create network meter") - - def get_parser(self, prog_name): - parser = super(CreateMeter, self).get_parser(prog_name) - - parser.add_argument( - '--description', - metavar='', - help=_("Create description for meter") - ) - parser.add_argument( - '--project', - metavar='', - help=_("Owner's project (name or ID)") - ) - - identity_common.add_project_domain_option_to_parser(parser) - share_group = parser.add_mutually_exclusive_group() - share_group.add_argument( - '--share', - action='store_true', - default=None, - help=_("Share meter between projects") - ) - share_group.add_argument( - '--no-share', - action='store_true', - help=_("Do not share meter between projects") - ) - parser.add_argument( - 'name', - metavar='', - help=_('Name of meter'), - ) - - return parser - - def take_action(self, parsed_args): - client = self.app.client_manager.network - attrs = _get_attrs(self.app.client_manager, parsed_args) - obj = client.create_metering_label(**attrs) - display_columns, columns = _get_columns(obj) - data = utils.get_item_properties(obj, columns, formatters={}) - - return (display_columns, data) - - -# TODO(ankur-gupta-f): Use the SDK resource mapped attribute names once the -# OSC minimum requirements include SDK 1.0. -class DeleteMeter(command.Command): - _description = _("Delete network meter") - - def get_parser(self, prog_name): - parser = super(DeleteMeter, self).get_parser(prog_name) - - parser.add_argument( - 'meter', - metavar='', - nargs='+', - help=_('Meter to delete (name or ID)') - ) - return parser - - def take_action(self, parsed_args): - client = self.app.client_manager.network - result = 0 - - for meter in parsed_args.meter: - try: - obj = client.find_metering_label(meter, ignore_missing=False) - client.delete_metering_label(obj) - except Exception as e: - result += 1 - LOG.error(_("Failed to delete meter with " - "ID '%(meter)s': %(e)s"), - {"meter": meter, "e": e}) - if result > 0: - total = len(parsed_args.meter) - msg = (_("%(result)s of %(total)s meters failed " - "to delete.") % {"result": result, "total": total}) - raise exceptions.CommandError(msg) - - -class ListMeter(command.Lister): - _description = _("List network meters") - - def take_action(self, parsed_args): - client = self.app.client_manager.network - - columns = ( - 'id', - 'name', - 'description', - 'shared', - ) - column_headers = ( - 'ID', - 'Name', - 'Description', - 'Shared', - ) - - data = client.metering_labels() - return (column_headers, - (utils.get_item_properties( - s, columns, - ) for s in data)) - - -class ShowMeter(command.ShowOne): - _description = _("Show network meter") - - def get_parser(self, prog_name): - parser = super(ShowMeter, self).get_parser(prog_name) - parser.add_argument( - 'meter', - metavar='', - help=_('Meter to display (name or ID)') - ) - return parser - - def take_action(self, parsed_args): - client = self.app.client_manager.network - obj = client.find_metering_label(parsed_args.meter, - ignore_missing=False) - display_columns, columns = _get_columns(obj) - data = utils.get_item_properties(obj, columns) - return display_columns, data diff --git a/openstackclient/network/v2/network_meter.py b/openstackclient/network/v2/network_meter.py new file mode 100644 index 00000000..df0e1da1 --- /dev/null +++ b/openstackclient/network/v2/network_meter.py @@ -0,0 +1,190 @@ +# 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. +# + +"""Metering Label Implementations""" + +import logging + +from osc_lib.command import command +from osc_lib import exceptions +from osc_lib import utils + +from openstackclient.i18n import _ +from openstackclient.identity import common as identity_common +from openstackclient.network import sdk_utils + +LOG = logging.getLogger(__name__) + + +def _get_columns(item): + column_map = { + 'is_shared': 'shared', + 'tenant_id': 'project_id', + } + return sdk_utils.get_osc_show_columns_for_sdk_resource(item, column_map) + + +def _get_attrs(client_manager, parsed_args): + attrs = {} + + if parsed_args.description is not None: + attrs['description'] = parsed_args.description + if parsed_args.project is not None and 'project' in parsed_args: + identity_client = client_manager.identity + project_id = identity_common.find_project( + identity_client, + parsed_args.project, + parsed_args.project_domain, + ).id + attrs['tenant_id'] = project_id + if parsed_args.share: + attrs['shared'] = True + if parsed_args.no_share: + attrs['shared'] = False + if parsed_args.name is not None: + attrs['name'] = parsed_args.name + + return attrs + + +# TODO(ankur-gupta-f): Use the SDK resource mapped attribute names once the +# OSC minimum requirements include SDK 1.0. +class CreateMeter(command.ShowOne): + _description = _("Create network meter") + + def get_parser(self, prog_name): + parser = super(CreateMeter, self).get_parser(prog_name) + + parser.add_argument( + '--description', + metavar='', + help=_("Create description for meter") + ) + parser.add_argument( + '--project', + metavar='', + help=_("Owner's project (name or ID)") + ) + + identity_common.add_project_domain_option_to_parser(parser) + share_group = parser.add_mutually_exclusive_group() + share_group.add_argument( + '--share', + action='store_true', + default=None, + help=_("Share meter between projects") + ) + share_group.add_argument( + '--no-share', + action='store_true', + help=_("Do not share meter between projects") + ) + parser.add_argument( + 'name', + metavar='', + help=_('Name of meter'), + ) + + return parser + + def take_action(self, parsed_args): + client = self.app.client_manager.network + attrs = _get_attrs(self.app.client_manager, parsed_args) + obj = client.create_metering_label(**attrs) + display_columns, columns = _get_columns(obj) + data = utils.get_item_properties(obj, columns, formatters={}) + + return (display_columns, data) + + +# TODO(ankur-gupta-f): Use the SDK resource mapped attribute names once the +# OSC minimum requirements include SDK 1.0. +class DeleteMeter(command.Command): + _description = _("Delete network meter") + + def get_parser(self, prog_name): + parser = super(DeleteMeter, self).get_parser(prog_name) + + parser.add_argument( + 'meter', + metavar='', + nargs='+', + help=_('Meter to delete (name or ID)') + ) + return parser + + def take_action(self, parsed_args): + client = self.app.client_manager.network + result = 0 + + for meter in parsed_args.meter: + try: + obj = client.find_metering_label(meter, ignore_missing=False) + client.delete_metering_label(obj) + except Exception as e: + result += 1 + LOG.error(_("Failed to delete meter with " + "ID '%(meter)s': %(e)s"), + {"meter": meter, "e": e}) + if result > 0: + total = len(parsed_args.meter) + msg = (_("%(result)s of %(total)s meters failed " + "to delete.") % {"result": result, "total": total}) + raise exceptions.CommandError(msg) + + +class ListMeter(command.Lister): + _description = _("List network meters") + + def take_action(self, parsed_args): + client = self.app.client_manager.network + + columns = ( + 'id', + 'name', + 'description', + 'shared', + ) + column_headers = ( + 'ID', + 'Name', + 'Description', + 'Shared', + ) + + data = client.metering_labels() + return (column_headers, + (utils.get_item_properties( + s, columns, + ) for s in data)) + + +class ShowMeter(command.ShowOne): + _description = _("Show network meter") + + def get_parser(self, prog_name): + parser = super(ShowMeter, self).get_parser(prog_name) + parser.add_argument( + 'meter', + metavar='', + help=_('Meter to display (name or ID)') + ) + return parser + + def take_action(self, parsed_args): + client = self.app.client_manager.network + obj = client.find_metering_label(parsed_args.meter, + ignore_missing=False) + display_columns, columns = _get_columns(obj) + data = utils.get_item_properties(obj, columns) + return display_columns, data diff --git a/openstackclient/tests/functional/network/v2/test_meter.py b/openstackclient/tests/functional/network/v2/test_meter.py deleted file mode 100644 index 7dce34e7..00000000 --- a/openstackclient/tests/functional/network/v2/test_meter.py +++ /dev/null @@ -1,102 +0,0 @@ -# Copyright (c) 2016, Intel Corporation. -# 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 re -import uuid - -from openstackclient.tests.functional import base - - -class TestMeter(base.TestCase): - """Functional tests for network meter.""" - - # NOTE(dtroyer): Do not normalize the setup and teardown of the resource - # creation and deletion. Little is gained when each test - # has its own needs and there are collisions when running - # tests in parallel. - - @classmethod - def setUpClass(cls): - # Set up some regex for matching below - cls.re_name = re.compile("name\s+\|\s+([^|]+?)\s+\|") - cls.re_shared = re.compile("shared\s+\|\s+(\S+)") - cls.re_description = re.compile("description\s+\|\s+([^|]+?)\s+\|") - - def test_meter_delete(self): - """Test create, delete multiple""" - name1 = uuid.uuid4().hex - name2 = uuid.uuid4().hex - - raw_output = self.openstack( - 'network meter create ' + name1, - ) - self.assertEqual( - name1, - re.search(self.re_name, raw_output).group(1), - ) - # Check if default shared values - self.assertEqual( - 'False', - re.search(self.re_shared, raw_output).group(1) - ) - - raw_output = self.openstack( - 'network meter create ' + name2, - ) - self.assertEqual( - name2, - re.search(self.re_name, raw_output).group(1), - ) - - raw_output = self.openstack( - 'network meter delete ' + name1 + ' ' + name2, - ) - self.assertOutput('', raw_output) - - def test_meter_list(self): - """Test create, list filters, delete""" - name1 = uuid.uuid4().hex - raw_output = self.openstack( - 'network meter create --description Test1 --share ' + name1, - ) - self.addCleanup(self.openstack, 'network meter delete ' + name1) - - self.assertEqual( - 'Test1', - re.search(self.re_description, raw_output).group(1), - ) - self.assertEqual( - 'True', - re.search(self.re_shared, raw_output).group(1), - ) - - name2 = uuid.uuid4().hex - raw_output = self.openstack( - 'network meter create --description Test2 --no-share ' + name2, - ) - self.addCleanup(self.openstack, 'network meter delete ' + name2) - - self.assertEqual( - 'Test2', - re.search(self.re_description, raw_output).group(1), - ) - self.assertEqual( - 'False', - re.search(self.re_shared, raw_output).group(1), - ) - - raw_output = self.openstack('network meter list') - self.assertIsNotNone(re.search(name1 + "\s+\|\s+Test1", raw_output)) - self.assertIsNotNone(re.search(name2 + "\s+\|\s+Test2", raw_output)) diff --git a/openstackclient/tests/functional/network/v2/test_network_meter.py b/openstackclient/tests/functional/network/v2/test_network_meter.py new file mode 100644 index 00000000..7dce34e7 --- /dev/null +++ b/openstackclient/tests/functional/network/v2/test_network_meter.py @@ -0,0 +1,102 @@ +# Copyright (c) 2016, Intel Corporation. +# 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 re +import uuid + +from openstackclient.tests.functional import base + + +class TestMeter(base.TestCase): + """Functional tests for network meter.""" + + # NOTE(dtroyer): Do not normalize the setup and teardown of the resource + # creation and deletion. Little is gained when each test + # has its own needs and there are collisions when running + # tests in parallel. + + @classmethod + def setUpClass(cls): + # Set up some regex for matching below + cls.re_name = re.compile("name\s+\|\s+([^|]+?)\s+\|") + cls.re_shared = re.compile("shared\s+\|\s+(\S+)") + cls.re_description = re.compile("description\s+\|\s+([^|]+?)\s+\|") + + def test_meter_delete(self): + """Test create, delete multiple""" + name1 = uuid.uuid4().hex + name2 = uuid.uuid4().hex + + raw_output = self.openstack( + 'network meter create ' + name1, + ) + self.assertEqual( + name1, + re.search(self.re_name, raw_output).group(1), + ) + # Check if default shared values + self.assertEqual( + 'False', + re.search(self.re_shared, raw_output).group(1) + ) + + raw_output = self.openstack( + 'network meter create ' + name2, + ) + self.assertEqual( + name2, + re.search(self.re_name, raw_output).group(1), + ) + + raw_output = self.openstack( + 'network meter delete ' + name1 + ' ' + name2, + ) + self.assertOutput('', raw_output) + + def test_meter_list(self): + """Test create, list filters, delete""" + name1 = uuid.uuid4().hex + raw_output = self.openstack( + 'network meter create --description Test1 --share ' + name1, + ) + self.addCleanup(self.openstack, 'network meter delete ' + name1) + + self.assertEqual( + 'Test1', + re.search(self.re_description, raw_output).group(1), + ) + self.assertEqual( + 'True', + re.search(self.re_shared, raw_output).group(1), + ) + + name2 = uuid.uuid4().hex + raw_output = self.openstack( + 'network meter create --description Test2 --no-share ' + name2, + ) + self.addCleanup(self.openstack, 'network meter delete ' + name2) + + self.assertEqual( + 'Test2', + re.search(self.re_description, raw_output).group(1), + ) + self.assertEqual( + 'False', + re.search(self.re_shared, raw_output).group(1), + ) + + raw_output = self.openstack('network meter list') + self.assertIsNotNone(re.search(name1 + "\s+\|\s+Test1", raw_output)) + self.assertIsNotNone(re.search(name2 + "\s+\|\s+Test2", raw_output)) diff --git a/openstackclient/tests/unit/network/v2/test_meter.py b/openstackclient/tests/unit/network/v2/test_meter.py deleted file mode 100644 index b393f7fa..00000000 --- a/openstackclient/tests/unit/network/v2/test_meter.py +++ /dev/null @@ -1,304 +0,0 @@ -# Copyright (c) 2016, Intel Corporation. -# 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 mock -from mock import call - -from osc_lib import exceptions - -from openstackclient.network.v2 import meter -from openstackclient.tests.unit.identity.v3 import fakes as identity_fakes_v3 -from openstackclient.tests.unit.network.v2 import fakes as network_fakes -from openstackclient.tests.unit import utils as tests_utils - - -class TestMeter(network_fakes.TestNetworkV2): - - def setUp(self): - super(TestMeter, self).setUp() - self.network = self.app.client_manager.network - self.projects_mock = self.app.client_manager.identity.projects - self.domains_mock = self.app.client_manager.identity.domains - - -class TestCreateMeter(TestMeter): - project = identity_fakes_v3.FakeProject.create_one_project() - domain = identity_fakes_v3.FakeDomain.create_one_domain() - - new_meter = ( - network_fakes.FakeNetworkMeter. - create_one_meter() - ) - columns = ( - 'description', - 'id', - 'name', - 'project_id', - 'shared', - ) - - data = ( - new_meter.description, - new_meter.id, - new_meter.name, - new_meter.project_id, - new_meter.shared, - ) - - def setUp(self): - super(TestCreateMeter, self).setUp() - self.network.create_metering_label = mock.Mock( - return_value=self.new_meter) - self.projects_mock.get.return_value = self.project - self.cmd = meter.CreateMeter(self.app, self.namespace) - - def test_create_no_options(self): - arglist = [] - verifylist = [] - - self.assertRaises(tests_utils.ParserException, self.check_parser, - self.cmd, arglist, verifylist) - - def test_create_default_options(self): - arglist = [ - self.new_meter.name, - ] - - verifylist = [ - ('name', self.new_meter.name), - ] - - parsed_args = self.check_parser(self.cmd, arglist, verifylist) - columns, data = (self.cmd.take_action(parsed_args)) - - self.network.create_metering_label.assert_called_once_with( - **{'name': self.new_meter.name} - ) - self.assertEqual(self.columns, columns) - self.assertEqual(self.data, data) - - def test_create_all_options(self): - arglist = [ - "--description", self.new_meter.description, - "--project", self.new_meter.project_id, - "--project-domain", self.domain.name, - "--share", - self.new_meter.name, - ] - - verifylist = [ - ('description', self.new_meter.description), - ('name', self.new_meter.name), - ('project', self.new_meter.project_id), - ('project_domain', self.domain.name), - ('share', True), - ] - - parsed_args = self.check_parser(self.cmd, arglist, verifylist) - columns, data = (self.cmd.take_action(parsed_args)) - - self.network.create_metering_label.assert_called_once_with( - **{'description': self.new_meter.description, - 'name': self.new_meter.name, - 'tenant_id': self.project.id, - 'shared': True, } - ) - self.assertEqual(self.columns, columns) - self.assertEqual(self.data, data) - - -class TestDeleteMeter(TestMeter): - - def setUp(self): - super(TestDeleteMeter, self).setUp() - - self.meter_list = \ - network_fakes.FakeNetworkMeter.create_meter(count=2) - - self.network.delete_metering_label = mock.Mock(return_value=None) - - self.network.find_metering_label = network_fakes \ - .FakeNetworkMeter.get_meter( - meter=self.meter_list - ) - - self.cmd = meter.DeleteMeter(self.app, self.namespace) - - def test_delete_one_meter(self): - arglist = [ - self.meter_list[0].name, - ] - verifylist = [ - ('meter', [self.meter_list[0].name]), - ] - - parsed_args = self.check_parser(self.cmd, arglist, verifylist) - - result = self.cmd.take_action(parsed_args) - - self.network.delete_metering_label.assert_called_once_with( - self.meter_list[0] - ) - self.assertIsNone(result) - - def test_delete_multiple_meters(self): - arglist = [] - for n in self.meter_list: - arglist.append(n.id) - verifylist = [ - ('meter', arglist), - ] - - parsed_args = self.check_parser(self.cmd, arglist, verifylist) - - result = self.cmd.take_action(parsed_args) - - calls = [] - for n in self.meter_list: - calls.append(call(n)) - self.network.delete_metering_label.assert_has_calls(calls) - self.assertIsNone(result) - - def test_delete_multiple_meter_exception(self): - arglist = [ - self.meter_list[0].id, - 'xxxx-yyyy-zzzz', - self.meter_list[1].id, - ] - verifylist = [ - ('meter', arglist), - ] - - parsed_args = self.check_parser(self.cmd, arglist, verifylist) - - return_find = [ - self.meter_list[0], - exceptions.NotFound('404'), - self.meter_list[1], - ] - self.network.find_meter = mock.Mock(side_effect=return_find) - - ret_delete = [ - None, - exceptions.NotFound('404'), - ] - self.network.delete_metering_label = mock.Mock(side_effect=ret_delete) - - self.assertRaises(exceptions.CommandError, self.cmd.take_action, - parsed_args) - - calls = [ - call(self.meter_list[0]), - call(self.meter_list[1]), - ] - self.network.delete_metering_label.assert_has_calls(calls) - - -class TestListMeter(TestMeter): - - meter_list = \ - network_fakes.FakeNetworkMeter.create_meter(count=2) - - columns = ( - 'ID', - 'Name', - 'Description', - 'Shared', - ) - - data = [] - - for meters in meter_list: - data.append(( - meters.id, - meters.name, - meters.description, - meters.shared, - )) - - def setUp(self): - super(TestListMeter, self).setUp() - - self.network.metering_labels = mock.Mock( - return_value=self.meter_list - ) - - self.cmd = meter.ListMeter(self.app, self.namespace) - - def test_meter_list(self): - arglist = [] - verifylist = [] - - parsed_args = self.check_parser(self.cmd, arglist, verifylist) - - columns, data = self.cmd.take_action(parsed_args) - - self.network.metering_labels.assert_called_with() - self.assertEqual(self.columns, columns) - self.assertEqual(self.data, list(data)) - - -class TestShowMeter(TestMeter): - new_meter = ( - network_fakes.FakeNetworkMeter. - create_one_meter() - ) - columns = ( - 'description', - 'id', - 'name', - 'project_id', - 'shared', - ) - - data = ( - new_meter.description, - new_meter.id, - new_meter.name, - new_meter.project_id, - new_meter.shared, - ) - - def setUp(self): - super(TestShowMeter, self).setUp() - - self.cmd = meter.ShowMeter(self.app, self.namespace) - - self.network.find_metering_label = \ - mock.Mock(return_value=self.new_meter) - - def test_show_no_options(self): - arglist = [] - verifylist = [] - - self.assertRaises(tests_utils.ParserException, self.check_parser, - self.cmd, arglist, verifylist) - - def test_meter_show_option(self): - arglist = [ - self.new_meter.name, - ] - verifylist = [ - ('meter', self.new_meter.name), - ] - parsed_args = self.check_parser(self.cmd, arglist, verifylist) - - columns, data = self.cmd.take_action(parsed_args) - - self.network.find_metering_label.assert_called_with( - self.new_meter.name, ignore_missing=False - ) - self.assertEqual(self.columns, columns) - self.assertEqual(self.data, data) diff --git a/openstackclient/tests/unit/network/v2/test_network_meter.py b/openstackclient/tests/unit/network/v2/test_network_meter.py new file mode 100644 index 00000000..2b96f7a6 --- /dev/null +++ b/openstackclient/tests/unit/network/v2/test_network_meter.py @@ -0,0 +1,304 @@ +# Copyright (c) 2016, Intel Corporation. +# 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 mock +from mock import call + +from osc_lib import exceptions + +from openstackclient.network.v2 import network_meter +from openstackclient.tests.unit.identity.v3 import fakes as identity_fakes_v3 +from openstackclient.tests.unit.network.v2 import fakes as network_fakes +from openstackclient.tests.unit import utils as tests_utils + + +class TestMeter(network_fakes.TestNetworkV2): + + def setUp(self): + super(TestMeter, self).setUp() + self.network = self.app.client_manager.network + self.projects_mock = self.app.client_manager.identity.projects + self.domains_mock = self.app.client_manager.identity.domains + + +class TestCreateMeter(TestMeter): + project = identity_fakes_v3.FakeProject.create_one_project() + domain = identity_fakes_v3.FakeDomain.create_one_domain() + + new_meter = ( + network_fakes.FakeNetworkMeter. + create_one_meter() + ) + columns = ( + 'description', + 'id', + 'name', + 'project_id', + 'shared', + ) + + data = ( + new_meter.description, + new_meter.id, + new_meter.name, + new_meter.project_id, + new_meter.shared, + ) + + def setUp(self): + super(TestCreateMeter, self).setUp() + self.network.create_metering_label = mock.Mock( + return_value=self.new_meter) + self.projects_mock.get.return_value = self.project + self.cmd = network_meter.CreateMeter(self.app, self.namespace) + + def test_create_no_options(self): + arglist = [] + verifylist = [] + + self.assertRaises(tests_utils.ParserException, self.check_parser, + self.cmd, arglist, verifylist) + + def test_create_default_options(self): + arglist = [ + self.new_meter.name, + ] + + verifylist = [ + ('name', self.new_meter.name), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + columns, data = (self.cmd.take_action(parsed_args)) + + self.network.create_metering_label.assert_called_once_with( + **{'name': self.new_meter.name} + ) + self.assertEqual(self.columns, columns) + self.assertEqual(self.data, data) + + def test_create_all_options(self): + arglist = [ + "--description", self.new_meter.description, + "--project", self.new_meter.project_id, + "--project-domain", self.domain.name, + "--share", + self.new_meter.name, + ] + + verifylist = [ + ('description', self.new_meter.description), + ('name', self.new_meter.name), + ('project', self.new_meter.project_id), + ('project_domain', self.domain.name), + ('share', True), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + columns, data = (self.cmd.take_action(parsed_args)) + + self.network.create_metering_label.assert_called_once_with( + **{'description': self.new_meter.description, + 'name': self.new_meter.name, + 'tenant_id': self.project.id, + 'shared': True, } + ) + self.assertEqual(self.columns, columns) + self.assertEqual(self.data, data) + + +class TestDeleteMeter(TestMeter): + + def setUp(self): + super(TestDeleteMeter, self).setUp() + + self.meter_list = \ + network_fakes.FakeNetworkMeter.create_meter(count=2) + + self.network.delete_metering_label = mock.Mock(return_value=None) + + self.network.find_metering_label = network_fakes \ + .FakeNetworkMeter.get_meter( + meter=self.meter_list + ) + + self.cmd = network_meter.DeleteMeter(self.app, self.namespace) + + def test_delete_one_meter(self): + arglist = [ + self.meter_list[0].name, + ] + verifylist = [ + ('meter', [self.meter_list[0].name]), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + result = self.cmd.take_action(parsed_args) + + self.network.delete_metering_label.assert_called_once_with( + self.meter_list[0] + ) + self.assertIsNone(result) + + def test_delete_multiple_meters(self): + arglist = [] + for n in self.meter_list: + arglist.append(n.id) + verifylist = [ + ('meter', arglist), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + result = self.cmd.take_action(parsed_args) + + calls = [] + for n in self.meter_list: + calls.append(call(n)) + self.network.delete_metering_label.assert_has_calls(calls) + self.assertIsNone(result) + + def test_delete_multiple_meter_exception(self): + arglist = [ + self.meter_list[0].id, + 'xxxx-yyyy-zzzz', + self.meter_list[1].id, + ] + verifylist = [ + ('meter', arglist), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + return_find = [ + self.meter_list[0], + exceptions.NotFound('404'), + self.meter_list[1], + ] + self.network.find_meter = mock.Mock(side_effect=return_find) + + ret_delete = [ + None, + exceptions.NotFound('404'), + ] + self.network.delete_metering_label = mock.Mock(side_effect=ret_delete) + + self.assertRaises(exceptions.CommandError, self.cmd.take_action, + parsed_args) + + calls = [ + call(self.meter_list[0]), + call(self.meter_list[1]), + ] + self.network.delete_metering_label.assert_has_calls(calls) + + +class TestListMeter(TestMeter): + + meter_list = \ + network_fakes.FakeNetworkMeter.create_meter(count=2) + + columns = ( + 'ID', + 'Name', + 'Description', + 'Shared', + ) + + data = [] + + for meters in meter_list: + data.append(( + meters.id, + meters.name, + meters.description, + meters.shared, + )) + + def setUp(self): + super(TestListMeter, self).setUp() + + self.network.metering_labels = mock.Mock( + return_value=self.meter_list + ) + + self.cmd = network_meter.ListMeter(self.app, self.namespace) + + def test_meter_list(self): + arglist = [] + verifylist = [] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + columns, data = self.cmd.take_action(parsed_args) + + self.network.metering_labels.assert_called_with() + self.assertEqual(self.columns, columns) + self.assertEqual(self.data, list(data)) + + +class TestShowMeter(TestMeter): + new_meter = ( + network_fakes.FakeNetworkMeter. + create_one_meter() + ) + columns = ( + 'description', + 'id', + 'name', + 'project_id', + 'shared', + ) + + data = ( + new_meter.description, + new_meter.id, + new_meter.name, + new_meter.project_id, + new_meter.shared, + ) + + def setUp(self): + super(TestShowMeter, self).setUp() + + self.cmd = network_meter.ShowMeter(self.app, self.namespace) + + self.network.find_metering_label = \ + mock.Mock(return_value=self.new_meter) + + def test_show_no_options(self): + arglist = [] + verifylist = [] + + self.assertRaises(tests_utils.ParserException, self.check_parser, + self.cmd, arglist, verifylist) + + def test_meter_show_option(self): + arglist = [ + self.new_meter.name, + ] + verifylist = [ + ('meter', self.new_meter.name), + ] + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + columns, data = self.cmd.take_action(parsed_args) + + self.network.find_metering_label.assert_called_with( + self.new_meter.name, ignore_missing=False + ) + self.assertEqual(self.columns, columns) + self.assertEqual(self.data, data) -- cgit v1.2.1