def test_generated_task_large_distro(self):
suite_name = "suite_1"
distro_name = "distro_1"
large_distro_name = "large_distro_1"
variant = "build_variant"
evg_conf_mock = MagicMock()
task_def_mock = evg_conf_mock.get_task.return_value
task_def_mock.is_generate_resmoke_task = True
task_def_mock.generate_resmoke_tasks_command = {"vars": {"use_large_distro": True}}
evg_conf_mock.get_variant.return_value.raw = {
"expansions": {
"large_distro_name": large_distro_name
}
} # yapf: disable
task_mock = create_variant_task_mock("task 1", suite_name, distro_name)
test_list = [f"test{i}.js" for i in range(3)]
tests_by_suite = {
suite_name: test_list,
"suite 2": [f"test{i}.js" for i in range(1)],
"suite 3": [f"test{i}.js" for i in range(2)],
}
task_info = under_test._gather_task_info(task_mock, tests_by_suite, evg_conf_mock, variant)
self.assertIn(suite_name, task_info["resmoke_args"])
for test in test_list:
self.assertIn(test, task_info["tests"])
self.assertIsNone(task_info["use_multiversion"])
self.assertEqual(large_distro_name, task_info["distro"])
def test_non_generated_task(self):
suite_name = "suite_1"
distro_name = "distro_1"
variant = "build_variant"
evg_conf_mock = MagicMock()
evg_conf_mock.get_task.return_value.is_generate_resmoke_task = False
task_mock = create_variant_task_mock("task 1", suite_name, distro_name)
test_list = [f"test{i}.js" for i in range(3)]
tests_by_suite = {
suite_name: test_list,
"suite 2": [f"test{i}.js" for i in range(1)],
"suite 3": [f"test{i}.js" for i in range(2)],
}
task_info = under_test._gather_task_info(task_mock, tests_by_suite, evg_conf_mock, variant)
self.assertIn(suite_name, task_info["resmoke_args"])
for test in test_list:
self.assertIn(test, task_info["tests"])
self.assertIsNone(task_info["use_multiversion"])
self.assertEqual(distro_name, task_info["distro"])
