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, {})
