def test_generate_evg_tasks_one_test_changed(self, create_tests_by_task_mock):
evg_conf_mock = get_evergreen_config()
create_tests_by_task_mock.return_value = {
"aggregation_mongos_passthrough": {
"display_task_name": "aggregation_mongos_passthrough",
"resmoke_args":
"--suites=aggregation_mongos_passthrough --storageEngine=wiredTiger",
"tests": ["jstests/aggregation/bugs/ifnull.js"],
"use_multiversion": None
}
} # yapf: disable
expansions_file_data = get_expansions_data()
buildvariant_map = {
"enterprise-rhel-62-64-bit-inmem": "enterprise-rhel-62-64-bit-inmem-required",
"enterprise-rhel-62-64-bit-majority-read-concern-off":
"enterprise-rhel-62-64-bit-majority-read-concern-off-required",
} # yapf: disable
shrub_config = ShrubProject.empty()
evergreen_api = MagicMock()
repo = MagicMock()
evergreen_api.test_stats_by_project.return_value = [
MagicMock(test_file="dir/test2.js", avg_duration_pass=10)
]
under_test._generate_evg_tasks(evergreen_api, shrub_config, expansions_file_data,
buildvariant_map, [repo], evg_conf_mock)
generated_config = shrub_config.as_dict()
self.assertEqual(len(generated_config["buildvariants"]), 2)
first_generated_build_variant = generated_config["buildvariants"][0]
self.assertIn(first_generated_build_variant["name"], buildvariant_map.values())
self.assertEqual(first_generated_build_variant["display_tasks"][0]["name"], "burn_in_tests")
self.assertEqual(
first_generated_build_variant["display_tasks"][0]["execution_tasks"][0],
f"burn_in:aggregation_mongos_passthrough_0_{first_generated_build_variant['name']}")
def test_generate_evg_tasks_one_test_changed(self,
create_tests_by_task_mock,
evergreen_mock):
evergreen_mock.parse_evergreen_file.return_value = get_evergreen_config(
)
create_tests_by_task_mock.return_value = {
"aggregation_mongers_passthrough": {
"resmoke_args":
"--suites=aggregation_mongers_passthrough --storageEngine=wiredTiger",
"tests": ["jstests/aggregation/bugs/ifnull.js"],
"use_multiversion": None
}
} # yapf: disable
expansions_file_data = get_expansions_data()
buildvariant_map = {
"enterprise-rhel-62-64-bit-inmem": "enterprise-rhel-62-64-bit-inmem-required",
"enterprise-rhel-62-64-bit-majority-read-concern-off":
"enterprise-rhel-62-64-bit-majority-read-concern-off-required",
} # yapf: disable
shrub_config = Configuration()
burn_in_tags._generate_evg_tasks(shrub_config, expansions_file_data,
buildvariant_map)
generated_config = shrub_config.to_map()
self.assertEqual(len(generated_config["buildvariants"]), 2)
first_generated_build_variant = generated_config["buildvariants"][0]
self.assertEqual(
first_generated_build_variant["display_tasks"][0]["name"],
"burn_in_tests")
self.assertEqual(
first_generated_build_variant["display_tasks"][0]
["execution_tasks"][0],
"burn_in:enterprise-rhel-62-64-bit-inmem-required_aggregation_mongers_passthrough_0"
)
def test_generate_evg_tasks_no_tests_changed(self, create_tests_by_task_mock, evergreen_mock):
evergreen_mock.parse_evergreen_file.return_value = get_evergreen_config()
create_tests_by_task_mock.return_value = {}
expansions_file_data = get_expansions_data()
buildvariant_map = {
"enterprise-rhel-62-64-bit-inmem": "enterprise-rhel-62-64-bit-inmem-required",
"enterprise-rhel-62-64-bit-majority-read-concern-off":
"enterprise-rhel-62-64-bit-majority-read-concern-off-required",
} # yapf: disable
shrub_config = Configuration()
evergreen_api = Mock()
burn_in_tags._generate_evg_tasks(evergreen_api, shrub_config, expansions_file_data,
buildvariant_map)
self.assertEqual(shrub_config.to_map(), {})
def test_generate_evg_tasks_no_tests_changed(self, create_tests_by_task_mock):
evg_conf_mock = get_evergreen_config()
create_tests_by_task_mock.return_value = {}
expansions_file_data = get_expansions_data()
buildvariant_map = {
"enterprise-rhel-62-64-bit-inmem": "enterprise-rhel-62-64-bit-inmem-required",
"enterprise-rhel-62-64-bit-majority-read-concern-off":
"enterprise-rhel-62-64-bit-majority-read-concern-off-required",
} # yapf: disable
shrub_config = ShrubProject()
evergreen_api = MagicMock()
repo = MagicMock()
under_test._generate_evg_tasks(evergreen_api, shrub_config, expansions_file_data,
buildvariant_map, [repo], evg_conf_mock)
self.assertEqual(shrub_config.as_dict(), EMPTY_PROJECT)
