def test_no_tasks_given(self):
build_variant = BuildVariant("build variant")
gen_config = MagicMock(run_build_variant="variant")
repeat_config = MagicMock()
under_test.create_generate_tasks_config(build_variant, {}, gen_config, repeat_config, None)
evg_config_dict = build_variant.as_dict()
self.assertEqual(0, len(evg_config_dict["tasks"]))
def test_fuzzer_for_large_distro_tasks_should_be_generated_on_large(self):
mock_distro = "my large distro"
mock_params = build_mock_fuzzer_params(use_large_distro=True, large_distro_name=mock_distro)
build_variant = BuildVariant("mock build variant")
service = build_mocked_service()
service.gen_task_options.large_distro_name = mock_distro
service.generate_fuzzer_task(mock_params, build_variant)
fuzzer_config = build_variant.as_dict()
self.assertTrue(all(mock_distro in task["distros"] for task in fuzzer_config["tasks"]))
def test_no_tasks_given(self):
build_variant = BuildVariant("build variant")
gen_config = MagicMock(run_build_variant="variant")
repeat_config = MagicMock()
mock_evg_api = MagicMock()
executor = under_test.GenerateBurnInExecutor(gen_config, repeat_config, mock_evg_api)
executor.generate_tasks_for_variant({}, build_variant)
evg_config_dict = build_variant.as_dict()
self.assertEqual(0, len(evg_config_dict["tasks"]))
def test_evg_config_does_not_fails_if_timeout_too_high_on_mainline(self):
options = self.generate_mock_options()
options.is_patch = False
options.repeat_suites = under_test.MAX_EXPECTED_TIMEOUT
suites = self.generate_mock_suites(3)
build_variant = BuildVariant("variant")
generator = under_test.EvergreenConfigGenerator(suites, options, MagicMock())
generator.generate_config(build_variant)
config = build_variant.as_dict()
self.assertEqual(len(config["tasks"]), len(suites) + 1)
def test_n_task_m_test(self):
n_tasks = 3
n_tests = 5
build_variant = BuildVariant("build variant")
gen_config = MagicMock(run_build_variant="variant", distro=None)
repeat_config = MagicMock()
tests_by_task = create_tests_by_task_mock(n_tasks, n_tests)
under_test.create_generate_tasks_config(build_variant, tests_by_task, gen_config,
repeat_config, None)
evg_config_dict = build_variant.as_dict()
self.assertEqual(n_tasks * n_tests, len(evg_config_dict["tasks"]))
def test_n_task_m_test(self):
n_tasks = 3
n_tests = 5
build_variant = BuildVariant("build variant")
gen_config = MagicMock(run_build_variant="variant", distro=None)
repeat_config = MagicMock()
tests_by_task = create_tests_by_task_mock(n_tasks, n_tests)
mock_evg_api = MagicMock()
executor = under_test.GenerateBurnInExecutor(gen_config, repeat_config, mock_evg_api)
executor.generate_tasks_for_variant(tests_by_task, build_variant)
evg_config_dict = build_variant.as_dict()
self.assertEqual(n_tasks * n_tests, len(evg_config_dict["tasks"]))
