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"])