Switch command server remove volume to sdk

Change-Id: If6f6cf93b55a67e767c54de8ce21f25252cf99ca

1 files changed, 42 insertions, 3 deletions

diff --git a/openstackclient/tests/unit/compute/v2/test_server.py b/openstackclient/tests/unit/compute/v2/test_server.py
index 203e47eb..10ea07ad 100644
--- a/openstackclient/tests/unit/compute/v2/test_server.py
+++ b/openstackclient/tests/unit/compute/v2/test_server.py
@@ -692,9 +692,6 @@ class TestServerVolume(TestServer):
             'create_volume_attachment': None,
         }
 
-        # Get the command object to test
-        self.cmd = server.AddServerVolume(self.app, None)
-
         self.servers = self.setup_sdk_servers_mock(count=1)
         self.volumes = self.setup_sdk_volumes_mock(count=1)
 
@@ -709,6 +706,15 @@ class TestServerVolume(TestServer):
         self.sdk_client.create_volume_attachment.return_value = \
             self.volume_attachment
 
+
+class TestServerAddVolume(TestServerVolume):
+
+    def setUp(self):
+        super(TestServerAddVolume, self).setUp()
+
+        # Get the command object to test
+        self.cmd = server.AddServerVolume(self.app, None)
+
     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
     def test_server_add_volume(self, sm_mock):
 
@@ -985,6 +991,39 @@ class TestServerVolume(TestServer):
                       'with argument --enable-delete-on-termination', str(ex))
 
 
+class TestServerRemoveVolume(TestServerVolume):
+
+    def setUp(self):
+        super(TestServerRemoveVolume, self).setUp()
+
+        # Get the command object to test
+        self.cmd = server.RemoveServerVolume(self.app, None)
+
+    def test_server_remove_volume(self):
+        self.sdk_client.volume_attachments.return_value = [
+            self.volume_attachment
+        ]
+
+        arglist = [
+            self.servers[0].id,
+            self.volumes[0].id,
+        ]
+
+        verifylist = [
+            ('server', self.servers[0].id),
+            ('volume', self.volumes[0].id),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        self.assertIsNone(result)
+        self.sdk_client.delete_volume_attachment.assert_called_once_with(
+            self.volume_attachment,
+            self.servers[0])
+
+
 class TestServerAddNetwork(TestServer):
 
     def setUp(self):