diff options
author | Victor Morales <victor.morales@intel.com> | 2017-09-27 11:21:07 -0700 |
---|---|---|
committer | Victor Morales <victor.morales@intel.com> | 2017-09-27 11:22:13 -0700 |
commit | 003e56fd3f537e37a791ad8f6e74005f84e352d6 (patch) | |
tree | ab35d27fb19edd9eaacb9eb58b5f894ba427700c | |
parent | fb3707152f43c9c9ef17182baea9a8f3e63e54c2 (diff) |
Refactor flavor UTs
The creation of base class helps to reduce code and improve existing
Unit Tests. This changes uses the latest improvements.
Change-Id: Ib0fa298c72df73761320ab415c22d6eabbc8caea
Signed-off-by: Victor Morales <victor.morales@intel.com>
Issue-Id: MULTICLOUD-83
-rw-r--r-- | newton/newton/requests/tests/test_flavor.py | 413 | ||||
-rw-r--r-- | newton/newton/requests/views/flavor.py | 90 |
2 files changed, 228 insertions, 275 deletions
diff --git a/newton/newton/requests/tests/test_flavor.py b/newton/newton/requests/tests/test_flavor.py index cb7a4ffe..7ddb1122 100644 --- a/newton/newton/requests/tests/test_flavor.py +++ b/newton/newton/requests/tests/test_flavor.py @@ -13,14 +13,13 @@ # 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.flavor import Flavors from newton.requests.views.util import VimDriverUtils -from newton.requests.tests import mock_info MOCK_GET_FLAVORS_RESPONSE = { @@ -37,6 +36,12 @@ MOCK_GET_FLAVOR_RESPONSE = { } } +MOCK_GET_EXTRA_SPECS = { + "extra_specs": { + "key": "test" + } +} + MOCK_POST_FLAVOR_REQUEST = { "id": "uuid_3", "name": "flavor_3" @@ -55,234 +60,198 @@ MOCK_POST_FLAVOR_RESPONSE = { } -class MockResponse(object): - status_code = 200 - content = '' - - def json(self): - pass - - -class TestFlavors(unittest.TestCase): - def setUp(self): - self.client = Client() - - def tearDown(self): - pass - - @mock.patch.object(Flavors, 'get_flavor_extra_specs') - @mock.patch.object(VimDriverUtils, 'get_session') - @mock.patch.object(VimDriverUtils, 'get_vim_info') - def test_get_flavors(self, mock_get_vim_info, mock_get_session, mock_get_flavor_extra_specs): - - mock_session_specs = ["get"] - mock_session = mock.Mock(name='mock_session', spec=mock_session_specs) - - mock_get_flavors_response_obj = mock.Mock(spec=MockResponse) - mock_get_flavors_response_obj.status_code = 200 - mock_get_flavors_response_obj.content = MOCK_GET_FLAVORS_RESPONSE - mock_get_flavors_response_obj.json.return_value = MOCK_GET_FLAVORS_RESPONSE - mock_session.get.return_value = mock_get_flavors_response_obj - - mock_extra_specs = mock.Mock(spec=MockResponse) +class TestFlavors(test_base.TestRequest): + + @mock.patch.object(Flavors, '_get_flavor_extra_specs') + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_flavors(self, mock_get_vim_info, mock_get_session, + mock_get_flavor_extra_specs): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], {"get": {"content": MOCK_GET_FLAVORS_RESPONSE}}), + ] + + mock_extra_specs = mock.Mock(spec=test_base.MockResponse) + mock_extra_specs.json.return_value = {"extra_specs": {}} + + mock_get_flavor_extra_specs.return_value = mock_extra_specs + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.get( + ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne" + "/fcca3cc49d5e42caae15459e27103efc/flavors"), + {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + context = response.json() + + self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertIsNotNone(context['flavors']) + self.assertEqual(MOCK_GET_FLAVORS_RESPONSE["flavors"], + context['flavors']) + + @mock.patch.object(Flavors, '_get_flavor_extra_specs') + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_flavor(self, mock_get_vim_info, mock_get_session, + mock_get_flavor_extra_specs): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], + {"get": {"content": MOCK_GET_FLAVOR_RESPONSE}}), + ] + + mock_extra_specs = mock.Mock(spec=test_base.MockResponse) + mock_extra_specs.json.return_value = {"extra_specs": {}} + + mock_get_flavor_extra_specs.return_value = mock_extra_specs + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.get( + ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne" + "/fcca3cc49d5e42caae15459e27103efc/flavors/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_FLAVOR_RESPONSE["id"], context["id"]) + + @mock.patch.object(Flavors, '_get_flavor_extra_specs') + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_get_flavor_not_found( + self, mock_get_vim_info, mock_get_session, + mock_get_flavor_extra_specs): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get"], + {"get": {"status_code":status.HTTP_404_NOT_FOUND}}), + ] + + mock_extra_specs = mock.Mock(spec=test_base.MockResponse) + mock_extra_specs.json.return_value = {"extra_specs": {}} + + mock_get_flavor_extra_specs.return_value = mock_extra_specs + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.get( + ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne" + "/fcca3cc49d5e42caae15459e27103efc/flavors/uuid_1"), + {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + + # TODO(sshank): 404 status is not possible. + 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_flavor(self, mock_get_vim_info, mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get", "post"], { + "get": {"content": MOCK_GET_FLAVORS_RESPONSE}, + "post": { + "content": MOCK_POST_FLAVOR_RESPONSE, + "status_code": status.HTTP_202_ACCEPTED, + } + } + ), + ] + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + + response = self.client.post( + ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne" + "/fcca3cc49d5e42caae15459e27103efc/flavors"), + MOCK_POST_FLAVOR_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(Flavors, '_get_flavor_extra_specs') + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_create_existing_flavor( + self, mock_get_vim_info, mock_get_session, + mock_get_flavor_extra_specs): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get", "post"], { + "get": {"content": MOCK_GET_FLAVORS_RESPONSE}, + "post": { + "content": MOCK_POST_FLAVOR_RESPONSE, + "status_code": status.HTTP_202_ACCEPTED, + } + }), + ] + mock_extra_specs = mock.Mock(spec=test_base.MockResponse) mock_extra_specs.json.return_value = {"extra_specs": {}} mock_get_flavor_extra_specs.return_value = mock_extra_specs 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/flavors", - {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - context = response.json() - - self.assertEquals(status.HTTP_200_OK, response.status_code) - self.assertIsNotNone(context['flavors']) - self.assertEqual(MOCK_GET_FLAVORS_RESPONSE["flavors"], context['flavors']) - - @mock.patch.object(Flavors, 'get_flavor_extra_specs') - @mock.patch.object(VimDriverUtils, 'get_session') - @mock.patch.object(VimDriverUtils, 'get_vim_info') - def test_get_flavor(self, mock_get_vim_info, mock_get_session, mock_get_flavor_extra_specs): - - mock_session_specs = ["get"] - mock_session = mock.Mock(name='mock_session', spec=mock_session_specs) - - mock_get_flavor_response_obj = mock.Mock(spec=MockResponse) - mock_get_flavor_response_obj.status_code = 200 - mock_get_flavor_response_obj.content = MOCK_GET_FLAVOR_RESPONSE - mock_get_flavor_response_obj.json.return_value = MOCK_GET_FLAVOR_RESPONSE - mock_session.get.return_value = mock_get_flavor_response_obj - - mock_extra_specs = mock.Mock(spec=MockResponse) - mock_extra_specs.json.return_value = {"extra_specs": {}} - - mock_get_flavor_extra_specs.return_value = mock_extra_specs - 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" - "/flavors/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_FLAVOR_RESPONSE["id"], context["id"]) - - @mock.patch.object(Flavors, 'get_flavor_extra_specs') - @mock.patch.object(VimDriverUtils, 'get_session') - @mock.patch.object(VimDriverUtils, 'get_vim_info') - def test_get_flavor_not_found(self, mock_get_vim_info, mock_get_session, mock_get_flavor_extra_specs): - mock_session_specs = ["get"] - mock_session = mock.Mock(name='mock_session', spec=mock_session_specs) - - mock_get_flavor_response_obj = mock.Mock(spec=MockResponse) - mock_get_flavor_response_obj.status_code = 404 - mock_get_flavor_response_obj.content = {} - mock_get_flavor_response_obj.json.return_value = {} - mock_session.get.return_value = mock_get_flavor_response_obj - - mock_extra_specs = mock.Mock(spec=MockResponse) - mock_extra_specs.json.return_value = {"extra_specs": {}} - - mock_get_flavor_extra_specs.return_value = mock_extra_specs - 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" - "/flavors/uuid_1", - {}, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - - # TODO(sshank): 404 status is not possible. - 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_post_flavor(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_flavors_response_obj = mock.Mock(spec=MockResponse) - mock_get_flavors_response_obj.status_code = 200 - mock_get_flavors_response_obj.content = MOCK_GET_FLAVORS_RESPONSE - mock_get_flavors_response_obj.json.return_value = MOCK_GET_FLAVORS_RESPONSE - mock_session.get.return_value = mock_get_flavors_response_obj - - mock_post_flavor_response_obj = mock.Mock(spec=MockResponse) - mock_post_flavor_response_obj.status_code = 202 - mock_post_flavor_response_obj.content = MOCK_POST_FLAVOR_RESPONSE - mock_post_flavor_response_obj.json.return_value = MOCK_POST_FLAVOR_RESPONSE - mock_session.post.return_value = mock_post_flavor_response_obj - - mock_extra_specs = mock.Mock(spec=MockResponse) - mock_extra_specs.json.return_value = {"extra_specs": {}} - - #mock_get_flavor_extra_specs.return_value = mock_extra_specs - 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/flavors", - MOCK_POST_FLAVOR_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(Flavors, 'get_flavor_extra_specs') - @mock.patch.object(VimDriverUtils, 'get_session') - @mock.patch.object(VimDriverUtils, 'get_vim_info') - def test_post_flavor_existing(self, mock_get_vim_info, mock_get_session, mock_get_flavor_extra_specs): - mock_session_specs = ["get", "post"] - mock_session = mock.Mock(name='mock_session', spec=mock_session_specs) - - mock_get_flavors_response_obj = mock.Mock(spec=MockResponse) - mock_get_flavors_response_obj.status_code = 200 - mock_get_flavors_response_obj.content = MOCK_GET_FLAVORS_RESPONSE - mock_get_flavors_response_obj.json.return_value = MOCK_GET_FLAVORS_RESPONSE - mock_session.get.return_value = mock_get_flavors_response_obj - - mock_post_flavor_response_obj = mock.Mock(spec=MockResponse) - mock_post_flavor_response_obj.status_code = 202 - mock_post_flavor_response_obj.content = MOCK_POST_FLAVOR_RESPONSE - mock_post_flavor_response_obj.json.return_value = MOCK_POST_FLAVOR_RESPONSE - mock_session.post.return_value = mock_post_flavor_response_obj - - mock_extra_specs = mock.Mock(spec=MockResponse) - mock_extra_specs.json.return_value = {"extra_specs": {}} - - mock_get_flavor_extra_specs.return_value = mock_extra_specs - 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/flavors", - MOCK_POST_FLAVOR_REQUEST_EXISTING, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) + ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/" + "fcca3cc49d5e42caae15459e27103efc/flavors"), + MOCK_POST_FLAVOR_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_flavor_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_flavors_response_obj = mock.Mock(spec=MockResponse) - mock_get_flavors_response_obj.status_code = 200 - mock_get_flavors_response_obj.content = MOCK_GET_FLAVORS_RESPONSE - mock_get_flavors_response_obj.json.return_value = MOCK_GET_FLAVORS_RESPONSE - mock_session.get.return_value = mock_get_flavors_response_obj - - mock_post_flavor_response_obj = mock.Mock(spec=MockResponse) - mock_post_flavor_response_obj.status_code = 202 - mock_post_flavor_response_obj.content = MOCK_POST_FLAVOR_RESPONSE - mock_post_flavor_response_obj.json.return_value = MOCK_POST_FLAVOR_RESPONSE - mock_session.post.return_value = mock_post_flavor_response_obj - - mock_extra_specs = mock.Mock(spec=MockResponse) - mock_extra_specs.json.return_value = {"extra_specs": {}} - - 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/flavors", - {}, 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(Flavors, 'get_flavor_extra_specs') - @mock.patch.object(VimDriverUtils, 'get_session') - @mock.patch.object(VimDriverUtils, 'get_vim_info') - def test_delete_flavor(self, mock_get_vim_info, mock_get_session, mock_get_flavor_extra_specs): - mock_session_specs = ["delete"] - mock_session = mock.Mock(name='mock_session', spec=mock_session_specs) - - mock_delete_flavor_response_obj = mock.Mock(spec=MockResponse) - mock_delete_flavor_response_obj.status_code = 204 - - mock_session.delete.return_value = mock_delete_flavor_response_obj - - mock_extra_specs = mock.Mock(spec=MockResponse) - mock_extra_specs.json.return_value = {"extra_specs": {}} - - 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" - "/flavors/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) + @mock.patch.object(VimDriverUtils, 'get_session') + @mock.patch.object(VimDriverUtils, 'get_vim_info') + def test_create_empty_flavor( + self, mock_get_vim_info,mock_get_session): + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get", "post"], + { + "get": {"content": MOCK_GET_FLAVORS_RESPONSE}, + "post": { + "content": MOCK_POST_FLAVOR_RESPONSE, + "status_code": status.HTTP_202_ACCEPTED + } + }) + ] + mock_extra_specs = mock.Mock(spec=test_base.MockResponse) + mock_extra_specs.json.return_value = {"extra_specs": {}} + + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + response = self.client.post( + ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/" + "fcca3cc49d5e42caae15459e27103efc/flavors"), + {}, 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_flavor(self, mock_get_vim_info, mock_get_session): + mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO + mock_get_session.side_effect = [ + test_base.get_mock_session( + ["get", "delete"], + { + "get": { "content": MOCK_GET_EXTRA_SPECS }, + "delete": {"status_code": status.HTTP_204_NO_CONTENT } + }), + ] + + response = self.client.delete( + ("/api/multicloud-newton/v0/windriver-hudson-dc_RegionOne/" + "fcca3cc49d5e42caae15459e27103efc/flavors/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/views/flavor.py b/newton/newton/requests/views/flavor.py index 87d5481e..2797461d 100644 --- a/newton/newton/requests/views/flavor.py +++ b/newton/newton/requests/views/flavor.py @@ -14,13 +14,13 @@ import logging import json import traceback + from keystoneauth1.exceptions import HttpError from rest_framework import status from rest_framework.response import Response from rest_framework.views import APIView from newton.pub.exceptions import VimDriverNewtonException - from newton.requests.views.util import VimDriverUtils logger = logging.getLogger(__name__) @@ -38,7 +38,7 @@ class Flavors(APIView): ("extra_specs", "extraSpecs"), ] - def convertExtraSpecs(self, extraSpecs, extra_specs, reverse=False): + def _convert_extra_specs(self, extraSpecs, extra_specs, reverse=False): if reverse == False: #from extraSpecs to extra_specs for spec in extraSpecs: @@ -58,16 +58,16 @@ class Flavors(APIView): vim = VimDriverUtils.get_vim_info(vimid) sess = VimDriverUtils.get_session(vim, tenantid) - resp = self.get_flavor(sess, request, flavorid) + resp = self._get_flavor(sess, request, flavorid) content = resp.json() if flavorid: flavor = content.pop("flavor", None) - extraResp = self.get_flavor_extra_specs(sess, flavor["id"]) + extraResp = self._get_flavor_extra_specs(sess, flavor["id"]) extraContent = extraResp.json() if extraContent["extra_specs"]: extraSpecs = [] - self.convertExtraSpecs(extraSpecs, extraContent["extra_specs"], True) + self._convert_extra_specs(extraSpecs, extraContent["extra_specs"], True) flavor["extraSpecs"] = extraSpecs VimDriverUtils.replace_key_by_mapping(flavor, self.keys_mapping) @@ -92,11 +92,11 @@ class Flavors(APIView): #iterate each flavor to get extra_specs for flavor in content["flavors"]: - extraResp = self.get_flavor_extra_specs(sess, flavor["id"]) + extraResp = self._get_flavor_extra_specs(sess, flavor["id"]) extraContent = extraResp.json() if extraContent["extra_specs"]: extraSpecs = [] - self.convertExtraSpecs(extraSpecs, extraContent["extra_specs"], True) + self._convert_extra_specs(extraSpecs, extraContent["extra_specs"], True) flavor["extraSpecs"] = extraSpecs VimDriverUtils.replace_key_by_mapping(flavor, self.keys_mapping) @@ -121,22 +121,19 @@ class Flavors(APIView): return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) - def get_flavor_extra_specs(self, sess, flavorid): - if not flavorid: - return {} - else: + def _get_flavor_extra_specs(self, sess, flavorid): + if flavorid: logger.debug("Flavors--get_extra_specs::> %s" % flavorid) # prepare request resource to vim instance req_resouce = "/flavors/%s/os-extra_specs" % flavorid resp = sess.get(req_resouce, endpoint_filter=self.service) return resp + return {} - def get_flavor(self, sess, request, flavorid=""): + def _get_flavor(self, sess, request, flavorid=""): logger.debug("Flavors--get basic") - if not sess: - return {} - else: + if sess: # prepare request resource to vim instance req_resouce = "/flavors" if flavorid: @@ -148,9 +145,8 @@ class Flavors(APIView): if query: req_resouce += "?%s" % query - resp = sess.get(req_resouce, endpoint_filter=self.service) - return resp - + return sess.get(req_resouce, endpoint_filter=self.service) + return {} def post(self, request, vimid="", tenantid="", flavorid=""): logger.debug("Flavors--post::> %s" % request.data) @@ -163,7 +159,7 @@ class Flavors(APIView): sess = VimDriverUtils.get_session(vim, tenantid) #check if the flavor is already created: name or id - tmpresp = self.get_flavor(sess, request) + tmpresp = self._get_flavor(sess, request) content = tmpresp.json() #iterate each flavor to get extra_specs existed = False @@ -175,12 +171,12 @@ class Flavors(APIView): existed = True break - if existed == True: - extraResp = self.get_flavor_extra_specs(sess, flavor["id"]) + if existed: + extraResp = self._get_flavor_extra_specs(sess, flavor["id"]) extraContent = extraResp.json() if extraContent["extra_specs"]: extraSpecs = [] - self.convertExtraSpecs(extraSpecs, extraContent["extra_specs"], True) + self._convert_extra_specs(extraSpecs, extraContent["extra_specs"], True) flavor["extraSpecs"] = extraSpecs VimDriverUtils.replace_key_by_mapping(flavor, self.keys_mapping) @@ -195,7 +191,7 @@ class Flavors(APIView): extraSpecs = request.data.pop("extraSpecs", None) #create flavor first - resp = self.create_flavor(sess, request) + resp = self._create_flavor(sess, request) if resp.status_code == 202: resp_body = resp.json()["flavor"] else: @@ -205,20 +201,20 @@ class Flavors(APIView): flavorid = resp_body['id'] if extraSpecs: extra_specs={} - self.convertExtraSpecs(extraSpecs, extra_specs, False) + self._convert_extra_specs(extraSpecs, extra_specs, False) # logger.debug("extraSpecs:%s" % extraSpecs) # logger.debug("extra_specs:%s" % extra_specs) - extraResp = self.create_flavor_extra_specs(sess, extra_specs, flavorid) + extraResp = self._create_flavor_extra_specs(sess, extra_specs, flavorid) if extraResp.status_code == 200: #combine the response body and return tmpSpecs = [] tmpRespBody = extraResp.json() - self.convertExtraSpecs(tmpSpecs, tmpRespBody['extra_specs'], True) + self._convert_extra_specs(tmpSpecs, tmpRespBody['extra_specs'], True) resp_body.update({"extraSpecs":tmpSpecs}) else: #rollback - self.delete_flavor(self, request, vimid, tenantid, flavorid) + self._delete_flavor(self, request, vimid, tenantid, flavorid) return extraResp VimDriverUtils.replace_key_by_mapping(resp_body, self.keys_mapping) @@ -232,7 +228,7 @@ class Flavors(APIView): return Response(data=resp_body, status=resp.status_code) except VimDriverNewtonException as e: if sess and resp and resp.status_code == 200: - self.delete_flavor(sess, flavorid) + self._delete_flavor(sess, flavorid) return Response(data={'error': e.content}, status=e.status_code) except HttpError as e: @@ -242,13 +238,12 @@ class Flavors(APIView): logger.error(traceback.format_exc()) if sess and resp and resp.status_code == 200: - self.delete_flavor(sess, flavorid) + self._delete_flavor(sess, flavorid) return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) - - def create_flavor(self, sess, request): + def _create_flavor(self, sess, request): logger.debug("Flavors--create::> %s" % request.data) # prepare request resource to vim instance req_resouce = "/flavors" @@ -261,9 +256,7 @@ class Flavors(APIView): return sess.post(req_resouce, data=req_body, endpoint_filter=self.service) - - - def create_flavor_extra_specs(self, sess, extraspecs, flavorid): + def _create_flavor_extra_specs(self, sess, extraspecs, flavorid): logger.debug("Flavors extra_specs--post::> %s" % extraspecs) # prepare request resource to vim instance req_resouce = "/flavors" @@ -279,10 +272,6 @@ class Flavors(APIView): return sess.post(req_resouce, data=req_body, endpoint_filter=self.service) - - - - def delete(self, request, vimid="", tenantid="", flavorid=""): logger.debug("Flavors--delete::> %s" % request.data) try: @@ -291,10 +280,10 @@ class Flavors(APIView): sess = VimDriverUtils.get_session(vim, tenantid) #delete extra specs one by one - resp = self.delete_flavor_extra_specs(sess, flavorid) + resp = self._delete_flavor_extra_specs(sess, flavorid) #delete flavor - resp = self.delete_flavor(sess, flavorid) + resp = self._delete_flavor(sess, flavorid) #return results return Response(status=resp.status_code) @@ -308,20 +297,19 @@ class Flavors(APIView): return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) - - def delete_flavor_extra_specs(self, sess, flavorid): + def _delete_flavor_extra_specs(self, sess, flavorid): logger.debug("Flavors--delete extra::> %s" % flavorid) #delete extra specs one by one - resp = self.get_flavor_extra_specs(sess, flavorid) + resp = self._get_flavor_extra_specs(sess, flavorid) extra_specs = resp.json() if extra_specs and extra_specs["extra_specs"]: for k, _ in extra_specs["extra_specs"].items(): # just try to delete extra spec, but do not care if succeeded - self.delete_flavor_one_extra_spec(sess, flavorid, k) + self._delete_flavor_one_extra_spec(sess, flavorid, k) return resp - def delete_flavor_one_extra_spec(self, sess, flavorid, extra_spec_key): + def _delete_flavor_one_extra_spec(self, sess, flavorid, extra_spec_key): logger.debug("Flavors--delete 1 extra::> %s" % extra_spec_key) # prepare request resource to vim instance try: @@ -332,10 +320,9 @@ class Flavors(APIView): else: raise VimDriverNewtonException(message="VIM newton exception", content="internal bug in deleting flavor extra specs: %s" % extra_spec_key, - status_code=500) + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR) - resp = sess.delete(req_resouce, endpoint_filter=self.service) - return resp + return sess.delete(req_resouce, endpoint_filter=self.service) except HttpError as e: logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json())) return Response(data=e.response.json(), status=e.http_status) @@ -344,8 +331,7 @@ class Flavors(APIView): return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) - - def delete_flavor(self, sess, flavorid): + def _delete_flavor(self, sess, flavorid): logger.debug("Flavors--delete basic::> %s" % flavorid) # prepare request resource to vim instance req_resouce = "/flavors" @@ -356,6 +342,4 @@ class Flavors(APIView): content="internal bug in deleting flavor", status_code=500) - resp = sess.delete(req_resouce, endpoint_filter=self.service) - return resp - + return sess.delete(req_resouce, endpoint_filter=self.service) |