summaryrefslogtreecommitdiff
path: root/openstackclient/tests/unit
diff options
context:
space:
mode:
Diffstat (limited to 'openstackclient/tests/unit')
-rw-r--r--openstackclient/tests/unit/compute/v2/test_server.py202
-rw-r--r--openstackclient/tests/unit/network/v2/fakes.py73
-rw-r--r--openstackclient/tests/unit/network/v2/test_address_group.py502
3 files changed, 742 insertions, 35 deletions
diff --git a/openstackclient/tests/unit/compute/v2/test_server.py b/openstackclient/tests/unit/compute/v2/test_server.py
index 0f33dd70..2c0cadfc 100644
--- a/openstackclient/tests/unit/compute/v2/test_server.py
+++ b/openstackclient/tests/unit/compute/v2/test_server.py
@@ -6434,16 +6434,126 @@ class TestServerShelve(TestServer):
# Get the command object to test
self.cmd = server.ShelveServer(self.app, None)
- # Set shelve method to be tested.
- self.methods = {
+ def test_shelve(self):
+ server_info = {'status': 'ACTIVE'}
+ server_methods = {
'shelve': None,
+ 'shelve_offload': None,
}
- def test_shelve_one_server(self):
- self.run_method_with_servers('shelve', 1)
+ server = compute_fakes.FakeServer.create_one_server(
+ attrs=server_info, methods=server_methods)
+ self.servers_mock.get.return_value = server
+
+ arglist = [server.name]
+ verifylist = [
+ ('servers', [server.name]),
+ ('wait', False),
+ ('offload', False),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ result = self.cmd.take_action(parsed_args)
+ self.assertIsNone(result)
- def test_shelve_multi_servers(self):
- self.run_method_with_servers('shelve', 3)
+ self.servers_mock.get.assert_called_once_with(server.name)
+ server.shelve.assert_called_once_with()
+ server.shelve_offload.assert_not_called()
+
+ def test_shelve_already_shelved(self):
+ server_info = {'status': 'SHELVED'}
+ server_methods = {
+ 'shelve': None,
+ 'shelve_offload': None,
+ }
+
+ server = compute_fakes.FakeServer.create_one_server(
+ attrs=server_info, methods=server_methods)
+ self.servers_mock.get.return_value = server
+
+ arglist = [server.name]
+ verifylist = [
+ ('servers', [server.name]),
+ ('wait', False),
+ ('offload', False),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ result = self.cmd.take_action(parsed_args)
+ self.assertIsNone(result)
+
+ self.servers_mock.get.assert_called_once_with(server.name)
+ server.shelve.assert_not_called()
+ server.shelve_offload.assert_not_called()
+
+ @mock.patch.object(common_utils, 'wait_for_status', return_value=True)
+ def test_shelve_with_wait(self, mock_wait_for_status):
+ server_info = {'status': 'ACTIVE'}
+ server_methods = {
+ 'shelve': None,
+ 'shelve_offload': None,
+ }
+
+ server = compute_fakes.FakeServer.create_one_server(
+ attrs=server_info, methods=server_methods)
+ self.servers_mock.get.return_value = server
+
+ arglist = ['--wait', server.name]
+ verifylist = [
+ ('servers', [server.name]),
+ ('wait', True),
+ ('offload', False),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ result = self.cmd.take_action(parsed_args)
+ self.assertIsNone(result)
+
+ self.servers_mock.get.assert_called_once_with(server.name)
+ server.shelve.assert_called_once_with()
+ server.shelve_offload.assert_not_called()
+ mock_wait_for_status.assert_called_once_with(
+ self.servers_mock.get,
+ server.id,
+ callback=mock.ANY,
+ success_status=('shelved', 'shelved_offloaded'),
+ )
+
+ @mock.patch.object(common_utils, 'wait_for_status', return_value=True)
+ def test_shelve_offload(self, mock_wait_for_status):
+ server_info = {'status': 'ACTIVE'}
+ server_methods = {
+ 'shelve': None,
+ 'shelve_offload': None,
+ }
+
+ server = compute_fakes.FakeServer.create_one_server(
+ attrs=server_info, methods=server_methods)
+ self.servers_mock.get.return_value = server
+
+ arglist = ['--offload', server.name]
+ verifylist = [
+ ('servers', [server.name]),
+ ('wait', False),
+ ('offload', True),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ result = self.cmd.take_action(parsed_args)
+ self.assertIsNone(result)
+
+ self.servers_mock.get.assert_has_calls([
+ mock.call(server.name),
+ mock.call(server.name),
+ ])
+ server.shelve.assert_called_once_with()
+ server.shelve_offload.assert_called_once_with()
+ mock_wait_for_status.assert_called_once_with(
+ self.servers_mock.get,
+ server.id,
+ callback=mock.ANY,
+ success_status=('shelved', 'shelved_offloaded'),
+ )
class TestServerShow(TestServer):
@@ -6859,6 +6969,9 @@ class TestServerUnshelve(TestServer):
self.methods = {
'unshelve': None,
}
+ self.attrs = {
+ 'status': 'SHELVED',
+ }
def test_unshelve_one_server(self):
self.run_method_with_servers('unshelve', 1)
@@ -6866,55 +6979,74 @@ class TestServerUnshelve(TestServer):
def test_unshelve_multi_servers(self):
self.run_method_with_servers('unshelve', 3)
- def test_unshelve_server_with_specified_az(self):
- server = compute_fakes.FakeServer.create_one_server()
+ def test_unshelve_with_specified_az(self):
+ self.app.client_manager.compute.api_version = \
+ api_versions.APIVersion('2.77')
+
+ server = compute_fakes.FakeServer.create_one_server(
+ attrs=self.attrs, methods=self.methods)
+ self.servers_mock.get.return_value = server
arglist = [
- server.id,
'--availability-zone', "foo-az",
+ server.id,
]
verifylist = [
('availability_zone', "foo-az"),
('server', [server.id])
]
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
- ex = self.assertRaises(exceptions.CommandError,
- self.cmd.take_action,
- parsed_args)
- self.assertIn(
- '--os-compute-api-version 2.77 or greater is required', str(ex))
-
-class TestServerUnshelveV277(TestServerUnshelve):
-
- def setUp(self):
- super(TestServerUnshelveV277, self).setUp()
-
- self.server = compute_fakes.FakeServer.create_one_server(
- methods=self.methods)
-
- # This is the return value for utils.find_resource()
- self.servers_mock.get.return_value = self.server
+ self.cmd.take_action(parsed_args)
- # Get the command object to test
- self.cmd = server.UnshelveServer(self.app, None)
+ self.servers_mock.get.assert_called_with(server.id)
+ server.unshelve.assert_called_with(availability_zone="foo-az")
- def test_specified_az_to_unshelve_with_v277(self):
- self.app.client_manager.compute.api_version = api_versions.APIVersion(
- '2.77')
+ def test_unshelve_with_specified_az_pre_v277(self):
+ self.app.client_manager.compute.api_version = \
+ api_versions.APIVersion('2.76')
+ server = compute_fakes.FakeServer.create_one_server(
+ attrs=self.attrs, methods=self.methods)
arglist = [
+ server.id,
'--availability-zone', "foo-az",
- self.server.id,
]
verifylist = [
('availability_zone', "foo-az"),
- ('server', [self.server.id])
+ ('server', [server.id])
]
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ ex = self.assertRaises(
+ exceptions.CommandError,
+ self.cmd.take_action,
+ parsed_args)
+ self.assertIn(
+ '--os-compute-api-version 2.77 or greater is required', str(ex))
- self.cmd.take_action(parsed_args)
- self.servers_mock.get.assert_called_with(self.server.id)
- self.server.unshelve.assert_called_with(availability_zone="foo-az")
+ @mock.patch.object(common_utils, 'wait_for_status', return_value=True)
+ def test_unshelve_with_wait(self, mock_wait_for_status):
+ server = compute_fakes.FakeServer.create_one_server(
+ attrs=self.attrs, methods=self.methods)
+ self.servers_mock.get.return_value = server
+
+ arglist = ['--wait', server.name]
+ verifylist = [
+ ('server', [server.name]),
+ ('wait', True),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ result = self.cmd.take_action(parsed_args)
+ self.assertIsNone(result)
+
+ self.servers_mock.get.assert_called_once_with(server.name)
+ server.unshelve.assert_called_once_with()
+ mock_wait_for_status.assert_called_once_with(
+ self.servers_mock.get,
+ server.id,
+ callback=mock.ANY,
+ success_status=('active', 'shutoff'),
+ )
class TestServerGeneral(TestServer):
diff --git a/openstackclient/tests/unit/network/v2/fakes.py b/openstackclient/tests/unit/network/v2/fakes.py
index 2db83d3b..798cfd96 100644
--- a/openstackclient/tests/unit/network/v2/fakes.py
+++ b/openstackclient/tests/unit/network/v2/fakes.py
@@ -83,6 +83,79 @@ class TestNetworkV2(utils.TestCommand):
)
+class FakeAddressGroup(object):
+ """Fake one or more address groups."""
+
+ @staticmethod
+ def create_one_address_group(attrs=None):
+ """Create a fake address group.
+
+ :param Dictionary attrs:
+ A dictionary with all attributes
+ :return:
+ A FakeResource object with name, id, etc.
+ """
+ attrs = attrs or {}
+
+ # Set default attributes.
+ address_group_attrs = {
+ 'name': 'address-group-name-' + uuid.uuid4().hex,
+ 'description': 'address-group-description-' + uuid.uuid4().hex,
+ 'id': 'address-group-id-' + uuid.uuid4().hex,
+ 'tenant_id': 'project-id-' + uuid.uuid4().hex,
+ 'addresses': ['10.0.0.1/32'],
+ }
+
+ # Overwrite default attributes.
+ address_group_attrs.update(attrs)
+
+ address_group = fakes.FakeResource(
+ info=copy.deepcopy(address_group_attrs),
+ loaded=True)
+
+ # Set attributes with special mapping in OpenStack SDK.
+ address_group.project_id = address_group_attrs['tenant_id']
+
+ return address_group
+
+ @staticmethod
+ def create_address_groups(attrs=None, count=2):
+ """Create multiple fake address groups.
+
+ :param Dictionary attrs:
+ A dictionary with all attributes
+ :param int count:
+ The number of address groups to fake
+ :return:
+ A list of FakeResource objects faking the address groups
+ """
+ address_groups = []
+ for i in range(0, count):
+ address_groups.append(
+ FakeAddressGroup.create_one_address_group(attrs))
+
+ return address_groups
+
+ @staticmethod
+ def get_address_groups(address_groups=None, count=2):
+ """Get an iterable Mock object with a list of faked address groups.
+
+ If address groups list is provided, then initialize the Mock object
+ with the list. Otherwise create one.
+
+ :param List address_groups:
+ A list of FakeResource objects faking address groups
+ :param int count:
+ The number of address groups to fake
+ :return:
+ An iterable Mock object with side_effect set to a list of faked
+ address groups
+ """
+ if address_groups is None:
+ address_groups = FakeAddressGroup.create_address_groups(count)
+ return mock.Mock(side_effect=address_groups)
+
+
class FakeAddressScope(object):
"""Fake one or more address scopes."""
diff --git a/openstackclient/tests/unit/network/v2/test_address_group.py b/openstackclient/tests/unit/network/v2/test_address_group.py
new file mode 100644
index 00000000..e4fa8ab3
--- /dev/null
+++ b/openstackclient/tests/unit/network/v2/test_address_group.py
@@ -0,0 +1,502 @@
+# 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.
+#
+
+from unittest import mock
+from unittest.mock import call
+
+from osc_lib import exceptions
+
+from openstackclient.network.v2 import address_group
+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 TestAddressGroup(network_fakes.TestNetworkV2):
+
+ def setUp(self):
+ super(TestAddressGroup, self).setUp()
+
+ # Get a shortcut to the network client
+ self.network = self.app.client_manager.network
+ # Get a shortcut to the ProjectManager Mock
+ self.projects_mock = self.app.client_manager.identity.projects
+ # Get a shortcut to the DomainManager Mock
+ self.domains_mock = self.app.client_manager.identity.domains
+
+
+class TestCreateAddressGroup(TestAddressGroup):
+
+ project = identity_fakes_v3.FakeProject.create_one_project()
+ domain = identity_fakes_v3.FakeDomain.create_one_domain()
+ # The new address group created.
+ new_address_group = (
+ network_fakes.FakeAddressGroup.create_one_address_group(
+ attrs={
+ 'tenant_id': project.id,
+ }
+ ))
+ columns = (
+ 'addresses',
+ 'description',
+ 'id',
+ 'name',
+ 'project_id',
+ )
+ data = (
+ new_address_group.addresses,
+ new_address_group.description,
+ new_address_group.id,
+ new_address_group.name,
+ new_address_group.project_id,
+ )
+
+ def setUp(self):
+ super(TestCreateAddressGroup, self).setUp()
+ self.network.create_address_group = mock.Mock(
+ return_value=self.new_address_group)
+
+ # Get the command object to test
+ self.cmd = address_group.CreateAddressGroup(self.app, self.namespace)
+
+ self.projects_mock.get.return_value = self.project
+ self.domains_mock.get.return_value = self.domain
+
+ def test_create_no_options(self):
+ arglist = []
+ verifylist = []
+
+ # Missing required args should bail here
+ self.assertRaises(tests_utils.ParserException, self.check_parser,
+ self.cmd, arglist, verifylist)
+
+ def test_create_default_options(self):
+ arglist = [
+ self.new_address_group.name,
+ ]
+ verifylist = [
+ ('project', None),
+ ('name', self.new_address_group.name),
+ ('description', None),
+ ('address', []),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ columns, data = (self.cmd.take_action(parsed_args))
+
+ self.network.create_address_group.assert_called_once_with(**{
+ 'name': self.new_address_group.name,
+ 'addresses': [],
+ })
+ self.assertEqual(self.columns, columns)
+ self.assertItemsEqual(self.data, data)
+
+ def test_create_all_options(self):
+ arglist = [
+ '--project', self.project.name,
+ '--project-domain', self.domain.name,
+ '--address', '10.0.0.1',
+ '--description', self.new_address_group.description,
+ self.new_address_group.name,
+ ]
+ verifylist = [
+ ('project', self.project.name),
+ ('project_domain', self.domain.name),
+ ('address', ['10.0.0.1']),
+ ('description', self.new_address_group.description),
+ ('name', self.new_address_group.name),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ columns, data = (self.cmd.take_action(parsed_args))
+
+ self.network.create_address_group.assert_called_once_with(**{
+ 'addresses': ['10.0.0.1/32'],
+ 'tenant_id': self.project.id,
+ 'name': self.new_address_group.name,
+ 'description': self.new_address_group.description,
+ })
+ self.assertEqual(self.columns, columns)
+ self.assertItemsEqual(self.data, data)
+
+
+class TestDeleteAddressGroup(TestAddressGroup):
+
+ # The address group to delete.
+ _address_groups = (
+ network_fakes.FakeAddressGroup.create_address_groups(count=2))
+
+ def setUp(self):
+ super(TestDeleteAddressGroup, self).setUp()
+ self.network.delete_address_group = mock.Mock(return_value=None)
+ self.network.find_address_group = (
+ network_fakes.FakeAddressGroup.get_address_groups(
+ address_groups=self._address_groups)
+ )
+
+ # Get the command object to test
+ self.cmd = address_group.DeleteAddressGroup(self.app, self.namespace)
+
+ def test_address_group_delete(self):
+ arglist = [
+ self._address_groups[0].name,
+ ]
+ verifylist = [
+ ('address_group', [self._address_groups[0].name]),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ result = self.cmd.take_action(parsed_args)
+ self.network.find_address_group.assert_called_once_with(
+ self._address_groups[0].name, ignore_missing=False)
+ self.network.delete_address_group.assert_called_once_with(
+ self._address_groups[0])
+ self.assertIsNone(result)
+
+ def test_multi_address_groups_delete(self):
+ arglist = []
+
+ for a in self._address_groups:
+ arglist.append(a.name)
+ verifylist = [
+ ('address_group', arglist),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ result = self.cmd.take_action(parsed_args)
+
+ calls = []
+ for a in self._address_groups:
+ calls.append(call(a))
+ self.network.delete_address_group.assert_has_calls(calls)
+ self.assertIsNone(result)
+
+ def test_multi_address_groups_delete_with_exception(self):
+ arglist = [
+ self._address_groups[0].name,
+ 'unexist_address_group',
+ ]
+ verifylist = [
+ ('address_group',
+ [self._address_groups[0].name, 'unexist_address_group']),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+ find_mock_result = [self._address_groups[0], exceptions.CommandError]
+ self.network.find_address_group = (
+ mock.Mock(side_effect=find_mock_result)
+ )
+
+ try:
+ self.cmd.take_action(parsed_args)
+ self.fail('CommandError should be raised.')
+ except exceptions.CommandError as e:
+ self.assertEqual('1 of 2 address groups failed to delete.', str(e))
+
+ self.network.find_address_group.assert_any_call(
+ self._address_groups[0].name, ignore_missing=False)
+ self.network.find_address_group.assert_any_call(
+ 'unexist_address_group', ignore_missing=False)
+ self.network.delete_address_group.assert_called_once_with(
+ self._address_groups[0]
+ )
+
+
+class TestListAddressGroup(TestAddressGroup):
+
+ # The address groups to list up.
+ address_groups = (
+ network_fakes.FakeAddressGroup.create_address_groups(count=3))
+ columns = (
+ 'ID',
+ 'Name',
+ 'Description',
+ 'Project',
+ 'Addresses',
+ )
+ data = []
+ for group in address_groups:
+ data.append((
+ group.id,
+ group.name,
+ group.description,
+ group.project_id,
+ group.addresses,
+ ))
+
+ def setUp(self):
+ super(TestListAddressGroup, self).setUp()
+ self.network.address_groups = mock.Mock(
+ return_value=self.address_groups)
+
+ # Get the command object to test
+ self.cmd = address_group.ListAddressGroup(self.app, self.namespace)
+
+ def test_address_group_list(self):
+ arglist = []
+ verifylist = []
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ columns, data = self.cmd.take_action(parsed_args)
+
+ self.network.address_groups.assert_called_once_with(**{})
+ self.assertEqual(self.columns, columns)
+ self.assertItemsEqual(self.data, list(data))
+
+ def test_address_group_list_name(self):
+ arglist = [
+ '--name', self.address_groups[0].name,
+ ]
+ verifylist = [
+ ('name', self.address_groups[0].name),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ columns, data = self.cmd.take_action(parsed_args)
+
+ self.network.address_groups.assert_called_once_with(
+ **{'name': self.address_groups[0].name})
+ self.assertEqual(self.columns, columns)
+ self.assertItemsEqual(self.data, list(data))
+
+ def test_address_group_list_project(self):
+ project = identity_fakes_v3.FakeProject.create_one_project()
+ self.projects_mock.get.return_value = project
+ arglist = [
+ '--project', project.id,
+ ]
+ verifylist = [
+ ('project', project.id),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ columns, data = self.cmd.take_action(parsed_args)
+
+ self.network.address_groups.assert_called_once_with(
+ project_id=project.id)
+ self.assertEqual(self.columns, columns)
+ self.assertItemsEqual(self.data, list(data))
+
+ def test_address_group_project_domain(self):
+ project = identity_fakes_v3.FakeProject.create_one_project()
+ self.projects_mock.get.return_value = project
+ arglist = [
+ '--project', project.id,
+ '--project-domain', project.domain_id,
+ ]
+ verifylist = [
+ ('project', project.id),
+ ]
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ columns, data = self.cmd.take_action(parsed_args)
+
+ self.network.address_groups.assert_called_once_with(
+ project_id=project.id)
+ self.assertEqual(self.columns, columns)
+ self.assertItemsEqual(self.data, list(data))
+
+
+class TestSetAddressGroup(TestAddressGroup):
+
+ # The address group to set.
+ _address_group = network_fakes.FakeAddressGroup.create_one_address_group()
+
+ def setUp(self):
+ super(TestSetAddressGroup, self).setUp()
+ self.network.update_address_group = mock.Mock(return_value=None)
+ self.network.find_address_group = mock.Mock(
+ return_value=self._address_group)
+ self.network.add_addresses_to_address_group = mock.Mock(
+ return_value=self._address_group)
+ # Get the command object to test
+ self.cmd = address_group.SetAddressGroup(self.app, self.namespace)
+
+ def test_set_nothing(self):
+ arglist = [self._address_group.name, ]
+ verifylist = [
+ ('address_group', self._address_group.name),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ result = self.cmd.take_action(parsed_args)
+
+ self.network.update_address_group.assert_not_called()
+ self.network.add_addresses_to_address_group.assert_not_called()
+ self.assertIsNone(result)
+
+ def test_set_name_and_description(self):
+ arglist = [
+ '--name', 'new_address_group_name',
+ '--description', 'new_address_group_description',
+ self._address_group.name,
+ ]
+ verifylist = [
+ ('name', 'new_address_group_name'),
+ ('description', 'new_address_group_description'),
+ ('address_group', self._address_group.name),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ result = self.cmd.take_action(parsed_args)
+ attrs = {
+ 'name': "new_address_group_name",
+ 'description': 'new_address_group_description',
+ }
+ self.network.update_address_group.assert_called_with(
+ self._address_group, **attrs)
+ self.assertIsNone(result)
+
+ def test_set_one_address(self):
+ arglist = [
+ self._address_group.name,
+ '--address', '10.0.0.2',
+ ]
+ verifylist = [
+ ('address_group', self._address_group.name),
+ ('address', ['10.0.0.2']),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ result = self.cmd.take_action(parsed_args)
+ self.network.add_addresses_to_address_group.assert_called_once_with(
+ self._address_group, ['10.0.0.2/32'])
+ self.assertIsNone(result)
+
+ def test_set_multiple_addresses(self):
+ arglist = [
+ self._address_group.name,
+ '--address', '10.0.0.2',
+ '--address', '2001::/16',
+ ]
+ verifylist = [
+ ('address_group', self._address_group.name),
+ ('address', ['10.0.0.2', '2001::/16']),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ result = self.cmd.take_action(parsed_args)
+ self.network.add_addresses_to_address_group.assert_called_once_with(
+ self._address_group, ['10.0.0.2/32', '2001::/16'])
+ self.assertIsNone(result)
+
+
+class TestShowAddressGroup(TestAddressGroup):
+
+ # The address group to show.
+ _address_group = network_fakes.FakeAddressGroup.create_one_address_group()
+ columns = (
+ 'addresses',
+ 'description',
+ 'id',
+ 'name',
+ 'project_id',
+ )
+ data = (
+ _address_group.addresses,
+ _address_group.description,
+ _address_group.id,
+ _address_group.name,
+ _address_group.project_id,
+ )
+
+ def setUp(self):
+ super(TestShowAddressGroup, self).setUp()
+ self.network.find_address_group = mock.Mock(
+ return_value=self._address_group)
+
+ # Get the command object to test
+ self.cmd = address_group.ShowAddressGroup(self.app, self.namespace)
+
+ def test_show_no_options(self):
+ arglist = []
+ verifylist = []
+
+ # Missing required args should bail here
+ self.assertRaises(tests_utils.ParserException, self.check_parser,
+ self.cmd, arglist, verifylist)
+
+ def test_show_all_options(self):
+ arglist = [
+ self._address_group.name,
+ ]
+ verifylist = [
+ ('address_group', self._address_group.name),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ columns, data = self.cmd.take_action(parsed_args)
+
+ self.network.find_address_group.assert_called_once_with(
+ self._address_group.name, ignore_missing=False)
+ self.assertEqual(self.columns, columns)
+ self.assertItemsEqual(self.data, list(data))
+
+
+class TestUnsetAddressGroup(TestAddressGroup):
+
+ # The address group to unset.
+ _address_group = network_fakes.FakeAddressGroup.create_one_address_group()
+
+ def setUp(self):
+ super(TestUnsetAddressGroup, self).setUp()
+ self.network.find_address_group = mock.Mock(
+ return_value=self._address_group)
+ self.network.remove_addresses_from_address_group = mock.Mock(
+ return_value=self._address_group)
+ # Get the command object to test
+ self.cmd = address_group.UnsetAddressGroup(self.app, self.namespace)
+
+ def test_unset_nothing(self):
+ arglist = [self._address_group.name, ]
+ verifylist = [
+ ('address_group', self._address_group.name),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ result = self.cmd.take_action(parsed_args)
+
+ self.network.remove_addresses_from_address_group.assert_not_called()
+ self.assertIsNone(result)
+
+ def test_unset_one_address(self):
+ arglist = [
+ self._address_group.name,
+ '--address', '10.0.0.2',
+ ]
+ verifylist = [
+ ('address_group', self._address_group.name),
+ ('address', ['10.0.0.2']),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ result = self.cmd.take_action(parsed_args)
+ self.network.remove_addresses_from_address_group.\
+ assert_called_once_with(self._address_group, ['10.0.0.2/32'])
+ self.assertIsNone(result)
+
+ def test_unset_multiple_addresses(self):
+ arglist = [
+ self._address_group.name,
+ '--address', '10.0.0.2',
+ '--address', '2001::/16',
+ ]
+ verifylist = [
+ ('address_group', self._address_group.name),
+ ('address', ['10.0.0.2', '2001::/16']),
+ ]
+
+ parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+ result = self.cmd.take_action(parsed_args)
+ self.network.remove_addresses_from_address_group.\
+ assert_called_once_with(self._address_group,
+ ['10.0.0.2/32', '2001::/16'])
+ self.assertIsNone(result)