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