diff options
| -rw-r--r-- | vio/vio/tests/test_volume_view.py | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/vio/vio/tests/test_volume_view.py b/vio/vio/tests/test_volume_view.py new file mode 100644 index 0000000..9d8395e --- /dev/null +++ b/vio/vio/tests/test_volume_view.py @@ -0,0 +1,203 @@ +# Copyright (c) 2018 VMware, Inc. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at: +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + +import mock +import unittest + +from vio.pub.msapi import extsys +from vio.swagger.views.volume import views +from vio.pub.vim.vimapi.cinder import OperateVolume +from vio.pub.vim.vimapi.glance import OperateImage + + +class TestGetDeleteVolumeView(unittest.TestCase): + + def setUp(self): + self.view = views.GetDeleteVolumeView() + + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_get(self, mock_getvim, mock_getvol): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + vol = mock.Mock() + vol.attachments = [] + vol.id = "vol-id" + vol.name = "vol-name" + vol.created_at = "create time" + vol.status = "ok" + vol.volume_type = "vmdk" + vol.size = 1 + vol.availability_zone = "nova" + mock_getvol.return_value = vol + ret = self.view.get(mock.Mock(), "vmware_nova", "tenant1", "vol-1") + self.assertEqual(200, ret.status_code) + + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_get_fail(self, mock_getvim, mock_getvol): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + mock_getvol.side_effect = [Exception("error here")] + ret = self.view.get(mock.Mock(), "vmware_nova", "tenant1", "vol-1") + self.assertEqual(500, ret.status_code) + + @mock.patch.object(OperateVolume.OperateVolume, "delete_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_delete(self, mock_getvim, mock_delvol): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + mock_delvol.return_value = None + ret = self.view.delete(mock.Mock(), "vmware_nova", "tenant1", "vol-1") + self.assertEqual(204, ret.status_code) + + @mock.patch.object(OperateVolume.OperateVolume, "delete_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_delete_fail(self, mock_getvim, mock_delvol): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + mock_delvol.side_effect = [Exception("error")] + ret = self.view.delete(mock.Mock(), "vmware_nova", "tenant1", "vol-1") + self.assertEqual(500, ret.status_code) + + +class TestCreateListVolumeView(unittest.TestCase): + + def setUp(self): + self.view = views.CreateListVolumeView() + + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volumes") + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_get(self, mock_getvim, mock_getvol, mock_getvols): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + vol = mock.Mock() + vol.attachments = [] + vol.id = "vol-id" + vol.name = "vol-name" + vol.created_at = "create time" + vol.status = "ok" + vol.volume_type = "vmdk" + vol.size = 1 + vol.availability_zone = "nova" + mock_getvol.return_value = vol + mock_getvols.return_value = [vol] + ret = self.view.get( + mock.Mock(query_params=[]), "vmware_nova", "tenant1") + self.assertEqual(200, ret.status_code) + + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volumes") + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_get_fail(self, mock_getvim, mock_getvol, mock_getvols): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + vol = mock.Mock() + vol.attachments = [] + vol.id = "vol-id" + vol.name = "vol-name" + vol.created_at = "create time" + vol.status = "ok" + vol.volume_type = "vmdk" + vol.size = 1 + vol.availability_zone = "nova" + mock_getvol.side_effect = [Exception("error")] + mock_getvols.return_value = [vol] + ret = self.view.get( + mock.Mock(query_params=[]), "vmware_nova", "tenant1") + self.assertEqual(500, ret.status_code) + + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volumes") + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_post_exist(self, mock_getvim, mock_getvol, mock_getvols): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + vol = mock.Mock() + vol.attachments = [] + vol.id = "vol-id" + vol.name = "vol-name" + vol.created_at = "create time" + vol.status = "ok" + vol.volume_type = "vmdk" + vol.size = 1 + vol.availability_zone = "nova" + mock_getvol.return_value = vol + mock_getvols.return_value = [vol] + req = mock.Mock() + req.body = """{ + "name": "vol-name" + }""" + ret = self.view.post(req, "vmware_nova", "tenant1") + self.assertEqual(200, ret.status_code) + + @mock.patch.object(OperateImage.OperateImage, "find_vim_image") + @mock.patch.object(OperateVolume.OperateVolume, "create_vim_volume") + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volumes") + @mock.patch.object(extsys, "get_vim_by_id") + def test_post_from_image(self, mock_getvim, mock_getvols, + mock_createvol, mock_findimg): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + mock_findimg.return_value = mock.Mock(id="image-id") + vol = mock.Mock() + vol.attachments = [] + vol.id = "vol-id" + vol.name = "vol-name" + vol.created_at = "create time" + vol.status = "ok" + vol.volume_type = "vmdk" + vol.size = 1 + vol.availability_zone = "nova" + mock_createvol.return_value = vol + mock_getvols.return_value = [] + req = mock.Mock() + req.body = """{ + "name": "vol-name", + "imageId": "image-id" + }""" + ret = self.view.post(req, "vmware_nova", "tenant1") + self.assertEqual(202, ret.status_code) + + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volumes") + @mock.patch.object(OperateVolume.OperateVolume, "get_vim_volume") + @mock.patch.object(extsys, "get_vim_by_id") + def test_post_fail(self, mock_getvim, mock_getvol, mock_getvols): + mock_getvim.return_value = { + "tenant": "tenant-id" + } + vol = mock.Mock() + vol.attachments = [] + vol.id = "vol-id" + vol.name = "vol-name" + vol.created_at = "create time" + vol.status = "ok" + vol.volume_type = "vmdk" + vol.size = 1 + vol.availability_zone = "nova" + mock_getvol.side_effect = [Exception("error")] + mock_getvols.return_value = [vol] + req = mock.Mock() + req.body = """{ + "name": "vol-name" + }""" + ret = self.view.post(req, "vmware_nova", "tenant1") + self.assertEqual(500, ret.status_code) |