Add UTs for requests/images

This patch adds unit tests for requests/images.

Change-Id: I214314ba8b9e78374e140803a25e5d64d78fd548
Issue-Id: MULTICLOUD-83
Signed-off-by: Shashank Kumar Shankar <shashank.kumar.shankar@intel.com>

1 files changed, 271 insertions, 0 deletions

diff --git a/newton/newton/requests/tests/test_image.py b/newton/newton/requests/tests/test_image.py
new file mode 100644
index 00000000..4b6bfb13
--- /dev/null
+++ b/newton/newton/requests/tests/test_image.py
@@ -0,0 +1,271 @@
+# Copyright (c) 2017 Intel Corporation, 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.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import mock
+import unittest
+
+from six.moves import urllib
+from django.test import Client
+from rest_framework import status
+
+from newton.requests.tests import mock_info
+from newton.requests.views.image import imageThread
+from newton.requests.views.util import VimDriverUtils
+
+
+MOCK_GET_IMAGES_RESPONSE = {
+    "images": [
+        {"id": "uuid_1", "name": "image_1"},
+        {"id": "uuid_2", "name": "image_2"}
+    ]
+}
+
+MOCK_GET_IMAGE_RESPONSE = {
+    "image": {
+        "id": "uuid_1",
+        "name": "image_1"
+    }
+}
+
+MOCK_POST_IMAGE_REQUEST = {
+    "id": "uuid_3",
+    "name": "image_3",
+    "imagePath": "test.com/image_3"
+}
+
+MOCK_POST_IMAGE_REQUEST_EXISTING = {
+    "id": "uuid_1",
+    "name": "image_1",
+    "imagePath": "test.com/image_1"
+}
+
+MOCK_POST_IMAGE_RESPONSE = {
+    "id": "uuid_3",
+    "name": "image_3"
+}
+
+
+class MockResponse(object):
+    status_code = 200
+    content = ''
+
+    def json(self):
+        pass
+
+
+class TestImage(unittest.TestCase):
+    def setUp(self):
+        self.client = Client()
+    def tearDown(self):
+        pass
+
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_get_images(self, mock_get_vim_info, mock_get_session):
+
+        mock_session_specs = ["get"]
+        mock_session = mock.Mock(name='mock_session', spec=mock_session_specs)
+
+        mock_get_images_response_obj = mock.Mock(spec=MockResponse)
+        mock_get_images_response_obj.status_code = 200
+        mock_get_images_response_obj.content = MOCK_GET_IMAGES_RESPONSE
+        mock_get_images_response_obj.json.return_value = MOCK_GET_IMAGES_RESPONSE
+        mock_session.get.return_value = mock_get_images_response_obj
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+
+        response = self.client.get(
+            "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/images",
+            {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+
+        context = response.json()
+        self.assertEquals(status.HTTP_200_OK, response.status_code)
+        self.assertIsNotNone(context['images'])
+        self.assertEqual(MOCK_GET_IMAGES_RESPONSE["images"], context["images"])
+
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_get_image(self, mock_get_vim_info, mock_get_session):
+
+        mock_session_specs = ["get"]
+        mock_session = mock.Mock(name='mock_session', spec=mock_session_specs)
+
+        mock_get_image_response_obj = mock.Mock(spec=MockResponse)
+        mock_get_image_response_obj.status_code = 200
+        mock_get_image_response_obj.content = MOCK_GET_IMAGE_RESPONSE
+        mock_get_image_response_obj.json.return_value = MOCK_GET_IMAGE_RESPONSE
+        mock_session.get.return_value = mock_get_image_response_obj
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+
+        response = self.client.get(
+            "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc"
+            "/images/uuid_1",
+            {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+        context = response.json()
+
+        self.assertEquals(status.HTTP_200_OK, response.status_code)
+        self.assertEqual(MOCK_GET_IMAGE_RESPONSE["image"], context["image"])
+
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_get_image_not_found(self, mock_get_vim_info, mock_get_session):
+
+        mock_session_specs = ["get"]
+        mock_session = mock.Mock(name='mock_session', spec=mock_session_specs)
+
+        mock_get_image_response_obj = mock.Mock(spec=MockResponse)
+        mock_get_image_response_obj.status_code = 404
+        mock_get_image_response_obj.content = {}
+        mock_get_image_response_obj.json.return_value = {}
+        mock_session.get.return_value = mock_get_image_response_obj
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+
+        response = self.client.get(
+            "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc"
+            "/images/uuid_3",
+            {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+        context = response.json()
+
+        self.assertEquals(status.HTTP_404_NOT_FOUND, response.status_code)
+        self.assertIsNone(context.get("image"))
+
+    @mock.patch.object(imageThread, 'run')
+    @mock.patch.object(urllib, 'request')
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_post_image(self, mock_get_vim_info, mock_get_session, mock_request, mock_run):
+        mock_session_specs = ["get", "post"]
+        mock_session = mock.Mock(name='mock_session', spec=mock_session_specs)
+
+        mock_get_images_response_obj = mock.Mock(spec=MockResponse)
+        mock_get_images_response_obj.status_code = 200
+        mock_get_images_response_obj.content = MOCK_GET_IMAGES_RESPONSE
+        mock_get_images_response_obj.json.return_value = MOCK_GET_IMAGES_RESPONSE
+
+        mock_post_image_response_obj = mock.Mock(spec=MockResponse)
+        mock_post_image_response_obj.status_code = 201
+        mock_post_image_response_obj.content = MOCK_POST_IMAGE_RESPONSE
+        mock_post_image_response_obj.json.return_value = MOCK_POST_IMAGE_RESPONSE
+
+        mock_session.get.return_value = mock_get_images_response_obj
+        mock_session.post.return_value = mock_post_image_response_obj
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+        mock_request.urlopen.return_value = "image"
+
+        response = self.client.post(
+            "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/images",
+            MOCK_POST_IMAGE_REQUEST, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+
+        context = response.json()
+        self.assertEquals(status.HTTP_201_CREATED, response.status_code)
+        self.assertIsNotNone(context['id'])
+        self.assertEqual(1, context['returnCode'])
+
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_post_image_existing(self, mock_get_vim_info, mock_get_session):
+        mock_session_specs = ["get", "post"]
+        mock_session = mock.Mock(name='mock_session', spec=mock_session_specs)
+
+        mock_get_images_response_obj = mock.Mock(spec=MockResponse)
+        mock_get_images_response_obj.status_code = 200
+        mock_get_images_response_obj.content = MOCK_GET_IMAGES_RESPONSE
+        mock_get_images_response_obj.json.return_value = MOCK_GET_IMAGES_RESPONSE
+
+        mock_post_image_response_obj = mock.Mock(spec=MockResponse)
+        mock_post_image_response_obj.status_code = 201
+        mock_post_image_response_obj.content = MOCK_POST_IMAGE_RESPONSE
+        mock_post_image_response_obj.json.return_value = MOCK_POST_IMAGE_RESPONSE
+
+        mock_session.get.return_value = mock_get_images_response_obj
+        mock_session.post.return_value = mock_post_image_response_obj
+
+        mock_session.get.return_value = mock_get_images_response_obj
+        mock_session.post.return_value = mock_post_image_response_obj
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+
+        response = self.client.post(
+            "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/images",
+            MOCK_POST_IMAGE_REQUEST_EXISTING, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+
+        context = response.json()
+        self.assertEquals(status.HTTP_200_OK, response.status_code)
+        self.assertIsNotNone(context['returnCode'])
+        self.assertEqual(0, context['returnCode'])
+
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_post_image_empty(self, mock_get_vim_info, mock_get_session):
+        mock_session_specs = ["get", "post"]
+        mock_session = mock.Mock(name='mock_session', spec=mock_session_specs)
+
+        mock_get_images_response_obj = mock.Mock(spec=MockResponse)
+        mock_get_images_response_obj.status_code = 200
+        mock_get_images_response_obj.content = MOCK_GET_IMAGES_RESPONSE
+        mock_get_images_response_obj.json.return_value = MOCK_GET_IMAGES_RESPONSE
+
+        mock_post_image_response_obj = mock.Mock(spec=MockResponse)
+        mock_post_image_response_obj.status_code = 201
+        mock_post_image_response_obj.content = MOCK_POST_IMAGE_RESPONSE
+        mock_post_image_response_obj.json.return_value = MOCK_POST_IMAGE_RESPONSE
+
+        mock_session.get.return_value = mock_get_images_response_obj
+        mock_session.post.return_value = mock_post_image_response_obj
+
+        mock_session.get.return_value = mock_get_images_response_obj
+        mock_session.post.return_value = mock_post_image_response_obj
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+
+        response = self.client.post(
+            "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/images",
+            {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+
+        context = response.json()
+        self.assertIn('error', context)
+        self.assertEquals(status.HTTP_500_INTERNAL_SERVER_ERROR, response.status_code)
+
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_delete_image(self, mock_get_vim_info, mock_get_session):
+        mock_session_specs = ["delete"]
+        mock_session = mock.Mock(name='mock_session', spec=mock_session_specs)
+
+        mock_delete_image_response_obj = mock.Mock(spec=MockResponse)
+        mock_delete_image_response_obj.status_code = 204
+
+        mock_session.delete.return_value = mock_delete_image_response_obj
+
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = mock_session
+
+        response = self.client.delete(
+            "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc"
+            "/images/uuid_1", HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+
+        self.assertEqual(status.HTTP_204_NO_CONTENT, response.status_code)
+        self.assertIsNone(response.data)