From fcd46acb69250205e18274cc9a59ab5f46d075c3 Mon Sep 17 00:00:00 2001
From: zhangbailin <zhangbailin@inspur.com>
Date: Fri, 14 Jun 2019 16:40:40 +0800
Subject: 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
---
 openstackclient/compute/v2/server.py               | 28 +++++++++--
 .../tests/unit/compute/v2/test_server.py           | 54 ++++++++++++++++++++++
 2 files changed, 78 insertions(+), 4 deletions(-)

(limited to 'openstackclient')

diff --git a/openstackclient/compute/v2/server.py b/openstackclient/compute/v2/server.py
index 9ba91812..538c9c4f 100644
--- a/openstackclient/compute/v2/server.py
+++ b/openstackclient/compute/v2/server.py
@@ -2811,12 +2811,32 @@ class UnshelveServer(command.Command):
             nargs='+',
             help=_('Server(s) to unshelve (name or ID)'),
         )
+        parser.add_argument(
+            '--availability-zone',
+            default=None,
+            help=_('Name of the availability zone in which to unshelve a '
+                   'SHELVED_OFFLOADED server (supported by '
+                   '--os-compute-api-version 2.77 or above)'),
+        )
         return parser
 
     def take_action(self, parsed_args):
         compute_client = self.app.client_manager.compute
+        support_az = compute_client.api_version >= api_versions.APIVersion(
+            '2.77')
+        if not support_az and parsed_args.availability_zone:
+            msg = _("--os-compute-api-version 2.77 or greater is required "
+                    "to support the '--availability-zone' option.")
+            raise exceptions.CommandError(msg)
+
         for server in parsed_args.server:
-            utils.find_resource(
-                compute_client.servers,
-                server,
-            ).unshelve()
+            if support_az:
+                utils.find_resource(
+                    compute_client.servers,
+                    server
+                ).unshelve(availability_zone=parsed_args.availability_zone)
+            else:
+                utils.find_resource(
+                    compute_client.servers,
+                    server,
+                ).unshelve()
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 = {
-- 
cgit v1.2.1