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)