diff options
author | 2018-04-12 13:38:21 +0100 | |
---|---|---|
committer | 2018-06-27 17:30:09 +0100 | |
commit | 5a3d0c166d1af6b54cc487b778c0d8c560c2275d (patch) | |
tree | 1decec12b7b6dc08e8ba89627d3affb764107822 /yardstick/tests/unit/common/test_openstack_utils.py | |
parent | 3976fdc1cdfb6e0e749a6886a732040cb7dc6ece (diff) |
Replace glance delete image with shade client.
Remove get_image_id function.
Function delete_image now uses shade client.
JIRA: YARDSTICK-892
Change-Id: I6e8510dfa49aa14786ed7ac3382b85c4e699fb9e
Signed-off-by: Shobhi Jain <shobhi.jain@intel.com>
(cherry picked from commit d6c1b8b7bfd257efd5fe2df30dd450bdb2c57c7d)
Diffstat (limited to 'yardstick/tests/unit/common/test_openstack_utils.py')
-rw-r--r-- | yardstick/tests/unit/common/test_openstack_utils.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/yardstick/tests/unit/common/test_openstack_utils.py b/yardstick/tests/unit/common/test_openstack_utils.py index 9b23c65b8..c9f9129df 100644 --- a/yardstick/tests/unit/common/test_openstack_utils.py +++ b/yardstick/tests/unit/common/test_openstack_utils.py @@ -667,3 +667,30 @@ class CreateImageTestCase(unittest.TestCase): self.name) mock_logger.error.assert_called_once() self.assertIsNone(output) + + +class DeleteImageTestCase(unittest.TestCase): + + def test_delete_image(self): + self.mock_shade_client = mock.Mock() + self.mock_shade_client.delete_image.return_value = True + output = openstack_utils.delete_image(self.mock_shade_client, + 'image_name_or_id') + self.assertTrue(output) + + def test_delete_image_fail(self): + self.mock_shade_client = mock.Mock() + self.mock_shade_client.delete_image.return_value = False + output = openstack_utils.delete_image(self.mock_shade_client, + 'image_name_or_id') + self.assertFalse(output) + + @mock.patch.object(openstack_utils, 'log') + def test_delete_image_exception(self, mock_logger): + self.mock_shade_client = mock.Mock() + self.mock_shade_client.delete_image.side_effect = ( + exc.OpenStackCloudException('error message')) + output = openstack_utils.delete_image(self.mock_shade_client, + 'image_name_or_id') + mock_logger.error.assert_called_once() + self.assertFalse(output) |