3 # Copyright (c) 2017 Orange and others.
5 # All rights reserved. This program and the accompanying materials
6 # are made available under the terms of the Apache License, Version 2.0
7 # which accompanies this distribution, and is available at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 """Define the class required to fully cover decorators."""
12 from datetime import datetime
21 from functest.utils import decorators
22 from functest.utils import functest_utils
23 from functest.utils.constants import CONST
25 __author__ = "Cedric Ollivier <cedric.ollivier@orange.com>"
29 FILE = '{}/null'.format(DIR)
30 URL = 'file://{}'.format(FILE)
33 class DecoratorsTesting(unittest.TestCase):
34 # pylint: disable=missing-docstring
37 _project_name = 'functest'
43 _deploy_scenario = 'foo'
44 _installer_type = 'debian'
47 os.environ['INSTALLER_TYPE'] = self._installer_type
48 os.environ['DEPLOY_SCENARIO'] = self._deploy_scenario
49 os.environ['NODE_NAME'] = self._node_name
50 os.environ['BUILD_TAG'] = self._build_tag
53 self.assertEqual(functest_utils.push_results_to_db.__name__,
57 stop_time = datetime.fromtimestamp(self._stop_time).strftime(
59 start_time = datetime.fromtimestamp(self._start_time).strftime(
61 data = {'project_name': self._project_name,
62 'stop_date': stop_time, 'start_date': start_time,
63 'case_name': self._case_name, 'build_tag': self._build_tag,
64 'pod_name': self._node_name, 'installer': self._installer_type,
65 'scenario': self._deploy_scenario, 'version': VERSION,
66 'details': {}, 'criteria': self._result}
67 return json.dumps(data, sort_keys=True)
69 @mock.patch('{}.get_version'.format(functest_utils.__name__),
71 @mock.patch('requests.post')
72 def test_http_shema(self, *args):
73 CONST.__setattr__('results_test_db_url', 'http://127.0.0.1')
74 self.assertTrue(functest_utils.push_results_to_db(
75 self._project_name, self._case_name, self._start_time,
76 self._stop_time, self._result, {}))
77 args[1].assert_called_once_with()
78 args[0].assert_called_once_with(
79 'http://127.0.0.1', data=self._get_json(),
80 headers={'Content-Type': 'application/json'})
82 def test_wrong_shema(self):
83 CONST.__setattr__('results_test_db_url', '/dev/null')
84 self.assertFalse(functest_utils.push_results_to_db(
85 self._project_name, self._case_name, self._start_time,
86 self._stop_time, self._result, {}))
88 @mock.patch('{}.get_version'.format(functest_utils.__name__),
90 def _test_dump(self, *args):
91 CONST.__setattr__('results_test_db_url', URL)
92 with mock.patch.object(decorators, 'open', mock.mock_open(),
93 create=True) as mock_open:
94 self.assertTrue(functest_utils.push_results_to_db(
95 self._project_name, self._case_name, self._start_time,
96 self._stop_time, self._result, {}))
97 mock_open.assert_called_once_with(FILE, 'a')
99 call_args, _ = handle.write.call_args
100 self.assertIn('POST', call_args[0])
101 self.assertIn(self._get_json(), call_args[0])
102 args[0].assert_called_once_with()
104 @mock.patch('os.makedirs')
105 def test_default_dump(self, mock_method=None):
107 mock_method.assert_called_once_with(DIR)
109 @mock.patch('os.makedirs', side_effect=OSError(errno.EEXIST, ''))
110 def test_makedirs_dir_exists(self, mock_method=None):
112 mock_method.assert_called_once_with(DIR)
114 @mock.patch('os.makedirs', side_effect=OSError)
115 def test_makedirs_exc(self, *args):
116 CONST.__setattr__('results_test_db_url', URL)
118 functest_utils.push_results_to_db(
119 self._project_name, self._case_name, self._start_time,
120 self._stop_time, self._result, {}))
121 args[0].assert_called_once_with(DIR)
124 if __name__ == "__main__":
125 logging.disable(logging.CRITICAL)
126 unittest.main(verbosity=2)