3 # All rights reserved. This program and the accompanying materials
4 # are made available under the terms of the Apache License, Version 2.0
5 # which accompanies this distribution, and is available at
6 # http://www.apache.org/licenses/LICENSE-2.0
8 # pylint: disable=missing-docstring
16 from functest.ci import run_tests
17 from functest.utils.constants import CONST
18 from functest.core.testcase import TestCase
21 class FakeModule(TestCase):
23 def run(self, **kwargs):
27 class RunTestsTesting(unittest.TestCase):
30 self.runner = run_tests.Runner()
31 mock_test_case = mock.Mock()
32 mock_test_case.is_successful.return_value = TestCase.EX_OK
33 self.runner.executed_test_cases['test1'] = mock_test_case
34 self.runner.executed_test_cases['test2'] = mock_test_case
36 self.creds = {'OS_AUTH_URL': 'http://test_ip:test_port/v2.0',
37 'OS_USERNAME': 'test_os_username',
38 'OS_TENANT_NAME': 'test_tenant',
39 'OS_PASSWORD': 'test_password'}
40 self.test = {'test_name': 'test_name'}
41 self.tier = mock.Mock()
43 test1.get_name.return_value = 'test1'
45 test2.get_name.return_value = 'test2'
46 attrs = {'get_name.return_value': 'test_tier',
47 'get_tests.return_value': [test1, test2],
48 'get_ci_loop.return_value': 'test_ci_loop',
49 'get_test_names.return_value': ['test1', 'test2']}
50 self.tier.configure_mock(**attrs)
52 self.tiers = mock.Mock()
53 attrs = {'get_tiers.return_value': [self.tier]}
54 self.tiers.configure_mock(**attrs)
56 self.run_tests_parser = run_tests.RunTestsParser()
58 @mock.patch('functest.ci.run_tests.Runner.get_dict_by_test')
59 def test_get_run_dict(self, *args):
60 retval = {'run': mock.Mock()}
61 args[0].return_value = retval
62 self.assertEqual(self.runner.get_run_dict('test_name'), retval['run'])
63 args[0].assert_called_once_with('test_name')
65 @mock.patch('functest.ci.run_tests.LOGGER.error')
66 @mock.patch('functest.ci.run_tests.Runner.get_dict_by_test',
68 def test_get_run_dict_config_ko(self, *args):
69 testname = 'test_name'
70 self.assertEqual(self.runner.get_run_dict(testname), None)
71 args[0].return_value = {}
72 self.assertEqual(self.runner.get_run_dict(testname), None)
73 calls = [mock.call(testname), mock.call(testname)]
74 args[0].assert_has_calls(calls)
75 calls = [mock.call("Cannot get %s's config options", testname),
76 mock.call("Cannot get %s's config options", testname)]
77 args[1].assert_has_calls(calls)
79 @mock.patch('functest.ci.run_tests.LOGGER.exception')
80 @mock.patch('functest.ci.run_tests.Runner.get_dict_by_test',
81 side_effect=Exception)
82 def test_get_run_dict_exception(self, *args):
83 testname = 'test_name'
84 self.assertEqual(self.runner.get_run_dict(testname), None)
85 args[1].assert_called_once_with(
86 "Cannot get %s's config options", testname)
88 def _test_source_envfile(self, msg, key='OS_TENANT_NAME', value='admin'):
91 except Exception: # pylint: disable=broad-except
94 with mock.patch('six.moves.builtins.open',
95 mock.mock_open(read_data=msg),
96 create=True) as mock_method:
97 mock_method.return_value.__iter__ = lambda self: iter(
99 self.runner.source_envfile(envfile)
100 mock_method.assert_called_once_with(envfile, 'r')
101 self.assertEqual(os.environ[key], value)
103 def test_source_envfile(self):
104 self._test_source_envfile('OS_TENANT_NAME=admin')
105 self._test_source_envfile('OS_TENANT_NAME= admin')
106 self._test_source_envfile('OS_TENANT_NAME = admin')
107 self._test_source_envfile('OS_TENANT_NAME = "admin"')
108 self._test_source_envfile('export OS_TENANT_NAME=admin')
109 self._test_source_envfile('export OS_TENANT_NAME =admin')
110 self._test_source_envfile('export OS_TENANT_NAME = admin')
111 self._test_source_envfile('export OS_TENANT_NAME = "admin"')
112 # This test will fail as soon as rc_file is fixed
113 self._test_source_envfile(
114 'export "\'OS_TENANT_NAME\'" = "\'admin\'"')
116 def test_get_dict_by_test(self):
117 with mock.patch('six.moves.builtins.open', mock.mock_open()), \
118 mock.patch('yaml.safe_load') as mock_yaml:
119 mock_obj = mock.Mock()
120 testcase_dict = {'case_name': 'testname',
122 attrs = {'get.return_value': [{'testcases': [testcase_dict]}]}
123 mock_obj.configure_mock(**attrs)
124 mock_yaml.return_value = mock_obj
125 self.assertDictEqual(
126 run_tests.Runner.get_dict_by_test('testname'),
129 @mock.patch('functest.ci.run_tests.Runner.get_run_dict',
131 def test_run_tests_import_exception(self, *args):
132 mock_test = mock.Mock()
133 kwargs = {'get_name.return_value': 'test_name',
134 'needs_clean.return_value': False}
135 mock_test.configure_mock(**kwargs)
136 with self.assertRaises(Exception) as context:
137 self.runner.run_test(mock_test)
138 args[0].assert_called_with('test_name')
139 msg = "Cannot import the class for the test case."
140 self.assertTrue(msg in str(context.exception))
142 @mock.patch('importlib.import_module', name="module",
143 return_value=mock.Mock(test_class=mock.Mock(
144 side_effect=FakeModule)))
145 @mock.patch('functest.ci.run_tests.Runner.get_dict_by_test')
146 def test_run_tests_default(self, *args):
147 mock_test = mock.Mock()
148 kwargs = {'get_name.return_value': 'test_name',
149 'needs_clean.return_value': True}
150 mock_test.configure_mock(**kwargs)
151 test_run_dict = {'module': 'test_module',
152 'class': 'test_class'}
153 with mock.patch('functest.ci.run_tests.Runner.get_run_dict',
154 return_value=test_run_dict):
155 self.runner.clean_flag = True
156 self.runner.run_test(mock_test)
157 args[0].assert_called_with('test_name')
158 args[1].assert_called_with('test_module')
159 self.assertEqual(self.runner.overall_result,
160 run_tests.Result.EX_OK)
162 @mock.patch('functest.ci.run_tests.Runner.run_test',
163 return_value=TestCase.EX_OK)
164 def test_run_tier_default(self, *mock_methods):
165 self.assertEqual(self.runner.run_tier(self.tier),
166 run_tests.Result.EX_OK)
167 mock_methods[0].assert_called_with(mock.ANY)
169 @mock.patch('functest.ci.run_tests.LOGGER.info')
170 def test_run_tier_missing_test(self, mock_logger_info):
171 self.tier.get_tests.return_value = None
172 self.assertEqual(self.runner.run_tier(self.tier),
173 run_tests.Result.EX_ERROR)
174 self.assertTrue(mock_logger_info.called)
176 @mock.patch('functest.ci.run_tests.LOGGER.info')
177 @mock.patch('functest.ci.run_tests.Runner.run_tier')
178 @mock.patch('functest.ci.run_tests.Runner.summary')
179 def test_run_all_default(self, *mock_methods):
180 CONST.__setattr__('CI_LOOP', 'test_ci_loop')
181 self.runner.run_all()
182 mock_methods[1].assert_not_called()
183 self.assertTrue(mock_methods[2].called)
185 @mock.patch('functest.ci.run_tests.LOGGER.info')
186 @mock.patch('functest.ci.run_tests.Runner.summary')
187 def test_run_all_missing_tier(self, *mock_methods):
188 CONST.__setattr__('CI_LOOP', 'loop_re_not_available')
189 self.runner.run_all()
190 self.assertTrue(mock_methods[1].called)
192 @mock.patch('functest.ci.run_tests.Runner.source_envfile',
193 side_effect=Exception)
194 @mock.patch('functest.ci.run_tests.Runner.summary')
195 def test_main_failed(self, *mock_methods):
196 kwargs = {'test': 'test_name', 'noclean': True, 'report': True}
197 args = {'get_tier.return_value': False,
198 'get_test.return_value': False}
199 self.runner.tiers = mock.Mock()
200 self.runner.tiers.configure_mock(**args)
201 self.assertEqual(self.runner.main(**kwargs),
202 run_tests.Result.EX_ERROR)
203 mock_methods[1].assert_called_once_with(
204 '/home/opnfv/functest/conf/env_file')
206 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
207 @mock.patch('functest.ci.run_tests.Runner.run_test',
208 return_value=TestCase.EX_OK)
209 @mock.patch('functest.ci.run_tests.Runner.summary')
210 def test_main_tier(self, *mock_methods):
211 mock_tier = mock.Mock()
212 test_mock = mock.Mock()
213 test_mock.get_name.return_value = 'test1'
214 args = {'get_name.return_value': 'tier_name',
215 'get_tests.return_value': [test_mock]}
216 mock_tier.configure_mock(**args)
217 kwargs = {'test': 'tier_name', 'noclean': True, 'report': True}
218 args = {'get_tier.return_value': mock_tier,
219 'get_test.return_value': None}
220 self.runner.tiers = mock.Mock()
221 self.runner.tiers.configure_mock(**args)
222 self.assertEqual(self.runner.main(**kwargs),
223 run_tests.Result.EX_OK)
224 mock_methods[1].assert_called()
226 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
227 @mock.patch('functest.ci.run_tests.Runner.run_test',
228 return_value=TestCase.EX_OK)
229 def test_main_test(self, *mock_methods):
230 kwargs = {'test': 'test_name', 'noclean': True, 'report': True}
231 args = {'get_tier.return_value': None,
232 'get_test.return_value': 'test_name'}
233 self.runner.tiers = mock.Mock()
234 mock_methods[1].return_value = self.creds
235 self.runner.tiers.configure_mock(**args)
236 self.assertEqual(self.runner.main(**kwargs),
237 run_tests.Result.EX_OK)
238 mock_methods[0].assert_called_once_with('test_name')
240 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
241 @mock.patch('functest.ci.run_tests.Runner.run_all')
242 @mock.patch('functest.ci.run_tests.Runner.summary')
243 def test_main_all_tier(self, *args):
244 kwargs = {'get_tier.return_value': None,
245 'get_test.return_value': None}
246 self.runner.tiers = mock.Mock()
247 self.runner.tiers.configure_mock(**kwargs)
249 self.runner.main(test='all', noclean=True, report=True),
250 run_tests.Result.EX_OK)
251 args[0].assert_called_once_with(None)
252 args[1].assert_called_once_with()
253 args[2].assert_called_once_with('/home/opnfv/functest/conf/env_file')
255 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
256 def test_main_any_tier_test_ko(self, *args):
257 kwargs = {'get_tier.return_value': None,
258 'get_test.return_value': None}
259 self.runner.tiers = mock.Mock()
260 self.runner.tiers.configure_mock(**kwargs)
262 self.runner.main(test='any', noclean=True, report=True),
263 run_tests.Result.EX_ERROR)
264 args[0].assert_called_once_with('/home/opnfv/functest/conf/env_file')
267 if __name__ == "__main__":
268 logging.disable(logging.CRITICAL)
269 unittest.main(verbosity=2)