DEVPROD-17817 add dependencies for multiversion binary selection on burn-in task references

CHANGELOG.md

@@ -1,4 +1,7 @@
 # Changelog
+## 3.0.1 - 2025-05-22
+* Add multiversion binary selection dependency to burn-in tasks when needed.
+
 ## 3.0.0 - 2025-05-07
 * Generate multiversion binary selection tasks
 

Cargo.toml

@@ -2,7 +2,7 @@
 name = "mongo-task-generator"
 description = "Dynamically split evergreen tasks into subtasks for testing the 10gen/mongo project."
 license = "Apache-2.0"
-version = "3.0.0"
+version = "3.0.1"
 repository = "https://github.com/mongodb/mongo-task-generator"
 authors = ["DevProd Correctness Team <devprod-correctness-team@mongodb.com>"]
 edition = "2018"

src/task_types/burn_in_tests.rs

@@ -11,8 +11,8 @@ use std::{
 
 use crate::evergreen::evg_config_utils::EvgConfigUtils;
 use crate::evergreen_names::{
-    BURN_IN_TASKS, BURN_IN_TASK_NAME, COMPILE_VARIANT, VERSION_BURN_IN_GEN_TASK,
-    VERSION_GEN_VARIANT,
+    BURN_IN_TASKS, BURN_IN_TASK_NAME, COMPILE_VARIANT, MULTIVERSION_BINARY_SELECTION,
+    VERSION_BURN_IN_GEN_TASK, VERSION_GEN_VARIANT,
 };
 use crate::{
     evergreen_names::BURN_IN_BYPASS,
@@ -451,6 +451,48 @@ impl BurnInService for BurnInServiceImpl {
             },
         ];
 
+        let mut includes_multiversion_tasks = false;
+        for task_ref in gen_config.gen_task_specs.iter_mut() {
+            // Find the corresponding subtask to check its dependencies
+            let depends_on_multiversion = generated_task
+                .sub_tasks()
+                .iter()
+                .find(|&t| t.evg_task.name == task_ref.name)
+                .map(|subtask| {
+                    subtask.evg_task.depends_on.as_ref().map_or(false, |deps| {
+                        deps.iter()
+                            .any(|dep| dep.name == MULTIVERSION_BINARY_SELECTION)
+                    })
+                })
+                .unwrap_or(false);
+
+            if depends_on_multiversion {
+                includes_multiversion_tasks = true;
+
+                // Combine the multiversion binary selection with variant task dependencies
+                let multiversion_dependency = TaskDependency {
+                    name: MULTIVERSION_BINARY_SELECTION.to_string(),
+                    variant: None,
+                };
+
+                task_ref.depends_on = Some(
+                    std::iter::once(multiversion_dependency)
+                        .chain(variant_task_dependencies.iter().cloned())
+                        .collect(),
+                );
+            }
+        }
+
+        // Add the multiversion binary selection task if needed
+        if includes_multiversion_tasks {
+            gen_config.gen_task_specs.push(TaskRef {
+                name: MULTIVERSION_BINARY_SELECTION.to_string(),
+                distros: None,
+                activate: Some(false),
+                depends_on: None,
+            });
+        }
+
         Ok(BuildVariant {
             name: gen_config.build_variant_name.clone(),
             tasks: gen_config.gen_task_specs.clone(),