def test_cap_on_task_generate(self, validate_mock, _, exit_mock):
evg_api = MagicMock()
gen_config = MagicMock(use_multiversion=False)
repeat_config = MagicMock()
tests_by_task = MagicMock()
validate_mock.return_value = False
exit_mock.side_effect = ValueError("exiting")
with self.assertRaises(ValueError):
under_test.create_generate_tasks_file(tests_by_task, gen_config, repeat_config, evg_api)
exit_mock.assert_called_once()
def test_gen_tasks_multiversion_configuration_is_returned(
self, gen_tasks_config_mock): # pylint: disable=invalid-name
evg_api = MagicMock()
gen_config = MagicMock(run_build_variant="variant",
project="project",
build_variant="build_variant",
task_id="task_id",
use_multiversion=True)
repeat_config = MagicMock()
tests_by_task = MagicMock()
evg_config = MagicMock()
evg_config.to_map.return_value = {
'buildvariants': [
{
'name': 'build_variant',
'display_tasks': [
{
'name': 'burn_in_tests_multiversion',
'execution_tasks': [
'burn_in_tests_multiversion_gen'
]
}
]
}
]
} # yapf: disable
gen_tasks_config_mock.return_value = evg_config
config = under_test.create_generate_tasks_file(tests_by_task,
gen_config,
repeat_config, evg_api)
self.assertEqual(config, evg_config.to_map.return_value)
def test_cap_on_task_generate(self, gen_tasks_config_mock, exit_mock):
evg_api = MagicMock()
gen_config = MagicMock()
repeat_config = MagicMock()
tests_by_task = MagicMock()
task_list = [f"task_{i}" for i in range(1005)]
evg_config = MagicMock()
evg_config.to_map.return_value = {
"tasks": task_list,
}
gen_tasks_config_mock.return_value = evg_config
exit_mock.side_effect = ValueError("exiting")
with self.assertRaises(ValueError):
under_test.create_generate_tasks_file(tests_by_task, gen_config, repeat_config, evg_api)
exit_mock.assert_called_once()
def test_gen_tasks_configuration_is_returned(self, gen_tasks_config_mock):
evg_api = MagicMock()
gen_config = MagicMock()
repeat_config = MagicMock()
tests_by_task = MagicMock()
task_list = [f"task_{i}" for i in range(10)]
evg_config = MagicMock()
evg_config.to_map.return_value = {
"tasks": task_list,
}
gen_tasks_config_mock.return_value = evg_config
config = under_test.create_generate_tasks_file(tests_by_task, gen_config, repeat_config,
evg_api)
self.assertEqual(config, evg_config.to_map.return_value)
def test_gen_tasks_configuration_is_returned(self, gen_tasks_config_mock):
evg_api = MagicMock()
gen_config = MagicMock(use_multiversion=False)
repeat_config = MagicMock()
tests_by_task = MagicMock()
task_list = [f"task_{i}" for i in range(10)]
evg_config = MagicMock()
evg_config.to_map.return_value = {
"tasks": task_list,
}
gen_tasks_config_mock.return_value = evg_config
evg_project_config = get_evergreen_config("etc/evergreen.yml")
config = under_test.create_generate_tasks_file(tests_by_task, gen_config, repeat_config,
evg_api, evg_project_config)
self.assertEqual(config, evg_config.to_map.return_value)
