def test_get_config_for_test_mapping_when_task_should_be_excluded( self, find_task_mock, exclude_task_mock): find_task_mock.return_value = build_mock_evg_task( "jsCore_auth", "run tests", "--suites=core_auth --storageEngine=wiredTiger") exclude_task_mock.return_value = True 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="", require_multiversion=None, distro="", ), } task_config_service = under_test.TaskConfigService() task_configs = task_config_service.get_task_configs_for_test_mappings( tests_by_task, MagicMock()) self.assertEqual(task_configs, {})
def test_get_config_for_task_mapping_when_task_does_not_exist( self, find_task_mock): find_task_mock.return_value = None tasks = ["task_1"] task_config_service = under_test.TaskConfigService() task_configs = task_config_service.get_task_configs_for_task_mappings( tasks, MagicMock()) self.assertEqual(task_configs, {})
def test_get_config_for_task_mapping_when_task_should_be_excluded( self, find_task_mock, exclude_task_mock): find_task_mock.return_value = build_mock_evg_task("task_1") exclude_task_mock.return_value = True tasks = ["task_1"] task_config_service = under_test.TaskConfigService() task_configs = task_config_service.get_task_configs_for_task_mappings( tasks, MagicMock()) self.assertEqual(task_configs, {})
def test_get_config_for_test_mapping(self, find_task_mock, exclude_task_mock): find_task_mock.side_effect = [ build_mock_evg_task("jsCore_auth", "run tests"), build_mock_evg_task( "auth_gen", "run tests", "--suites=core_auth --storageEngine=wiredTiger"), ] 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="", require_multiversion_setup=False, distro="", suite="core_auth", build_variant="dummy_variant", ), "auth_gen": TaskInfo( display_task_name="task 2", tests=["jstests/auth/auth3.js"], resmoke_args="", require_multiversion_setup=False, distro="", suite="auth", build_variant="dummy_variant", ), } task_config_service = under_test.TaskConfigService() task_configs = task_config_service.get_task_configs_for_test_mappings( tests_by_task, MagicMock()) self.assertEqual(task_configs["jsCore_auth"]["resmoke_args"], "--storageEngine=wiredTiger") 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"]["suite"], "auth_gen") self.assertEqual(task_configs["auth_gen"]["selected_tests_to_run"], {'jstests/auth/auth3.js'})
def test_task_is_a_generate_resmoke_task(self): build_variant_conf = MagicMock() build_variant_conf.name = "variant" task = build_mock_evg_task("auth_gen") task_config_service = under_test.TaskConfigService() evg_task_config = task_config_service.get_evg_task_config( task, build_variant_conf) self.assertEqual(evg_task_config["task_name"], "auth") self.assertEqual(evg_task_config["build_variant"], "variant") self.assertIsNone(evg_task_config.get("suite")) self.assertEqual( evg_task_config["resmoke_args"], "--storageEngine=wiredTiger", )
def test_get_config_for_task_mapping(self, find_task_mock, exclude_task_mock): find_task_mock.side_effect = [ build_mock_evg_task("task_1"), build_mock_evg_task("task_2") ] exclude_task_mock.return_value = False tasks = ["task_1", "task_2"] task_config_service = under_test.TaskConfigService() task_configs = task_config_service.get_task_configs_for_task_mappings( tasks, MagicMock()) self.assertEqual(task_configs["task_1"]["resmoke_args"], "--storageEngine=wiredTiger") self.assertEqual(task_configs["task_2"]["resmoke_args"], "--storageEngine=wiredTiger")
def test_task_is_not_a_generate_resmoke_task(self): build_variant_conf = MagicMock() build_variant_conf.name = "variant" task = build_mock_evg_task( "jsCore_auth", "run tests", "--suites=core_auth --storageEngine=wiredTiger") task_config_service = under_test.TaskConfigService() evg_task_config = task_config_service.get_evg_task_config( task, build_variant_conf) self.assertEqual(evg_task_config["task_name"], "jsCore_auth") self.assertEqual(evg_task_config["build_variant"], "variant") self.assertEqual(evg_task_config["suite"], "core_auth") self.assertEqual( evg_task_config["resmoke_args"], "--storageEngine=wiredTiger", )
def test_get_config_for_test_mapping_when_task_does_not_exist( self, find_task_mock): find_task_mock.return_value = None 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="", require_multiversion=None, distro="", ), } task_config_service = under_test.TaskConfigService() task_configs = task_config_service.get_task_configs_for_test_mappings( tests_by_task, MagicMock()) self.assertEqual(task_configs, {})