Merge "Use SDK to get compute API extensions"

This commit is contained in:
Zuul 2019-11-04 22:09:23 +00:00 committed by Gerrit Code Review
commit 10747754a6
2 changed files with 10 additions and 15 deletions

View File

@ -89,9 +89,9 @@ class ListExtension(command.Lister):
LOG.warning(message) LOG.warning(message)
if parsed_args.compute or show_all: if parsed_args.compute or show_all:
compute_client = self.app.client_manager.compute compute_client = self.app.client_manager.sdk_connection.compute
try: try:
data += compute_client.list_extensions.show_all() data += compute_client.extensions()
except Exception: except Exception:
message = _("Extensions list not supported by Compute API") message = _("Extensions list not supported by Compute API")
LOG.warning(message) LOG.warning(message)

View File

@ -36,13 +36,9 @@ class TestExtension(utils.TestCommand):
self.identity_extensions_mock = identity_client.extensions self.identity_extensions_mock = identity_client.extensions
self.identity_extensions_mock.reset_mock() self.identity_extensions_mock.reset_mock()
compute_client = compute_fakes.FakeComputev2Client( sdk_connection = mock.Mock()
endpoint=fakes.AUTH_URL, self.app.client_manager.sdk_connection = sdk_connection
token=fakes.AUTH_TOKEN, self.compute_extensions_mock = sdk_connection.compute.extensions
)
self.app.client_manager.compute = compute_client
compute_client.list_extensions = mock.Mock()
self.compute_extensions_mock = compute_client.list_extensions
self.compute_extensions_mock.reset_mock() self.compute_extensions_mock.reset_mock()
volume_client = volume_fakes.FakeVolumeClient( volume_client = volume_fakes.FakeVolumeClient(
@ -80,8 +76,7 @@ class TestExtensionList(TestExtension):
self.identity_extensions_mock.list.return_value = [ self.identity_extensions_mock.list.return_value = [
self.identity_extension] self.identity_extension]
self.compute_extensions_mock.show_all.return_value = [ self.compute_extensions_mock.return_value = [self.compute_extension]
self.compute_extension]
self.volume_extensions_mock.show_all.return_value = [ self.volume_extensions_mock.show_all.return_value = [
self.volume_extension] self.volume_extension]
self.network_extensions_mock.return_value = [self.network_extension] self.network_extensions_mock.return_value = [self.network_extension]
@ -131,7 +126,7 @@ class TestExtensionList(TestExtension):
) )
self._test_extension_list_helper(arglist, verifylist, datalist) self._test_extension_list_helper(arglist, verifylist, datalist)
self.identity_extensions_mock.list.assert_called_with() self.identity_extensions_mock.list.assert_called_with()
self.compute_extensions_mock.show_all.assert_called_with() self.compute_extensions_mock.assert_called_with()
self.volume_extensions_mock.show_all.assert_called_with() self.volume_extensions_mock.show_all.assert_called_with()
self.network_extensions_mock.assert_called_with() self.network_extensions_mock.assert_called_with()
@ -178,7 +173,7 @@ class TestExtensionList(TestExtension):
) )
self._test_extension_list_helper(arglist, verifylist, datalist, True) self._test_extension_list_helper(arglist, verifylist, datalist, True)
self.identity_extensions_mock.list.assert_called_with() self.identity_extensions_mock.list.assert_called_with()
self.compute_extensions_mock.show_all.assert_called_with() self.compute_extensions_mock.assert_called_with()
self.volume_extensions_mock.show_all.assert_called_with() self.volume_extensions_mock.show_all.assert_called_with()
self.network_extensions_mock.assert_called_with() self.network_extensions_mock.assert_called_with()
@ -248,7 +243,7 @@ class TestExtensionList(TestExtension):
self.compute_extension.description, self.compute_extension.description,
), ) ), )
self._test_extension_list_helper(arglist, verifylist, datalist) self._test_extension_list_helper(arglist, verifylist, datalist)
self.compute_extensions_mock.show_all.assert_called_with() self.compute_extensions_mock.assert_called_with()
def test_extension_list_compute_and_network(self): def test_extension_list_compute_and_network(self):
arglist = [ arglist = [
@ -272,7 +267,7 @@ class TestExtensionList(TestExtension):
), ),
) )
self._test_extension_list_helper(arglist, verifylist, datalist) self._test_extension_list_helper(arglist, verifylist, datalist)
self.compute_extensions_mock.show_all.assert_called_with() self.compute_extensions_mock.assert_called_with()
self.network_extensions_mock.assert_called_with() self.network_extensions_mock.assert_called_with()
def test_extension_list_volume(self): def test_extension_list_volume(self):