Add two consistency group commands

Add commands:
    consistency group add volume
    consistency group remove volume
in volume v2 (v2 only)

Change-Id: I70ff287d3b5df78196b8f4b9e50402c471aef284
Implements: bp cinder-command-support
Closes-Bug: #1613964

1 files changed, 206 insertions, 0 deletions

diff --git a/openstackclient/tests/unit/volume/v2/test_consistency_group.py b/openstackclient/tests/unit/volume/v2/test_consistency_group.py
index bc99ca8d..6eeeae39 100644
--- a/openstackclient/tests/unit/volume/v2/test_consistency_group.py
+++ b/openstackclient/tests/unit/volume/v2/test_consistency_group.py
@@ -36,10 +36,115 @@ class TestConsistencyGroup(volume_fakes.TestVolume):
             self.app.client_manager.volume.cgsnapshots)
         self.cgsnapshots_mock.reset_mock()
 
+        self.volumes_mock = (
+            self.app.client_manager.volume.volumes)
+        self.volumes_mock.reset_mock()
+
         self.types_mock = self.app.client_manager.volume.volume_types
         self.types_mock.reset_mock()
 
 
+class TestConsistencyGroupAddVolume(TestConsistencyGroup):
+
+    _consistency_group = (
+        volume_fakes.FakeConsistencyGroup.create_one_consistency_group())
+
+    def setUp(self):
+        super(TestConsistencyGroupAddVolume, self).setUp()
+
+        self.consistencygroups_mock.get.return_value = (
+            self._consistency_group)
+        # Get the command object to test
+        self.cmd = \
+            consistency_group.AddVolumeToConsistencyGroup(self.app, None)
+
+    def test_add_one_volume_to_consistency_group(self):
+        volume = volume_fakes.FakeVolume.create_one_volume()
+        self.volumes_mock.get.return_value = volume
+        arglist = [
+            self._consistency_group.id,
+            volume.id,
+        ]
+        verifylist = [
+            ('consistency_group', self._consistency_group.id),
+            ('volumes', [volume.id]),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'add_volumes': volume.id,
+        }
+        self.consistencygroups_mock.update.assert_called_once_with(
+            self._consistency_group.id,
+            **kwargs
+        )
+        self.assertIsNone(result)
+
+    def test_add_multiple_volumes_to_consistency_group(self):
+        volumes = volume_fakes.FakeVolume.create_volumes(count=2)
+        self.volumes_mock.get = volume_fakes.FakeVolume.get_volumes(volumes)
+        arglist = [
+            self._consistency_group.id,
+            volumes[0].id,
+            volumes[1].id,
+        ]
+        verifylist = [
+            ('consistency_group', self._consistency_group.id),
+            ('volumes', [volumes[0].id, volumes[1].id]),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'add_volumes': volumes[0].id + ',' + volumes[1].id,
+        }
+        self.consistencygroups_mock.update.assert_called_once_with(
+            self._consistency_group.id,
+            **kwargs
+        )
+        self.assertIsNone(result)
+
+    @mock.patch.object(consistency_group.LOG, 'error')
+    def test_add_multiple_volumes_to_consistency_group_with_exception(
+            self, mock_error):
+        volume = volume_fakes.FakeVolume.create_one_volume()
+        arglist = [
+            self._consistency_group.id,
+            volume.id,
+            'unexist_volume',
+        ]
+        verifylist = [
+            ('consistency_group', self._consistency_group.id),
+            ('volumes', [volume.id, 'unexist_volume']),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        find_mock_result = [volume,
+                            exceptions.CommandError,
+                            self._consistency_group]
+        with mock.patch.object(utils, 'find_resource',
+                               side_effect=find_mock_result) as find_mock:
+            result = self.cmd.take_action(parsed_args)
+            mock_error.assert_called_with("1 of 2 volumes failed to add.")
+            self.assertIsNone(result)
+            find_mock.assert_any_call(self.consistencygroups_mock,
+                                      self._consistency_group.id)
+            find_mock.assert_any_call(self.volumes_mock,
+                                      volume.id)
+            find_mock.assert_any_call(self.volumes_mock,
+                                      'unexist_volume')
+            self.assertEqual(3, find_mock.call_count)
+            self.consistencygroups_mock.update.assert_called_once_with(
+                self._consistency_group.id, add_volumes=volume.id
+            )
+
+
 class TestConsistencyGroupCreate(TestConsistencyGroup):
 
     volume_type = volume_fakes.FakeType.create_one_type()
@@ -394,6 +499,107 @@ class TestConsistencyGroupList(TestConsistencyGroup):
         self.assertEqual(self.data_long, list(data))
 
 
+class TestConsistencyGroupRemoveVolume(TestConsistencyGroup):
+
+    _consistency_group = (
+        volume_fakes.FakeConsistencyGroup.create_one_consistency_group())
+
+    def setUp(self):
+        super(TestConsistencyGroupRemoveVolume, self).setUp()
+
+        self.consistencygroups_mock.get.return_value = (
+            self._consistency_group)
+        # Get the command object to test
+        self.cmd = \
+            consistency_group.RemoveVolumeFromConsistencyGroup(self.app, None)
+
+    def test_remove_one_volume_from_consistency_group(self):
+        volume = volume_fakes.FakeVolume.create_one_volume()
+        self.volumes_mock.get.return_value = volume
+        arglist = [
+            self._consistency_group.id,
+            volume.id,
+        ]
+        verifylist = [
+            ('consistency_group', self._consistency_group.id),
+            ('volumes', [volume.id]),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'remove_volumes': volume.id,
+        }
+        self.consistencygroups_mock.update.assert_called_once_with(
+            self._consistency_group.id,
+            **kwargs
+        )
+        self.assertIsNone(result)
+
+    def test_remove_multi_volumes_from_consistency_group(self):
+        volumes = volume_fakes.FakeVolume.create_volumes(count=2)
+        self.volumes_mock.get = volume_fakes.FakeVolume.get_volumes(volumes)
+        arglist = [
+            self._consistency_group.id,
+            volumes[0].id,
+            volumes[1].id,
+        ]
+        verifylist = [
+            ('consistency_group', self._consistency_group.id),
+            ('volumes', [volumes[0].id, volumes[1].id]),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'remove_volumes': volumes[0].id + ',' + volumes[1].id,
+        }
+        self.consistencygroups_mock.update.assert_called_once_with(
+            self._consistency_group.id,
+            **kwargs
+        )
+        self.assertIsNone(result)
+
+    @mock.patch.object(consistency_group.LOG, 'error')
+    def test_remove_multiple_volumes_from_consistency_group_with_exception(
+            self, mock_error):
+        volume = volume_fakes.FakeVolume.create_one_volume()
+        arglist = [
+            self._consistency_group.id,
+            volume.id,
+            'unexist_volume',
+        ]
+        verifylist = [
+            ('consistency_group', self._consistency_group.id),
+            ('volumes', [volume.id, 'unexist_volume']),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        find_mock_result = [volume,
+                            exceptions.CommandError,
+                            self._consistency_group]
+        with mock.patch.object(utils, 'find_resource',
+                               side_effect=find_mock_result) as find_mock:
+            result = self.cmd.take_action(parsed_args)
+            mock_error.assert_called_with("1 of 2 volumes failed to remove.")
+            self.assertIsNone(result)
+            find_mock.assert_any_call(self.consistencygroups_mock,
+                                      self._consistency_group.id)
+            find_mock.assert_any_call(self.volumes_mock,
+                                      volume.id)
+            find_mock.assert_any_call(self.volumes_mock,
+                                      'unexist_volume')
+            self.assertEqual(3, find_mock.call_count)
+            self.consistencygroups_mock.update.assert_called_once_with(
+                self._consistency_group.id, remove_volumes=volume.id
+            )
+
+
 class TestConsistencyGroupSet(TestConsistencyGroup):
 
     consistency_group = (