def test_get_config_for_task_mapping(self, get_evg_task_config_mock): tasks = ["task_1", "task_2"] 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_task_mappings({}, tasks, MagicMock()) self.assertEqual(task_configs["task_1"]["task_config_key"], "task_config_value_1") self.assertEqual(task_configs["task_2"]["task_config_key"], "task_config_value_2")
def test_get_config_for_task_mapping_when_task_does_not_exist(self, find_task_mock, get_evg_task_config_mock): find_task_mock.return_value = None tasks = ["task_1"] get_evg_task_config_mock.return_value = {"task_config_key": "task_config_value_1"} task_configs = under_test._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, get_evg_task_config_mock): find_task_mock.return_value = _evergreen.Task({"name": "task_1"}) exclude_task_mock.return_value = True tasks = ["task_1"] get_evg_task_config_mock.return_value = {"task_config_key": "task_config_value_1"} task_configs = under_test._get_task_configs_for_task_mappings({}, tasks, MagicMock()) self.assertEqual(task_configs, {})
def test_get_config_for_task_mapping(self, find_task_mock, exclude_task_mock, get_evg_task_config_mock): find_task_mock.side_effect = [ _evergreen.Task({"name": "task_1"}), _evergreen.Task({"name": "task_2"}) ] exclude_task_mock.return_value = False tasks = ["task_1", "task_2"] 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_task_mappings({}, tasks, MagicMock()) self.assertEqual(task_configs["task_1"]["task_config_key"], "task_config_value_1") self.assertEqual(task_configs["task_2"]["task_config_key"], "task_config_value_2")
def test_get_config_for_task_mapping(self, get_evg_task_config_mock): tasks = ["task_1", "task_2"] 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_task_mappings( expansion_file=MagicMock(), related_tasks=tasks, build_variant_config=MagicMock()) self.assertEqual(task_configs["task_1"]["task_config_key"], "task_config_value_1") self.assertEqual(task_configs["task_2"]["task_config_key"], "task_config_value_2")