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)