diff options
Diffstat (limited to 'pike/pike/resource/tests')
-rw-r--r-- | pike/pike/resource/tests/test_capacity.py | 46 | ||||
-rw-r--r-- | pike/pike/resource/tests/tests_infra_workload.py | 28 |
2 files changed, 53 insertions, 21 deletions
diff --git a/pike/pike/resource/tests/test_capacity.py b/pike/pike/resource/tests/test_capacity.py index d6f3ea32..cfee8075 100644 --- a/pike/pike/resource/tests/test_capacity.py +++ b/pike/pike/resource/tests/test_capacity.py @@ -1,3 +1,6 @@ +''' +test capability +''' # Copyright (c) 2018 Intel Corporation. # # Licensed under the Apache License, Version 2.0 (the "License"); @@ -12,8 +15,8 @@ # See the License for the specific language governing permissions and # limitations under the License. -import mock import json +import mock from rest_framework import status @@ -115,9 +118,9 @@ TEST_REQ_FAILED_SOURCE = { class TestCapacity(test_base.TestRequest): - def setUp(self): - super(TestCapacity, self).setUp() - + ''' + TestCapacity + ''' def _get_mock_response(self, return_value=None): mock_response = mock.Mock(spec=test_base.MockResponse) mock_response.status_code = status.HTTP_200_OK @@ -127,6 +130,9 @@ class TestCapacity(test_base.TestRequest): @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_capacity_check_success(self, mock_get_vim_info, mock_get_session): + ''' + test_capacity_check_success + ''' mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO mock_get_session.return_value = test_base.get_mock_session( ["get"], { @@ -142,12 +148,15 @@ class TestCapacity(test_base.TestRequest): TEST_REQ_SUCCESS_SOURCE, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual({"result": True}, response.data) @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_capacity_check_nova_limits_failed(self, mock_get_vim_info, mock_get_session): + ''' + test_capacity_check_nova_limits_failed + ''' mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO mock_get_session.return_value = test_base.get_mock_session( ["get"], { @@ -163,12 +172,15 @@ class TestCapacity(test_base.TestRequest): TEST_REQ_FAILED_SOURCE, HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual({"result": False}, response.data) @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_capacity_check_nova_hypervisor_outofram(self, mock_get_vim_info, mock_get_session): + ''' + test_capacity_check_nova_hypervisor_outofram + ''' mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO mock_get_session.return_value = test_base.get_mock_session( ["get"], { @@ -185,12 +197,15 @@ class TestCapacity(test_base.TestRequest): content_type='application/json', HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual({"result": False}, response.data) @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_capacity_check_nova_hypervisor_outofstorage(self, mock_get_vim_info, mock_get_session): + ''' + test_capacity_check_nova_hypervisor_outofstorage + ''' mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO mock_get_session.return_value = test_base.get_mock_session( ["get"], { @@ -207,12 +222,15 @@ class TestCapacity(test_base.TestRequest): content_type='application/json', HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual({"result": False}, response.data) @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_capacity_check_nova_hypervisor_outofvcpu(self, mock_get_vim_info, mock_get_session): + ''' + test_capacity_check_nova_hypervisor_outofvcpu + ''' mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO mock_get_session.return_value = test_base.get_mock_session( ["get"], { @@ -229,12 +247,15 @@ class TestCapacity(test_base.TestRequest): content_type='application/json', HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual({"result": False}, response.data) @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_capacity_check_nova_limits_outofram(self, mock_get_vim_info, mock_get_session): + ''' + test_capacity_check_nova_limits_outofram + ''' mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO mock_get_session.return_value = test_base.get_mock_session( ["get"], { @@ -251,12 +272,15 @@ class TestCapacity(test_base.TestRequest): content_type='application/json', HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual({"result": True}, response.data) @mock.patch.object(VimDriverUtils, 'get_session') @mock.patch.object(VimDriverUtils, 'get_vim_info') def test_capacity_check_volume_limits_outofstorage(self, mock_get_vim_info, mock_get_session): + ''' + test_capacity_check_volume_limits_outofstorage + ''' mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO mock_get_session.return_value = test_base.get_mock_session( ["get"], { @@ -273,5 +297,5 @@ class TestCapacity(test_base.TestRequest): content_type='application/json', HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID) - self.assertEquals(status.HTTP_200_OK, response.status_code) + self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual({"result": False}, response.data) diff --git a/pike/pike/resource/tests/tests_infra_workload.py b/pike/pike/resource/tests/tests_infra_workload.py index 52a76e58..c4df70b4 100644 --- a/pike/pike/resource/tests/tests_infra_workload.py +++ b/pike/pike/resource/tests/tests_infra_workload.py @@ -1,3 +1,6 @@ +''' +test infra workload +''' # Copyright (c) 2018 Intel Corporation. # # Licensed under the Apache License, Version 2.0 (the "License"); @@ -101,7 +104,8 @@ MOCK_HEAT_CREATE_BODY1 = { "get_param": "flavor" }, "image": "40be8d1a-3eb9-40de-8abd-43237517384f", - "user_data": "#!/bin/bash -xv\necho \"hello world\" > /root/hello-world.txt\n" + "user_data": + "#!/bin/bash -xv\necho \"hello world\" > /root/hello-world.txt\n" } } } @@ -154,7 +158,8 @@ MOCK_HEAT_CREATE_BODY2 = { "get_param": "flavor" }, "image": "40be8d1a-3eb9-40de-8abd-43237517384f", - "user_data": "#!/bin/bash -xv\necho \"hello world\" > /root/hello-world.txt\n" + "user_data": + "#!/bin/bash -xv\necho \"hello world\" > /root/hello-world.txt\n" } } } @@ -166,7 +171,7 @@ MOCK_HEAT_CREATE_BODY2 = { class InfraWorkloadTest(unittest.TestCase): def setUp(self): - self._InfraWorkload = InfraWorkload() + self._infra_workload = InfraWorkload() pass def tearDown(self): @@ -189,13 +194,14 @@ class InfraWorkloadTest(unittest.TestCase): vimid = "CloudOwner_Region1" - response = self._InfraWorkload.post(mock_request, vimid) + response = self._infra_workload.post(mock_request, vimid) self.assertEqual(response.status_code, status.HTTP_201_CREATED) pass @mock.patch.object(helper, 'MultiCloudServiceHelper') @mock.patch.object(helper, 'MultiCloudIdentityHelper') - def test_post_wo_oof_directive(self, mock_MultiCloudIdentityHelper, mock_MultiCloudServiceHelper): + def test_post_wo_oof_directive(self, mock_MultiCloudIdentityHelper, + mock_MultiCloudServiceHelper): mock_request = mock.Mock() mock_request.META = {"testkey": "testvalue"} mock_request.data = MOCK_HEAT_CREATE_BODY2 @@ -210,7 +216,7 @@ class InfraWorkloadTest(unittest.TestCase): vimid = "CloudOwner_Region1" - response = self._InfraWorkload.post(mock_request, vimid) + response = self._infra_workload.post(mock_request, vimid) self.assertEqual(response.status_code, status.HTTP_201_CREATED) pass @@ -231,7 +237,7 @@ class InfraWorkloadTest(unittest.TestCase): vimid = "CloudOwner_Region1" mock_stack_id = "MOCKED_HEAT_STACK_ID1" - response = self._InfraWorkload.get(mock_request, vimid, mock_stack_id) + response = self._infra_workload.get(mock_request, vimid, mock_stack_id) self.assertEqual(response.status_code, status.HTTP_200_OK) pass @@ -252,7 +258,7 @@ class InfraWorkloadTest(unittest.TestCase): vimid = "CloudOwner_Region1" mock_stack_id = "MOCKED_HEAT_STACK_ID1" - response = self._InfraWorkload.delete(mock_request, vimid, mock_stack_id) + response = self._infra_workload.delete(mock_request, vimid, mock_stack_id) self.assertEqual(response.status_code, status.HTTP_200_OK) pass @@ -305,7 +311,8 @@ class APIv1InfraWorkloadTest(unittest.TestCase): cloud_region_id = "Region1" mock_stack_id = "MOCKED_HEAT_STACK_ID1" - response = self._APIv1InfraWorkload.get(mock_request, cloud_owner, cloud_region_id, mock_stack_id) + response = self._APIv1InfraWorkload.get(mock_request, cloud_owner, + cloud_region_id, mock_stack_id) self.assertEqual(response.status_code, status.HTTP_200_OK) pass @@ -327,6 +334,7 @@ class APIv1InfraWorkloadTest(unittest.TestCase): cloud_region_id = "Region1" mock_stack_id = "MOCKED_HEAT_STACK_ID1" - response = self._APIv1InfraWorkload.delete(mock_request, cloud_owner, cloud_region_id, mock_stack_id) + response = self._APIv1InfraWorkload.delete(mock_request, cloud_owner, + cloud_region_id, mock_stack_id) self.assertEqual(response.status_code, status.HTTP_200_OK) pass |