diff options
author | emartin <ephraim.martin@est.tech> | 2019-12-11 11:48:53 +0000 |
---|---|---|
committer | emartin <ephraim.martin@est.tech> | 2020-01-07 15:28:36 +0000 |
commit | 331a93a59c60f570c2ab231b425f9cd46c07ccf4 (patch) | |
tree | ec144b15122053336538ed6d590a0c1b6aa4cafa /components/pm-subscription-handler/tests/config_handler_test.py | |
parent | ab4c8376241c153e3f8ab986bd29a8baa8226450 (diff) |
Pull config from CBS
Signed-off-by: emartin <ephraim.martin@est.tech>
Issue-ID: DCAEGEN2-1843
Change-Id: Id97960dc02aca5531be016d70ea388473b556683
Diffstat (limited to 'components/pm-subscription-handler/tests/config_handler_test.py')
-rwxr-xr-x | components/pm-subscription-handler/tests/config_handler_test.py | 112 |
1 files changed, 112 insertions, 0 deletions
diff --git a/components/pm-subscription-handler/tests/config_handler_test.py b/components/pm-subscription-handler/tests/config_handler_test.py new file mode 100755 index 00000000..fcc25d60 --- /dev/null +++ b/components/pm-subscription-handler/tests/config_handler_test.py @@ -0,0 +1,112 @@ +# ============LICENSE_START=================================================== +# Copyright (C) 2019-2020 Nordix Foundation. +# ============================================================================ +# 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. +# +# SPDX-License-Identifier: Apache-2.0 +# ============LICENSE_END===================================================== + +import json +import unittest +from os import environ +from os import path +from unittest.mock import patch + +import requests +import responses +from tenacity import wait_none + +from pmsh_service.mod.config_handler import ConfigHandler + + +class ConfigHandlerTest(unittest.TestCase): + + def setUp(self): + self.env_vars = {'CONFIG_BINDING_SERVICE_SERVICE_HOST': 'cbs_hostname', + 'CONFIG_BINDING_SERVICE_SERVICE_PORT': '10000', + 'HOSTNAME': 'hostname'} + for key, value in self.env_vars.items(): + environ[key] = value + self.cbs_url = 'http://cbs_hostname:10000/service_component_all/hostname' + self.expected_config = self._get_expected_config() + + def test_missing_environment_variable(self): + for key, value in self.env_vars.items(): + with self.assertRaises(KeyError): + environ.pop(key) + test_value = globals()[value] + test_value() + environ[key] = value + + @responses.activate + def test_get_config_success(self): + responses.add(responses.GET, self.cbs_url, json=json.dumps(self.expected_config), + status=200) + + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + + self.assertEqual(self.expected_config, config_handler.get_config()) + + def test_get_config_already_exists(self): + config_handler = ConfigHandler() + expected_config = self._get_expected_config() + config_handler._config = expected_config + + with patch.object(requests, 'get') as mock_get_request: + actual_config = config_handler.get_config() + + self.assertEqual(0, mock_get_request.call_count) + self.assertEqual(expected_config, actual_config) + + @responses.activate + def test_get_config_error(self): + responses.add(responses.GET, self.cbs_url, status=404) + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + + with self.assertRaises(Exception): + config_handler.get_config() + + @responses.activate + def test_get_config_max_retries_error(self): + retry_limit = 5 + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + + for __ in range(retry_limit): + responses.add(responses.GET, self.cbs_url, status=500) + + with self.assertRaises(Exception): + config_handler.get_config() + self.assertEqual(retry_limit, len(responses.calls)) + + @responses.activate + def test_get_config_less_than_5_retries_success(self): + retry_attempts = 4 + responses.add(responses.GET, self.cbs_url, status=500) + responses.add(responses.GET, self.cbs_url, status=400) + responses.add(responses.GET, self.cbs_url, status=300) + responses.add(responses.GET, self.cbs_url, json=json.dumps(self.expected_config), + status=200) + + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + config_handler.get_config() + + self.assertEqual(retry_attempts, len(responses.calls)) + + @staticmethod + def _get_expected_config(): + with open(path.join(path.dirname(__file__), 'expected_config.json'))as json_file: + return json.load(json_file) |