python-openstackclient 6.3.0__py3-none-any.whl → 6.5.0__py3-none-any.whl

openstackclient/tests/unit/compute/v2/test_aggregate.py

@@ -55,29 +55,15 @@ class TestAggregate(compute_fakes.TestComputev2):
         fake_ag.uuid,
     )
 
-    def setUp(self):
-        super(TestAggregate, self).setUp()
-
-        # Get a shortcut to the AggregateManager Mock
-        self.app.client_manager.sdk_connection = mock.Mock()
-        self.app.client_manager.sdk_connection.compute = mock.Mock()
-        self.sdk_client = self.app.client_manager.sdk_connection.compute
-        self.sdk_client.aggregates = mock.Mock()
-        self.sdk_client.find_aggregate = mock.Mock()
-        self.sdk_client.create_aggregate = mock.Mock()
-        self.sdk_client.update_aggregate = mock.Mock()
-        self.sdk_client.update_aggregate = mock.Mock()
-        self.sdk_client.set_aggregate_metadata = mock.Mock()
-        self.sdk_client.add_host_to_aggregate = mock.Mock()
-        self.sdk_client.remove_host_from_aggregate = mock.Mock()
-
 
 class TestAggregateAddHost(TestAggregate):
     def setUp(self):
         super(TestAggregateAddHost, self).setUp()
 
-        self.sdk_client.find_aggregate.return_value = self.fake_ag
-        self.sdk_client.add_host_to_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.find_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.add_host_to_aggregate.return_value = (
+            self.fake_ag
+        )
         self.cmd = aggregate.AddAggregateHost(self.app, None)
 
     def test_aggregate_add_host(self):
@@ -91,10 +77,10 @@ class TestAggregateAddHost(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.sdk_client.add_host_to_aggregate.assert_called_once_with(
+        self.compute_sdk_client.add_host_to_aggregate.assert_called_once_with(
             self.fake_ag.id, parsed_args.host
         )
         self.assertEqual(self.columns, columns)
@@ -105,8 +91,10 @@ class TestAggregateCreate(TestAggregate):
     def setUp(self):
         super(TestAggregateCreate, self).setUp()
 
-        self.sdk_client.create_aggregate.return_value = self.fake_ag
-        self.sdk_client.set_aggregate_metadata.return_value = self.fake_ag
+        self.compute_sdk_client.create_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.set_aggregate_metadata.return_value = (
+            self.fake_ag
+        )
         self.cmd = aggregate.CreateAggregate(self.app, None)
 
     def test_aggregate_create(self):
@@ -118,7 +106,7 @@ class TestAggregateCreate(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_aggregate.assert_called_once_with(
+        self.compute_sdk_client.create_aggregate.assert_called_once_with(
             name=parsed_args.name
         )
         self.assertEqual(self.columns, columns)
@@ -137,7 +125,7 @@ class TestAggregateCreate(TestAggregate):
 
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_aggregate.assert_called_once_with(
+        self.compute_sdk_client.create_aggregate.assert_called_once_with(
             name=parsed_args.name, availability_zone=parsed_args.zone
         )
         self.assertEqual(self.columns, columns)
@@ -157,10 +145,10 @@ class TestAggregateCreate(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_aggregate.assert_called_once_with(
+        self.compute_sdk_client.create_aggregate.assert_called_once_with(
             name=parsed_args.name
         )
-        self.sdk_client.set_aggregate_metadata.assert_called_once_with(
+        self.compute_sdk_client.set_aggregate_metadata.assert_called_once_with(
             self.fake_ag.id, parsed_args.properties
         )
         self.assertEqual(self.columns, columns)
@@ -173,7 +161,7 @@ class TestAggregateDelete(TestAggregate):
     def setUp(self):
         super(TestAggregateDelete, self).setUp()
 
-        self.sdk_client.find_aggregate = compute_fakes.get_aggregates(
+        self.compute_sdk_client.find_aggregate = compute_fakes.get_aggregates(
             self.fake_ags
         )
         self.cmd = aggregate.DeleteAggregate(self.app, None)
@@ -185,10 +173,10 @@ class TestAggregateDelete(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             self.fake_ags[0].id, ignore_missing=False
         )
-        self.sdk_client.delete_aggregate.assert_called_once_with(
+        self.compute_sdk_client.delete_aggregate.assert_called_once_with(
             self.fake_ags[0].id, ignore_missing=False
         )
 
@@ -206,8 +194,8 @@ class TestAggregateDelete(TestAggregate):
         calls = []
         for a in self.fake_ags:
             calls.append(call(a.id, ignore_missing=False))
-        self.sdk_client.find_aggregate.assert_has_calls(calls)
-        self.sdk_client.delete_aggregate.assert_has_calls(calls)
+        self.compute_sdk_client.find_aggregate.assert_has_calls(calls)
+        self.compute_sdk_client.delete_aggregate.assert_has_calls(calls)
 
     def test_delete_multiple_agggregates_with_exception(self):
         arglist = [
@@ -220,7 +208,7 @@ class TestAggregateDelete(TestAggregate):
 
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 
-        self.sdk_client.find_aggregate.side_effect = [
+        self.compute_sdk_client.find_aggregate.side_effect = [
             self.fake_ags[0],
             sdk_exceptions.NotFoundException,
         ]
@@ -233,8 +221,8 @@ class TestAggregateDelete(TestAggregate):
         calls = []
         for a in arglist:
             calls.append(call(a, ignore_missing=False))
-        self.sdk_client.find_aggregate.assert_has_calls(calls)
-        self.sdk_client.delete_aggregate.assert_called_with(
+        self.compute_sdk_client.find_aggregate.assert_has_calls(calls)
+        self.compute_sdk_client.delete_aggregate.assert_called_with(
             self.fake_ags[0].id, ignore_missing=False
         )
 
@@ -281,7 +269,7 @@ class TestAggregateList(TestAggregate):
     def setUp(self):
         super(TestAggregateList, self).setUp()
 
-        self.sdk_client.aggregates.return_value = [self.fake_ag]
+        self.compute_sdk_client.aggregates.return_value = [self.fake_ag]
         self.cmd = aggregate.ListAggregate(self.app, None)
 
     def test_aggregate_list(self):
@@ -309,8 +297,10 @@ class TestAggregateRemoveHost(TestAggregate):
     def setUp(self):
         super(TestAggregateRemoveHost, self).setUp()
 
-        self.sdk_client.find_aggregate.return_value = self.fake_ag
-        self.sdk_client.remove_host_from_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.find_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.remove_host_from_aggregate.return_value = (
+            self.fake_ag
+        )
         self.cmd = aggregate.RemoveAggregateHost(self.app, None)
 
     def test_aggregate_remove_host(self):
@@ -324,10 +314,10 @@ class TestAggregateRemoveHost(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.sdk_client.remove_host_from_aggregate.assert_called_once_with(
+        self.compute_sdk_client.remove_host_from_aggregate.assert_called_once_with(
             self.fake_ag.id, parsed_args.host
         )
         self.assertEqual(self.columns, columns)
@@ -338,7 +328,7 @@ class TestAggregateSet(TestAggregate):
     def setUp(self):
         super(TestAggregateSet, self).setUp()
 
-        self.sdk_client.find_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.find_aggregate.return_value = self.fake_ag
         self.cmd = aggregate.SetAggregate(self.app, None)
 
     def test_aggregate_set_no_option(self):
@@ -351,11 +341,11 @@ class TestAggregateSet(TestAggregate):
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
 
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.assertNotCalled(self.sdk_client.update_aggregate)
-        self.assertNotCalled(self.sdk_client.set_aggregate_metadata)
+        self.assertNotCalled(self.compute_sdk_client.update_aggregate)
+        self.assertNotCalled(self.compute_sdk_client.set_aggregate_metadata)
         self.assertIsNone(result)
 
     def test_aggregate_set_with_name(self):
@@ -371,13 +361,13 @@ class TestAggregateSet(TestAggregate):
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
 
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.sdk_client.update_aggregate.assert_called_once_with(
+        self.compute_sdk_client.update_aggregate.assert_called_once_with(
             self.fake_ag.id, name=parsed_args.name
         )
-        self.assertNotCalled(self.sdk_client.set_aggregate_metadata)
+        self.assertNotCalled(self.compute_sdk_client.set_aggregate_metadata)
         self.assertIsNone(result)
 
     def test_aggregate_set_with_zone(self):
@@ -393,13 +383,13 @@ class TestAggregateSet(TestAggregate):
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
 
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.sdk_client.update_aggregate.assert_called_once_with(
+        self.compute_sdk_client.update_aggregate.assert_called_once_with(
             self.fake_ag.id, availability_zone=parsed_args.zone
         )
-        self.assertNotCalled(self.sdk_client.set_aggregate_metadata)
+        self.assertNotCalled(self.compute_sdk_client.set_aggregate_metadata)
         self.assertIsNone(result)
 
     def test_aggregate_set_with_property(self):
@@ -417,11 +407,11 @@ class TestAggregateSet(TestAggregate):
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
 
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.assertNotCalled(self.sdk_client.update_aggregate)
-        self.sdk_client.set_aggregate_metadata.assert_called_once_with(
+        self.assertNotCalled(self.compute_sdk_client.update_aggregate)
+        self.compute_sdk_client.set_aggregate_metadata.assert_called_once_with(
             self.fake_ag.id, parsed_args.properties
         )
         self.assertIsNone(result)
@@ -440,11 +430,11 @@ class TestAggregateSet(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.assertNotCalled(self.sdk_client.update_aggregate)
-        self.sdk_client.set_aggregate_metadata.assert_called_once_with(
+        self.assertNotCalled(self.compute_sdk_client.update_aggregate)
+        self.compute_sdk_client.set_aggregate_metadata.assert_called_once_with(
             self.fake_ag.id, {'key1': None, 'key2': 'value2'}
         )
         self.assertIsNone(result)
@@ -460,11 +450,11 @@ class TestAggregateSet(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.assertNotCalled(self.sdk_client.update_aggregate)
-        self.sdk_client.set_aggregate_metadata.assert_called_once_with(
+        self.assertNotCalled(self.compute_sdk_client.update_aggregate)
+        self.compute_sdk_client.set_aggregate_metadata.assert_called_once_with(
             self.fake_ag.id, {'key1': None}
         )
         self.assertIsNone(result)
@@ -483,13 +473,13 @@ class TestAggregateSet(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.sdk_client.update_aggregate.assert_called_once_with(
+        self.compute_sdk_client.update_aggregate.assert_called_once_with(
             self.fake_ag.id, availability_zone=parsed_args.zone
         )
-        self.sdk_client.set_aggregate_metadata.assert_called_once_with(
+        self.compute_sdk_client.set_aggregate_metadata.assert_called_once_with(
             self.fake_ag.id, {'key1': None}
         )
         self.assertIsNone(result)
@@ -525,7 +515,7 @@ class TestAggregateShow(TestAggregate):
     def setUp(self):
         super(TestAggregateShow, self).setUp()
 
-        self.sdk_client.find_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.find_aggregate.return_value = self.fake_ag
         self.cmd = aggregate.ShowAggregate(self.app, None)
 
     def test_aggregate_show(self):
@@ -537,7 +527,7 @@ class TestAggregateShow(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
 
@@ -549,7 +539,7 @@ class TestAggregateUnset(TestAggregate):
     def setUp(self):
         super(TestAggregateUnset, self).setUp()
 
-        self.sdk_client.find_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.find_aggregate.return_value = self.fake_ag
         self.cmd = aggregate.UnsetAggregate(self.app, None)
 
     def test_aggregate_unset(self):
@@ -565,7 +555,7 @@ class TestAggregateUnset(TestAggregate):
 
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
-        self.sdk_client.set_aggregate_metadata.assert_called_once_with(
+        self.compute_sdk_client.set_aggregate_metadata.assert_called_once_with(
             self.fake_ag.id, {'unset_key': None}
         )
         self.assertIsNone(result)
@@ -585,7 +575,7 @@ class TestAggregateUnset(TestAggregate):
 
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
-        self.sdk_client.set_aggregate_metadata.assert_called_once_with(
+        self.compute_sdk_client.set_aggregate_metadata.assert_called_once_with(
             self.fake_ag.id, {'unset_key1': None, 'unset_key2': None}
         )
         self.assertIsNone(result)
@@ -600,7 +590,7 @@ class TestAggregateUnset(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         result = self.cmd.take_action(parsed_args)
-        self.assertNotCalled(self.sdk_client.set_aggregate_metadata)
+        self.assertNotCalled(self.compute_sdk_client.set_aggregate_metadata)
         self.assertIsNone(result)
 
 
@@ -610,7 +600,7 @@ class TestAggregateCacheImage(TestAggregate):
     def setUp(self):
         super(TestAggregateCacheImage, self).setUp()
 
-        self.sdk_client.find_aggregate.return_value = self.fake_ag
+        self.compute_sdk_client.find_aggregate.return_value = self.fake_ag
         self.find_image_mock = mock.Mock(side_effect=self.images)
         self.app.client_manager.sdk_connection.image.find_image = (
             self.find_image_mock
@@ -639,10 +629,10 @@ class TestAggregateCacheImage(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.sdk_client.aggregate_precache_images.assert_called_once_with(
+        self.compute_sdk_client.aggregate_precache_images.assert_called_once_with(
             self.fake_ag.id, [self.images[0].id]
         )
 
@@ -659,9 +649,9 @@ class TestAggregateCacheImage(TestAggregate):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         self.cmd.take_action(parsed_args)
-        self.sdk_client.find_aggregate.assert_called_once_with(
+        self.compute_sdk_client.find_aggregate.assert_called_once_with(
             parsed_args.aggregate, ignore_missing=False
         )
-        self.sdk_client.aggregate_precache_images.assert_called_once_with(
+        self.compute_sdk_client.aggregate_precache_images.assert_called_once_with(
             self.fake_ag.id, [self.images[0].id, self.images[1].id]
         )

openstackclient/tests/unit/compute/v2/test_console.py

@@ -20,25 +20,13 @@ from openstackclient.tests.unit.compute.v2 import fakes as compute_fakes
 from openstackclient.tests.unit import utils
 
 
-class TestConsole(compute_fakes.TestComputev2):
-    def setUp(self):
-        super(TestConsole, self).setUp()
-
-        # SDK mock
-        self.app.client_manager.sdk_connection = mock.Mock()
-        self.app.client_manager.sdk_connection.compute = mock.Mock()
-        self.sdk_client = self.app.client_manager.sdk_connection.compute
-        self.sdk_client.find_server = mock.Mock()
-        self.sdk_client.get_server_console_output = mock.Mock()
-
-
-class TestConsoleLog(TestConsole):
+class TestConsoleLog(compute_fakes.TestComputev2):
     _server = compute_fakes.create_one_server()
 
     def setUp(self):
         super(TestConsoleLog, self).setUp()
 
-        self.sdk_client.find_server.return_value = self._server
+        self.compute_sdk_client.find_server.return_value = self._server
 
         self.cmd = console.ShowConsoleLog(self.app, None)
 
@@ -59,13 +47,13 @@ class TestConsoleLog(TestConsole):
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 
         output = {'output': '1st line\n2nd line\n'}
-        self.sdk_client.get_server_console_output.return_value = output
+        self.compute_sdk_client.get_server_console_output.return_value = output
         self.cmd.take_action(parsed_args)
 
-        self.sdk_client.find_server.assert_called_with(
+        self.compute_sdk_client.find_server.assert_called_with(
             name_or_id='fake_server', ignore_missing=False
         )
-        self.sdk_client.get_server_console_output.assert_called_with(
+        self.compute_sdk_client.get_server_console_output.assert_called_with(
             self._server.id, length=None
         )
         stdout = self.app.stdout.content
@@ -77,29 +65,29 @@ class TestConsoleLog(TestConsole):
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 
         output = {'output': '1st line\n2nd line'}
-        self.sdk_client.get_server_console_output.return_value = output
+        self.compute_sdk_client.get_server_console_output.return_value = output
         self.cmd.take_action(parsed_args)
 
-        self.sdk_client.find_server.assert_called_with(
+        self.compute_sdk_client.find_server.assert_called_with(
             name_or_id='fake_server', ignore_missing=False
         )
-        self.sdk_client.get_server_console_output.assert_called_with(
+        self.compute_sdk_client.get_server_console_output.assert_called_with(
             self._server.id, length=15
         )
 
 
-class TestConsoleUrlShow(TestConsole):
+class TestConsoleUrlShow(compute_fakes.TestComputev2):
     _server = compute_fakes.create_one_server()
 
     def setUp(self):
         super(TestConsoleUrlShow, self).setUp()
-        self.sdk_client.find_server.return_value = self._server
+        self.compute_sdk_client.find_server.return_value = self._server
         fake_console_data = {
             'url': 'http://localhost',
             'protocol': 'fake_protocol',
             'type': 'fake_type',
         }
-        self.sdk_client.create_console = mock.Mock(
+        self.compute_sdk_client.create_console = mock.Mock(
             return_value=fake_console_data
         )
 
@@ -126,7 +114,7 @@ class TestConsoleUrlShow(TestConsole):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_console.assert_called_once_with(
+        self.compute_sdk_client.create_console.assert_called_once_with(
             self._server.id, console_type='novnc'
         )
         self.assertEqual(self.columns, columns)
@@ -143,7 +131,7 @@ class TestConsoleUrlShow(TestConsole):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_console.assert_called_once_with(
+        self.compute_sdk_client.create_console.assert_called_once_with(
             self._server.id, console_type='novnc'
         )
         self.assertEqual(self.columns, columns)
@@ -160,7 +148,7 @@ class TestConsoleUrlShow(TestConsole):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_console.assert_called_once_with(
+        self.compute_sdk_client.create_console.assert_called_once_with(
             self._server.id, console_type='xvpvnc'
         )
         self.assertEqual(self.columns, columns)
@@ -177,7 +165,7 @@ class TestConsoleUrlShow(TestConsole):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_console.assert_called_once_with(
+        self.compute_sdk_client.create_console.assert_called_once_with(
             self._server.id, console_type='spice-html5'
         )
         self.assertEqual(self.columns, columns)
@@ -194,7 +182,7 @@ class TestConsoleUrlShow(TestConsole):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_console.assert_called_once_with(
+        self.compute_sdk_client.create_console.assert_called_once_with(
             self._server.id, console_type='rdp-html5'
         )
         self.assertEqual(self.columns, columns)
@@ -211,7 +199,7 @@ class TestConsoleUrlShow(TestConsole):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_console.assert_called_once_with(
+        self.compute_sdk_client.create_console.assert_called_once_with(
             self._server.id, console_type='serial'
         )
         self.assertEqual(self.columns, columns)
@@ -228,7 +216,7 @@ class TestConsoleUrlShow(TestConsole):
         ]
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
         columns, data = self.cmd.take_action(parsed_args)
-        self.sdk_client.create_console.assert_called_once_with(
+        self.compute_sdk_client.create_console.assert_called_once_with(
             self._server.id, console_type='webmks'
         )
         self.assertEqual(self.columns, columns)