def test_n_task_m_test(self, mock_hash):
mock_hash.return_value = MONGO_4_2_HASH
n_tasks = 2
n_tests = 3
target_file = "target_file.json"
mock_evg_api = MagicMock()
split_config = build_mock_split_task_config()
configure_dependencies(mock_evg_api, split_config)
tests_by_task = create_multiversion_tests_by_task_mock(
n_tasks, n_tests)
orchestrator = under_test.MultiversionBurnInOrchestrator()
generated_config = orchestrator.generate_configuration(
tests_by_task, target_file, "build_variant")
evg_config = [
config for config in generated_config.file_list
if config.file_name == target_file
]
self.assertEqual(1, len(evg_config))
evg_config = evg_config[0]
evg_config_dict = json.loads(evg_config.content)
tasks = evg_config_dict["tasks"]
self.assertEqual(len(tasks),
(NUM_REPL_MIXED_VERSION_CONFIGS +
NUM_SHARDED_MIXED_VERSION_CONFIGS) * n_tests)
def tests_no_tasks_given(self):
target_file = "target_file.json"
mock_evg_api = MagicMock()
split_config = build_mock_split_task_config()
configure_dependencies(mock_evg_api, split_config)
orchestrator = under_test.MultiversionBurnInOrchestrator()
generated_config = orchestrator.generate_configuration({}, target_file,
"build_variant")
evg_config = [
config for config in generated_config.file_list
if config.file_name == target_file
]
self.assertEqual(1, len(evg_config))
evg_config = evg_config[0]
evg_config_dict = json.loads(evg_config.content)
self.assertEqual(0, len(evg_config_dict["tasks"]))
