diff options
Diffstat (limited to 'openstackclient/tests/common')
| -rw-r--r-- | openstackclient/tests/common/test_extension.py | 52 |
1 files changed, 50 insertions, 2 deletions
diff --git a/openstackclient/tests/common/test_extension.py b/openstackclient/tests/common/test_extension.py index 2e6e7050..5561345b 100644 --- a/openstackclient/tests/common/test_extension.py +++ b/openstackclient/tests/common/test_extension.py @@ -18,6 +18,7 @@ from openstackclient.tests import fakes from openstackclient.tests import utils from openstackclient.tests.identity.v2_0 import fakes as identity_fakes +from openstackclient.tests.network.v2 import fakes as network_fakes class TestExtension(utils.TestCommand): @@ -29,12 +30,15 @@ class TestExtension(utils.TestCommand): endpoint=fakes.AUTH_URL, token=fakes.AUTH_TOKEN, ) - - # Get shortcuts to the ExtensionManager Mocks self.identity_extensions_mock = ( self.app.client_manager.identity.extensions) self.identity_extensions_mock.reset_mock() + network = network_fakes.FakeNetworkV2Client() + self.app.client_manager.network = network + self.network_extensions_mock = network.list_extensions + self.network_extensions_mock.reset_mock() + class TestExtensionList(TestExtension): @@ -48,6 +52,13 @@ class TestExtensionList(TestExtension): loaded=True, ), ] + self.network_extensions_mock.list.return_value = [ + fakes.FakeResource( + None, + copy.deepcopy(identity_fakes.EXTENSION), + loaded=True, + ), + ] # Get the command object to test self.cmd = extension.ListExtension(self.app, None) @@ -71,6 +82,11 @@ class TestExtensionList(TestExtension): identity_fakes.extension_alias, identity_fakes.extension_description, ), + ( + network_fakes.extension_name, + network_fakes.extension_alias, + network_fakes.extension_description, + ), ) self.assertEqual(tuple(data), datalist) @@ -101,6 +117,14 @@ class TestExtensionList(TestExtension): identity_fakes.extension_updated, identity_fakes.extension_links, ), + ( + network_fakes.extension_name, + network_fakes.extension_namespace, + network_fakes.extension_description, + network_fakes.extension_alias, + network_fakes.extension_updated, + network_fakes.extension_links, + ), ) self.assertEqual(tuple(data), datalist) @@ -126,3 +150,27 @@ class TestExtensionList(TestExtension): identity_fakes.extension_description, ), ) self.assertEqual(tuple(data), datalist) + + def test_extension_list_network(self): + arglist = [ + '--network', + ] + verifylist = [ + ('network', True), + ] + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + columns, data = self.cmd.take_action(parsed_args) + + self.network_extensions_mock.assert_called_with() + + collist = ('Name', 'Alias', 'Description') + self.assertEqual(columns, collist) + datalist = ( + ( + network_fakes.extension_name, + network_fakes.extension_alias, + network_fakes.extension_description, + ), + ) + self.assertEqual(tuple(data), datalist) |