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.core.testcase import TestCase
20 class FakeModule(TestCase):
22 def run(self, **kwargs):
26 class RunTestsTesting(unittest.TestCase):
29 self.runner = run_tests.Runner()
30 mock_test_case = mock.Mock()
31 mock_test_case.is_successful.return_value = TestCase.EX_OK
32 self.runner.executed_test_cases['test1'] = mock_test_case
33 self.runner.executed_test_cases['test2'] = mock_test_case
35 self.creds = {'OS_AUTH_URL': 'http://test_ip:test_port/v2.0',
36 'OS_USERNAME': 'test_os_username',
37 'OS_TENANT_NAME': 'test_tenant',
38 'OS_PASSWORD': 'test_password'}
39 self.test = {'test_name': 'test_name'}
40 self.tier = mock.Mock()
42 test1.get_name.return_value = 'test1'
44 test2.get_name.return_value = 'test2'
45 attrs = {'get_name.return_value': 'test_tier',
46 'get_tests.return_value': [test1, test2],
47 'get_ci_loop.return_value': 'test_ci_loop',
48 'get_test_names.return_value': ['test1', 'test2']}
49 self.tier.configure_mock(**attrs)
51 self.tiers = mock.Mock()
52 attrs = {'get_tiers.return_value': [self.tier]}
53 self.tiers.configure_mock(**attrs)
55 self.run_tests_parser = run_tests.RunTestsParser()
57 @mock.patch('functest.ci.run_tests.Runner.get_dict_by_test')
58 def test_get_run_dict(self, *args):
59 retval = {'run': mock.Mock()}
60 args[0].return_value = retval
61 self.assertEqual(self.runner.get_run_dict('test_name'), retval['run'])
62 args[0].assert_called_once_with('test_name')
64 @mock.patch('functest.ci.run_tests.LOGGER.error')
65 @mock.patch('functest.ci.run_tests.Runner.get_dict_by_test',
67 def test_get_run_dict_config_ko(self, *args):
68 testname = 'test_name'
69 self.assertEqual(self.runner.get_run_dict(testname), None)
70 args[0].return_value = {}
71 self.assertEqual(self.runner.get_run_dict(testname), None)
72 calls = [mock.call(testname), mock.call(testname)]
73 args[0].assert_has_calls(calls)
74 calls = [mock.call("Cannot get %s's config options", testname),
75 mock.call("Cannot get %s's config options", testname)]
76 args[1].assert_has_calls(calls)
78 @mock.patch('functest.ci.run_tests.LOGGER.exception')
79 @mock.patch('functest.ci.run_tests.Runner.get_dict_by_test',
80 side_effect=Exception)
81 def test_get_run_dict_exception(self, *args):
82 testname = 'test_name'
83 self.assertEqual(self.runner.get_run_dict(testname), None)
84 args[1].assert_called_once_with(
85 "Cannot get %s's config options", testname)
87 def _test_source_envfile(self, msg, key='OS_TENANT_NAME', value='admin'):
90 except Exception: # pylint: disable=broad-except
93 with mock.patch('six.moves.builtins.open',
94 mock.mock_open(read_data=msg),
95 create=True) as mock_method,\
96 mock.patch('os.path.isfile', return_value=True):
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 os.environ['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 os.environ['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()
205 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
206 @mock.patch('functest.ci.run_tests.Runner.run_test',
207 return_value=TestCase.EX_OK)
208 @mock.patch('functest.ci.run_tests.Runner.summary')
209 def test_main_tier(self, *mock_methods):
210 mock_tier = mock.Mock()
211 test_mock = mock.Mock()
212 test_mock.get_name.return_value = 'test1'
213 args = {'get_name.return_value': 'tier_name',
214 'get_tests.return_value': [test_mock]}
215 mock_tier.configure_mock(**args)
216 kwargs = {'test': 'tier_name', 'noclean': True, 'report': True}
217 args = {'get_tier.return_value': mock_tier,
218 'get_test.return_value': None}
219 self.runner.tiers = mock.Mock()
220 self.runner.tiers.configure_mock(**args)
221 self.assertEqual(self.runner.main(**kwargs),
222 run_tests.Result.EX_OK)
223 mock_methods[1].assert_called()
225 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
226 @mock.patch('functest.ci.run_tests.Runner.run_test',
227 return_value=TestCase.EX_OK)
228 def test_main_test(self, *mock_methods):
229 kwargs = {'test': 'test_name', 'noclean': True, 'report': True}
230 args = {'get_tier.return_value': None,
231 'get_test.return_value': 'test_name'}
232 self.runner.tiers = mock.Mock()
233 mock_methods[1].return_value = self.creds
234 self.runner.tiers.configure_mock(**args)
235 self.assertEqual(self.runner.main(**kwargs),
236 run_tests.Result.EX_OK)
237 mock_methods[0].assert_called_once_with('test_name')
239 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
240 @mock.patch('functest.ci.run_tests.Runner.run_all')
241 @mock.patch('functest.ci.run_tests.Runner.summary')
242 def test_main_all_tier(self, *args):
243 kwargs = {'get_tier.return_value': None,
244 'get_test.return_value': None}
245 self.runner.tiers = mock.Mock()
246 self.runner.tiers.configure_mock(**kwargs)
248 self.runner.main(test='all', noclean=True, report=True),
249 run_tests.Result.EX_OK)
250 args[0].assert_called_once_with(None)
251 args[1].assert_called_once_with()
252 args[2].assert_called_once_with()
254 @mock.patch('functest.ci.run_tests.Runner.source_envfile')
255 def test_main_any_tier_test_ko(self, *args):
256 kwargs = {'get_tier.return_value': None,
257 'get_test.return_value': None}
258 self.runner.tiers = mock.Mock()
259 self.runner.tiers.configure_mock(**kwargs)
261 self.runner.main(test='any', noclean=True, report=True),
262 run_tests.Result.EX_ERROR)
263 args[0].assert_called_once_with()
266 if __name__ == "__main__":
267 logging.disable(logging.CRITICAL)
268 unittest.main(verbosity=2)