Microversion 2.77: Support Specifying AZ to unshelve

This patch adds a new parameter ``--availability-zone`` to
``openstack server unshelve`` command. This can help users to specify
an ``availability_zone`` to unshelve a shelve offloaded server from
2.77 microversion.

Depends-On: https://review.opendev.org/679295

Implements: blueprint support-specifying-az-when-restore-shelved-server

Change-Id: Ia431e27c2a17fe16466707cc362532860ecf22df

1 files changed, 54 insertions, 0 deletions

diff --git a/openstackclient/tests/unit/compute/v2/test_server.py b/openstackclient/tests/unit/compute/v2/test_server.py
index 0793116a..5c98188a 100644
--- a/openstackclient/tests/unit/compute/v2/test_server.py
+++ b/openstackclient/tests/unit/compute/v2/test_server.py
@@ -109,6 +109,10 @@ class TestServer(compute_fakes.TestComputev2):
                 version = self.app.client_manager.compute.api_version
                 if version >= api_versions.APIVersion('2.73'):
                     method.assert_called_with(reason=None)
+            elif method_name == 'unshelve':
+                version = self.app.client_manager.compute.api_version
+                if version >= api_versions.APIVersion('2.77'):
+                    method.assert_called_with(availability_zone=None)
                 else:
                     method.assert_called_with()
             else:
@@ -4777,6 +4781,56 @@ 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()
+        arglist = [
+            server.id,
+            '--availability-zone', "foo-az",
+        ]
+        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
+
+        # Get the command object to test
+        self.cmd = server.UnshelveServer(self.app, None)
+
+    def test_specified_az_to_unshelve_with_v277(self):
+        self.app.client_manager.compute.api_version = api_versions.APIVersion(
+            '2.77')
+
+        arglist = [
+            '--availability-zone', "foo-az",
+            self.server.id,
+        ]
+        verifylist = [
+            ('availability_zone', "foo-az"),
+            ('server', [self.server.id])
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        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")
+
 
 class TestServerGeneral(TestServer):
     OLD = {