diff options
| -rw-r--r-- | newton/newton/requests/tests/test_image.py | 137 | ||||
| -rw-r--r-- | newton/newton/requests/tests/test_network.py | 127 | ||||
| -rw-r--r-- | newton/newton/requests/tests/test_subnet.py | 188 | ||||
| -rw-r--r-- | newton/newton/requests/tests/test_volume.py | 188 |
4 files changed, 457 insertions, 183 deletions
diff --git a/newton/newton/requests/tests/test_image.py b/newton/newton/requests/tests/test_image.py index 4b6bfb13..467e4710 100644 --- a/newton/newton/requests/tests/test_image.py +++ b/newton/newton/requests/tests/test_image.py @@ -20,6 +20,7 @@ from django.test import Client from rest_framework import status from newton.requests.tests import mock_info +from newton.requests.tests import test_base from newton.requests.views.image import imageThread from newton.requests.views.util import VimDriverUtils @@ -56,14 +57,6 @@ MOCK_POST_IMAGE_RESPONSE = { } -class MockResponse(object): - status_code = 200 - content = '' - - def json(self): - pass - - class TestImage(unittest.TestCase): def setUp(self): self.client = Client() @@ -74,17 +67,12 @@ class TestImage(unittest.TestCase): @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_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_IMAGES_RESPONSE}}), + ] 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", @@ -99,17 +87,12 @@ class TestImage(unittest.TestCase): @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_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_IMAGE_RESPONSE}}), + ] 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" @@ -124,20 +107,13 @@ class TestImage(unittest.TestCase): @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_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": {}, + "status_code": 404}}) + ] 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" @@ -153,24 +129,17 @@ class TestImage(unittest.TestCase): @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_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_IMAGES_RESPONSE}}), + test_base.get_mock_session( + ["post"], {"post": {"content": MOCK_POST_IMAGE_RESPONSE, + "status_code": 201}}), + ] 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( @@ -185,27 +154,16 @@ class TestImage(unittest.TestCase): @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_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_IMAGES_RESPONSE}}), + test_base.get_mock_session( + ["post"], {"post": {"content": MOCK_POST_IMAGE_RESPONSE, + "status_code": 201}}), + ] 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", @@ -219,27 +177,16 @@ class TestImage(unittest.TestCase): @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_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_IMAGES_RESPONSE}}), + test_base.get_mock_session( + ["post"], {"post": {"content": MOCK_POST_IMAGE_RESPONSE, + "status_code": 201}}), + ] 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", @@ -252,16 +199,14 @@ class TestImage(unittest.TestCase): @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_session.side_effect = [ + test_base.get_mock_session( + ["delete"], {"delete": {"content": {}, + "status_code": 204}}) + ] 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" diff --git a/newton/newton/requests/tests/test_network.py b/newton/newton/requests/tests/test_network.py index 3eb8b0e6..448c098b 100644 --- a/newton/newton/requests/tests/test_network.py +++ b/newton/newton/requests/tests/test_network.py @@ -20,6 +20,7 @@ from rest_framework import status from newton.requests.views.util import VimDriverUtils from newton.requests.tests import mock_info +from newton.requests.tests import test_base MOCK_GET_NETWORKS_RESPONSE = { @@ -51,14 +52,6 @@ MOCK_POST_NETWORK_RESPONSE = { } -class MockResponse(object): - status_code = 200 - content = '' - - def json(self): - pass - - class TestNetwork(unittest.TestCase): def setUp(self): self.client = Client() @@ -70,16 +63,12 @@ class TestNetwork(unittest.TestCase): @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_get_networks(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_networks_response_obj = mock.Mock(spec=MockResponse) - mock_get_networks_response_obj.status_code = 200 - mock_get_networks_response_obj.content = MOCK_GET_NETWORKS_RESPONSE - mock_get_networks_response_obj.json.return_value = MOCK_GET_NETWORKS_RESPONSE - mock_session.get.return_value = mock_get_networks_response_obj + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_NETWORKS_RESPONSE}}), + ] 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/networks", @@ -94,16 +83,12 @@ class TestNetwork(unittest.TestCase): @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_get_network(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_network_response_obj = mock.Mock(spec=MockResponse) - mock_get_network_response_obj.status_code = 200 - mock_get_network_response_obj.content = MOCK_GET_NETWORK_RESPONSE - mock_get_network_response_obj.json.return_value = MOCK_GET_NETWORK_RESPONSE - mock_session.get.return_value = mock_get_network_response_obj + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_NETWORK_RESPONSE}}), + ] 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" @@ -118,16 +103,13 @@ class TestNetwork(unittest.TestCase): @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_get_network_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_network_response_obj = mock.Mock(spec=MockResponse) - mock_get_network_response_obj.status_code = 404 - mock_get_network_response_obj.context = {} - mock_get_network_response_obj.json.return_value = {} - mock_session.get.return_value = mock_get_network_response_obj + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": {}, + "status_code": 404}}), + ] 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" @@ -142,24 +124,15 @@ class TestNetwork(unittest.TestCase): @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_post(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_networks_response_obj = mock.Mock(spec=MockResponse) - mock_get_networks_response_obj.status_code = 200 - mock_get_networks_response_obj.content = MOCK_GET_NETWORKS_RESPONSE - mock_get_networks_response_obj.json.return_value = MOCK_GET_NETWORKS_RESPONSE - - mock_post_network_response_obj = mock.Mock(spec=MockResponse) - mock_post_network_response_obj.status_code = 202 - mock_post_network_response_obj.content = MOCK_POST_NETWORK_RESPONSE - mock_post_network_response_obj.json.return_value = MOCK_POST_NETWORK_RESPONSE - - mock_session.get.return_value = mock_get_networks_response_obj - mock_session.post.return_value = mock_post_network_response_obj + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_NETWORKS_RESPONSE}}), + test_base.get_mock_session( + ["post"], {"post": {"content": MOCK_POST_NETWORK_RESPONSE, + "status_code": 202}}), + ] 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/networks", @@ -173,24 +146,16 @@ class TestNetwork(unittest.TestCase): @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_post_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_networks_response_obj = mock.Mock(spec=MockResponse) - mock_get_networks_response_obj.status_code = 200 - mock_get_networks_response_obj.content = MOCK_GET_NETWORKS_RESPONSE - mock_get_networks_response_obj.json.return_value = MOCK_GET_NETWORKS_RESPONSE - mock_post_network_response_obj = mock.Mock(spec=MockResponse) - mock_post_network_response_obj.status_code = 202 - mock_post_network_response_obj.content = MOCK_POST_NETWORK_RESPONSE - mock_post_network_response_obj.json.return_value = MOCK_POST_NETWORK_RESPONSE - - mock_session.get.return_value = mock_get_networks_response_obj - mock_session.post.return_value = mock_post_network_response_obj + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_NETWORKS_RESPONSE}}), + test_base.get_mock_session( + ["post"], {"post": {"content": MOCK_POST_NETWORK_RESPONSE, + "status_code": 202}}), + ] 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/networks", @@ -204,24 +169,16 @@ class TestNetwork(unittest.TestCase): @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_post_empty_body(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_networks_response_obj = mock.Mock(spec=MockResponse) - mock_get_networks_response_obj.status_code = 200 - mock_get_networks_response_obj.content = MOCK_GET_NETWORKS_RESPONSE - mock_get_networks_response_obj.json.return_value = MOCK_GET_NETWORKS_RESPONSE - - mock_post_network_response_obj = mock.Mock(spec=MockResponse) - mock_post_network_response_obj.status_code = 202 - mock_post_network_response_obj.content = MOCK_POST_NETWORK_RESPONSE - mock_post_network_response_obj.json.return_value = MOCK_POST_NETWORK_RESPONSE - mock_session.get.return_value = mock_get_networks_response_obj - mock_session.post.return_value = mock_post_network_response_obj + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_NETWORKS_RESPONSE}}), + test_base.get_mock_session( + ["post"], {"post": {"content": MOCK_POST_NETWORK_RESPONSE, + "status_code": 202}}), + ] 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/networks", @@ -234,17 +191,13 @@ class TestNetwork(unittest.TestCase): @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_delete(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_network_response_obj = mock.Mock(spec=MockResponse) - mock_delete_network_response_obj.status_code = 204 - - mock_session.delete.return_value = mock_delete_network_response_obj + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["delete"], {"delete": {"content": {}, + "status_code": 204}}) + ] 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" diff --git a/newton/newton/requests/tests/test_subnet.py b/newton/newton/requests/tests/test_subnet.py new file mode 100644 index 00000000..e4b2805e --- /dev/null +++ b/newton/newton/requests/tests/test_subnet.py @@ -0,0 +1,188 @@ +# 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 django.test import Client +from rest_framework import status + +from newton.requests.tests import test_base +from newton.requests.tests import mock_info +from newton.requests.views.util import VimDriverUtils + + +MOCK_GET_SUBNETS_RESPONSE = { + "subnets": [ + {"id": "uuid_1", "name": "subnet_1"}, + {"id": "uuid_2", "name": "subnet_2"} + ] +} + +MOCK_GET_SUBNET_RESPONSE = { + "subnet": { + "id": "uuid_1", + "name": "subnet_1" + } +} + +MOCK_GET_SUBNET_RESPONSE_NOT_FOUND = { + "subnet": {} +} + +MOCK_POST_SUBNET_REQUEST = { + "id": "uuid_3", + "name": "subnet_3", +} + +MOCK_POST_SUBNET_REQUEST_EXISTING = { + "id": "uuid_1", + "name": "subnet_1", +} + +MOCK_POST_SUBNET_RESPONSE = { + "subnet": { + "id": "uuid_3", + "name": "subnet_3" + } +} + + +class TestSubnet(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_subnets(self, mock_get_vim_info, mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_SUBNETS_RESPONSE}}) + ] + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.get( + "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/subnets", + {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + + context = response.json() + self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertIsNotNone(context['subnets']) + self.assertEqual(MOCK_GET_SUBNETS_RESPONSE["subnets"], context["subnets"]) + + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_subnet(self, mock_get_vim_info, mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_SUBNET_RESPONSE}}) + ] + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.get( + "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/subnets" + "/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_SUBNET_RESPONSE["id"], context["id"]) + + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_subnet_not_found(self, mock_get_vim_info, mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_SUBNET_RESPONSE_NOT_FOUND, + "status_code": 404}}), + ] + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.get( + "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/subnets" + "/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("subnet")) + + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_create_subnet_success(self, mock_get_vim_info, mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_SUBNETS_RESPONSE}}), + test_base.get_mock_session( + ["post"], + {"post": {"content": MOCK_POST_SUBNET_RESPONSE, + "status_code": 201}}) + ] + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.post( + "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/subnets", + MOCK_POST_SUBNET_REQUEST, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + context = response.json() + + self.assertEquals(status.HTTP_201_CREATED, response.status_code) + self.assertEqual(MOCK_POST_SUBNET_RESPONSE["subnet"], context) + + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_create_subnet_fail_existing(self, mock_get_vim_info, mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_SUBNETS_RESPONSE}}), + test_base.get_mock_session( + ["post"], + {"post": {"content": MOCK_POST_SUBNET_RESPONSE, + "status_code": 404}}) + ] + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.post( + "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/subnets", + MOCK_POST_SUBNET_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_delete_subnet_success(self, mock_get_vim_info, mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["delete"], + {"delete": {"content": None, + "status_code": 204}}) + ] + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.delete( + "/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/fcca3cc49d5e42caae15459e27103efc/subnets" + "/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) diff --git a/newton/newton/requests/tests/test_volume.py b/newton/newton/requests/tests/test_volume.py new file mode 100644 index 00000000..8281e23d --- /dev/null +++ b/newton/newton/requests/tests/test_volume.py @@ -0,0 +1,188 @@ +# 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 django.test import Client +from rest_framework import status + +from newton.requests.tests import mock_info +from newton.requests.tests import test_base +from newton.requests.views.util import VimDriverUtils +from newton.requests.views.volume import Volumes + +MOCK_GET_VOLUMES_RESPONSE = { + "volumes": [ + {"id": "uuid_1", "name": "volume_1"}, + {"id": "uuid_2", "name": "volume_2"} + ] +} + +MOCK_GET_VOLUME_RESPONSE = { + "volume": { + "id": "uuid_1", + "name": "volume_1" + } +} + +MOCK_POST_VOLUME_REQUEST = { + "id": "uuid_3", + "name": "volume_3" +} + +MOCK_POST_VOLUME_REQUEST_EXISTING = { + "id": "uuid_1", + "name": "volume_1" +} + +MOCK_POST_VOLUME_RESPONSE = { + "volume": { + "id": "uuid_3", + "name": "volume_3" + } +} + +class TestVolumes(test_base.TestRequest): + + url = ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/" + "fcca3cc49d5e42caae15459e27103efc/") + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_volumes(self, mock_get_vim_info, mock_get_session): + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.return_value = test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_VOLUMES_RESPONSE}}) + + response = self.client.get( + self.url + "volumes", + {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + + context = response.json() + self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertIsNotNone(context['volumes']) + self.assertEqual(MOCK_GET_VOLUMES_RESPONSE["volumes"], context['volumes']) + + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_volume(self, mock_get_vim_info, mock_get_session): + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.return_value = test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_VOLUME_RESPONSE}}) + + response = self.client.get( + self.url + "volumes/uuid_1", + {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + + context = response.json() + self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEquals(MOCK_GET_VOLUME_RESPONSE['id'], context['id']) + + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_volume_not_found(self, mock_get_vim_info, mock_get_session): + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.return_value = test_base.get_mock_session( + ["get"], + {"get": {"content": None, "status_code": 404}}) + + response = self.client.get( + self.url + "volumes/uuid_3", + {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + + self.assertEquals(status.HTTP_500_INTERNAL_SERVER_ERROR, response.status_code) + self.assertIn('error', response.data) + + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_create_volume_successfully(self, mock_get_vim_info, mock_get_session): + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.return_value = test_base.get_mock_session( + ["get", "post"], + { + "get" : {"content": MOCK_GET_VOLUMES_RESPONSE}, + "post": {"content": MOCK_POST_VOLUME_RESPONSE, "status_code": 202} + }) + + response = self.client.post( + self.url + "volumes", + MOCK_POST_VOLUME_REQUEST, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + + context = response.json() + self.assertEquals(status.HTTP_202_ACCEPTED, 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_create_existing_volume(self, mock_get_vim_info, mock_get_session): + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.return_value = test_base.get_mock_session( + ["get", "post"], + { + "get" : {"content": MOCK_GET_VOLUMES_RESPONSE}, + "post": {"content": MOCK_POST_VOLUME_RESPONSE, "status_code": 202} + }) + + response = self.client.post( + self.url + "volumes", + MOCK_POST_VOLUME_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_create_volume_unsuccessfully(self, mock_get_vim_info, mock_get_session): + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.return_value = test_base.get_mock_session( + ["get", "post"], + { + "get" : {"content": MOCK_GET_VOLUMES_RESPONSE}, + "post": {"content": MOCK_POST_VOLUME_RESPONSE, "status_code": 202} + }) + + response = self.client.post( + self.url + "volumes", + {}, 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_volume(self, mock_get_vim_info, mock_get_session): + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.return_value = test_base.get_mock_session( + ["delete"], + {"delete" : {"content": None, "status_code": 204}}) + + response = self.client.delete( + self.url + "volumes/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) |