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