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