def test_get_config_for_test_mapping(self, get_evg_task_config_mock): tests_by_task = tests_by_task_stub() get_evg_task_config_mock.side_effect = [{ "task_config_key": "task_config_value_1" }, { "task_config_key": "task_config_value_2" }] task_configs = under_test._get_task_configs_for_test_mappings( expansion_file=MagicMock(), tests_by_task=tests_by_task, build_variant_config=MagicMock()) self.assertEqual(task_configs["jsCore_auth"]["task_config_key"], "task_config_value_1") self.assertEqual( task_configs["jsCore_auth"]["selected_tests_to_run"], { "jstests/core/currentop_waiting_for_latch.js", "jstests/core/latch_analyzer.js" }) self.assertEqual(task_configs["auth_gen"]["task_config_key"], "task_config_value_2") self.assertEqual(task_configs["auth_gen"]["selected_tests_to_run"], {'jstests/auth/auth3.js'})
def test_get_config_for_test_mapping(self, find_task_mock, exclude_task_mock, get_evg_task_config_mock): find_task_mock.side_effect = [ _evergreen.Task({"name": "jsCore_auth"}), _evergreen.Task({"name": "auth_gen"}) ] exclude_task_mock.return_value = False tests_by_task = { "jsCore_auth": { "tests": [ "jstests/core/currentop_waiting_for_latch.js", "jstests/core/latch_analyzer.js", ], }, "auth_gen": {"tests": ["jstests/auth/auth3.js"], }, } get_evg_task_config_mock.side_effect = [{"task_config_key": "task_config_value_1"}, {"task_config_key": "task_config_value_2"}] task_configs = under_test._get_task_configs_for_test_mappings({}, tests_by_task, MagicMock()) self.assertEqual(task_configs["jsCore_auth"]["task_config_key"], "task_config_value_1") self.assertEqual( task_configs["jsCore_auth"]["selected_tests_to_run"], {"jstests/core/currentop_waiting_for_latch.js", "jstests/core/latch_analyzer.js"}) self.assertEqual(task_configs["auth_gen"]["task_config_key"], "task_config_value_2") self.assertEqual(task_configs["auth_gen"]["selected_tests_to_run"], {'jstests/auth/auth3.js'})
def test_get_config_for_test_mapping(self, find_task_mock, exclude_task_mock, get_evg_task_config_mock): find_task_mock.side_effect = [ _evergreen.Task({"name": "jsCore_auth"}), _evergreen.Task({"name": "auth_gen"}) ] exclude_task_mock.return_value = False tests_by_task = { "jsCore_auth": TaskInfo( display_task_name="task 1", tests=[ "jstests/core/currentop_waiting_for_latch.js", "jstests/core/latch_analyzer.js", ], resmoke_args="", use_multiversion=None, distro="", ), "auth_gen": TaskInfo( display_task_name="task 2", tests=["jstests/auth/auth3.js"], resmoke_args="", use_multiversion=None, distro="", ), } get_evg_task_config_mock.side_effect = [{ "task_config_key": "task_config_value_1" }, { "task_config_key": "task_config_value_2" }] task_configs = under_test._get_task_configs_for_test_mappings( {}, tests_by_task, MagicMock()) self.assertEqual(task_configs["jsCore_auth"]["task_config_key"], "task_config_value_1") self.assertEqual( task_configs["jsCore_auth"]["selected_tests_to_run"], { "jstests/core/currentop_waiting_for_latch.js", "jstests/core/latch_analyzer.js" }) self.assertEqual(task_configs["auth_gen"]["task_config_key"], "task_config_value_2") self.assertEqual(task_configs["auth_gen"]["selected_tests_to_run"], {'jstests/auth/auth3.js'})
def test_get_config_for_test_mapping_when_task_does_not_exist(self, find_task_mock, get_evg_task_config_mock): find_task_mock.return_value = None tests_by_task = { "jsCore_auth": { "tests": [ "jstests/core/currentop_waiting_for_latch.js", "jstests/core/latch_analyzer.js", ], }, } get_evg_task_config_mock.return_value = {"task_config_key": "task_config_value_1"} task_configs = under_test._get_task_configs_for_test_mappings({}, tests_by_task, MagicMock()) self.assertEqual(task_configs, {})
def test_get_config_for_test_mapping_when_task_should_be_excluded( self, find_task_mock, exclude_task_mock, get_evg_task_config_mock): find_task_mock.return_value = _evergreen.Task({"name": "jsCore_auth"}) exclude_task_mock.return_value = True tests_by_task = { "jsCore_auth": { "tests": [ "jstests/core/currentop_waiting_for_latch.js", "jstests/core/latch_analyzer.js", ], }, } get_evg_task_config_mock.return_value = {"task_config_key": "task_config_value_1"} task_configs = under_test._get_task_configs_for_test_mappings({}, tests_by_task, MagicMock()) self.assertEqual(task_configs, {})