From a1a470693e264e991b1dc9497512769bb8d510d8 Mon Sep 17 00:00:00 2001 From: Tang Chen <chen.tang@easystack.cn> Date: Wed, 23 Mar 2016 11:27:37 +0800 Subject: [PATCH] Add --project to "subnet pool create" This patch adds --project and --project-domain options to "subnet pool create" command. Change-Id: I2fe006013a194861299a9c77234a7cf988a8dad8 Partial-Bug: #1544586 --- doc/source/command-objects/subnet-pool.rst | 10 ++++ openstackclient/network/v2/subnet_pool.py | 25 +++++++-- .../tests/network/v2/test_subnet_pool.py | 52 +++++++++++++++++++ .../notes/bug-1544586-0e6ca9a09dac0726.yaml | 3 ++ 4 files changed, 86 insertions(+), 4 deletions(-) diff --git a/doc/source/command-objects/subnet-pool.rst b/doc/source/command-objects/subnet-pool.rst index 439c4f5033..cb86e2273c 100644 --- a/doc/source/command-objects/subnet-pool.rst +++ b/doc/source/command-objects/subnet-pool.rst @@ -17,6 +17,7 @@ Create subnet pool [--default-prefix-length <default-prefix-length>] [--min-prefix-length <min-prefix-length>] [--max-prefix-length <max-prefix-length>] + [--project <project> [--project-domain <project-domain>]] <name> .. option:: --pool-prefix <pool-prefix> @@ -36,6 +37,15 @@ Create subnet pool Set subnet pool maximum prefix length +.. option:: --project <project> + + Owner's project (name or ID) + +.. option:: --project-domain <project-domain> + + Domain the project belongs to (name or ID). This can be used in case + collisions between project names exist. + .. _subnet_pool_create-name: .. describe:: <name> diff --git a/openstackclient/network/v2/subnet_pool.py b/openstackclient/network/v2/subnet_pool.py index 834760fb43..8c90b95511 100644 --- a/openstackclient/network/v2/subnet_pool.py +++ b/openstackclient/network/v2/subnet_pool.py @@ -17,6 +17,7 @@ from openstackclient.common import command from openstackclient.common import exceptions from openstackclient.common import parseractions from openstackclient.common import utils +from openstackclient.identity import common as identity_common def _get_columns(item): @@ -32,7 +33,7 @@ _formatters = { } -def _get_attrs(parsed_args): +def _get_attrs(client_manager, parsed_args): attrs = {} if parsed_args.name is not None: attrs['name'] = str(parsed_args.name) @@ -45,6 +46,16 @@ def _get_attrs(parsed_args): if parsed_args.max_prefix_length is not None: attrs['max_prefixlen'] = parsed_args.max_prefix_length + # "subnet pool set" command doesn't support setting project. + if 'project' in parsed_args and parsed_args.project is not None: + 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 + return attrs @@ -84,16 +95,22 @@ class CreateSubnetPool(command.ShowOne): parser = super(CreateSubnetPool, self).get_parser(prog_name) parser.add_argument( 'name', - metavar="<name>", + metavar='<name>', help='Name of the new subnet pool' ) _add_prefix_options(parser) + parser.add_argument( + '--project', + metavar='<project>', + help="Owner's project (name or ID)", + ) + identity_common.add_project_domain_option_to_parser(parser) return parser def take_action(self, parsed_args): client = self.app.client_manager.network - attrs = _get_attrs(parsed_args) + attrs = _get_attrs(self.app.client_manager, parsed_args) obj = client.create_subnet_pool(**attrs) columns = _get_columns(obj) data = utils.get_item_properties(obj, columns, formatters=_formatters) @@ -192,7 +209,7 @@ class SetSubnetPool(command.Command): obj = client.find_subnet_pool(parsed_args.subnet_pool, ignore_missing=False) - attrs = _get_attrs(parsed_args) + attrs = _get_attrs(self.app.client_manager, parsed_args) if attrs == {}: msg = "Nothing specified to be set" raise exceptions.CommandError(msg) diff --git a/openstackclient/tests/network/v2/test_subnet_pool.py b/openstackclient/tests/network/v2/test_subnet_pool.py index 6ed2352d67..c79b91799c 100644 --- a/openstackclient/tests/network/v2/test_subnet_pool.py +++ b/openstackclient/tests/network/v2/test_subnet_pool.py @@ -12,11 +12,14 @@ # import argparse +import copy import mock from openstackclient.common import exceptions from openstackclient.common import utils from openstackclient.network.v2 import subnet_pool +from openstackclient.tests import fakes +from openstackclient.tests.identity.v3 import fakes as identity_fakes_v3 from openstackclient.tests.network.v2 import fakes as network_fakes from openstackclient.tests import utils as tests_utils @@ -73,6 +76,30 @@ class TestCreateSubnetPool(TestSubnetPool): # Get the command object to test self.cmd = subnet_pool.CreateSubnetPool(self.app, self.namespace) + # Set identity client. And get a shortcut to Identity client. + identity_client = identity_fakes_v3.FakeIdentityv3Client( + endpoint=fakes.AUTH_URL, + token=fakes.AUTH_TOKEN, + ) + self.app.client_manager.identity = identity_client + self.identity = self.app.client_manager.identity + + # Get a shortcut to the ProjectManager Mock + self.projects_mock = self.identity.projects + self.projects_mock.get.return_value = fakes.FakeResource( + None, + copy.deepcopy(identity_fakes_v3.PROJECT), + loaded=True, + ) + + # Get a shortcut to the DomainManager Mock + self.domains_mock = self.identity.domains + self.domains_mock.get.return_value = fakes.FakeResource( + None, + copy.deepcopy(identity_fakes_v3.DOMAIN), + loaded=True, + ) + def test_create_no_options(self): arglist = [] verifylist = [] @@ -140,6 +167,31 @@ class TestCreateSubnetPool(TestSubnetPool): self.assertRaises(argparse.ArgumentTypeError, self.check_parser, self.cmd, arglist, verifylist) + def test_create_project_domain(self): + arglist = [ + '--pool-prefix', '10.0.10.0/24', + "--project", identity_fakes_v3.project_name, + "--project-domain", identity_fakes_v3.domain_name, + self._subnet_pool.name, + ] + verifylist = [ + ('prefixes', ['10.0.10.0/24']), + ('project', identity_fakes_v3.project_name), + ('project_domain', identity_fakes_v3.domain_name), + ('name', self._subnet_pool.name), + ] + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + columns, data = (self.cmd.take_action(parsed_args)) + + self.network.create_subnet_pool.assert_called_once_with(**{ + 'prefixes': ['10.0.10.0/24'], + 'tenant_id': identity_fakes_v3.project_id, + 'name': self._subnet_pool.name, + }) + self.assertEqual(self.columns, columns) + self.assertEqual(self.data, data) + class TestDeleteSubnetPool(TestSubnetPool): diff --git a/releasenotes/notes/bug-1544586-0e6ca9a09dac0726.yaml b/releasenotes/notes/bug-1544586-0e6ca9a09dac0726.yaml index e1595ed374..fdbd6fd18b 100644 --- a/releasenotes/notes/bug-1544586-0e6ca9a09dac0726.yaml +++ b/releasenotes/notes/bug-1544586-0e6ca9a09dac0726.yaml @@ -2,3 +2,6 @@ features: - Add ``subnet pool create`` command. [Bug `1544586 <https://bugs.launchpad.net/python-openstackclient/+bug/1544586>`_] + - Command ``subnet pool create`` now supports ``--project`` and + ``--project-domain`` options. + [Bug `1544586 <https://bugs.launchpad.net/python-openstackclient/+bug/1544586>`_]