DEVPROD-18487 Respect use_large_distro and use_xlarge_distro when generating jstestfuzz tasks (#106)

Author: sleaux

CHANGELOG.md

@@ -1,4 +1,7 @@
 # Changelog
+## 3.4.1 - 2025-06-23
+* Respect `use_large_distro` and `use_xlarge_distro` when generating jstestfuzz tasks
+
 ## 3.4.0 - 2025-06-12
 * Allow fuzzer tasks to use the `no_multiversion_generate_tasks` tag
 

Cargo.lock

@@ -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.4.0"
+version = "3.4.1"
 repository = "https://github.com/mongodb/mongo-task-generator"
 authors = ["DevProd Correctness Team <[email protected]>"]
 edition = "2018"

Cargo.toml

@@ -198,6 +198,16 @@ impl ConfigExtractionService for ConfigExtractionServiceImpl {
             task_name,
             variant: build_variant.name.to_string(),
             suite,
+            use_large_distro: self.evg_config_utils.lookup_default_param_bool(
+                task_def,
+                USE_LARGE_DISTRO,
+                false,
+            )?,
+            use_xlarge_distro: self.evg_config_utils.lookup_default_param_bool(
+                task_def,
+                USE_XLARGE_DISTRO,
+                false,
+            )?,
             num_files,
             num_tasks: evg_config_utils.lookup_required_param_u64(task_def, NUM_FUZZER_TASKS)?,
             resmoke_args: evg_config_utils.lookup_required_param_str(task_def, RESMOKE_ARGS)?,

src/services/config_extraction.rs

@@ -38,6 +38,10 @@ pub struct FuzzerGenTaskParams {
     pub suite: String,
     /// The bazel test target, if it is a bazel-based resmoke task.
     pub bazel_target: Option<String>,
+    /// Should the generated tasks run on a 'large' distro.
+    pub use_large_distro: bool,
+    /// Should the generated tasks run on a 'xlarge' distro.
+    pub use_xlarge_distro: bool,
     /// Number of javascript files fuzzer should generate.
     pub num_files: String,
     /// Number of sub-tasks fuzzer should generate.
@@ -165,12 +169,16 @@ impl FuzzerGenTaskParams {
 }
 
 /// A Generated Fuzzer task.
-#[derive(Debug)]
+#[derive(Debug, Default)]
 pub struct FuzzerTask {
     /// Name for generated task.
     pub task_name: String,
     /// Sub-tasks comprising generated task.
     pub sub_tasks: Vec<EvgTask>,
+    /// Should the generated tasks run on a 'large' distro.
+    pub use_large_distro: bool,
+    /// Should the generated tasks run on a 'xlarge' distro.
+    pub use_xlarge_distro: bool,
 }
 
 impl GeneratedSuite for FuzzerTask {
@@ -186,8 +194,8 @@ impl GeneratedSuite for FuzzerTask {
             .into_iter()
             .map(|sub_task| GeneratedSubTask {
                 evg_task: sub_task,
-                use_large_distro: false,
-                use_xlarge_distro: false,
+                use_large_distro: self.use_large_distro,
+                use_xlarge_distro: self.use_xlarge_distro,
             })
             .collect()
     }
@@ -256,6 +264,8 @@ impl GenFuzzerService for GenFuzzerServiceImpl {
         Ok(Box::new(FuzzerTask {
             task_name: params.task_name.to_string(),
             sub_tasks,
+            use_large_distro: params.use_large_distro,
+            use_xlarge_distro: params.use_xlarge_distro,
         }))
     }
 }
@@ -443,11 +453,24 @@ mod tests {
         let fuzzer_task = FuzzerTask {
             task_name: "my fuzzer".to_string(),
             sub_tasks: vec![],
+            ..Default::default()
         };
 
         assert_eq!(fuzzer_task.display_name(), "my fuzzer".to_string());
     }
 
+    #[test]
+    fn test_large_distro() {
+        let fuzzer_task = FuzzerTask {
+            task_name: "my fuzzer".to_string(),
+            sub_tasks: vec![],
+            use_large_distro: true,
+            ..Default::default()
+        };
+
+        assert!(fuzzer_task.use_large_distro);
+    }
+
     #[test]
     fn test_sub_tasks() {
         let fuzzer_task = FuzzerTask {
@@ -460,6 +483,7 @@ mod tests {
                     ..Default::default()
                 },
             ],
+            ..Default::default()
         };
 
         assert_eq!(fuzzer_task.sub_tasks().len(), 2);
@@ -477,6 +501,7 @@ mod tests {
                     ..Default::default()
                 },
             ],
+            ..Default::default()
         };
 
         let task_refs = fuzzer_task.build_task_ref(Some("distro".to_string()), None);

src/task_types/fuzzer_tasks.rs

@@ -3303,6 +3303,7 @@ tasks:
       npm_command: initsync-fuzzer
       suite: initial_sync_fuzzer
       resmoke_args: "--mongodSetParameters='{logComponentVerbosity: {command: 2}}'"
+      use_large_distro: "true"
 
 ## initial sync generational fuzzer ##
 - <<: *jstestfuzz_template
@@ -3317,6 +3318,7 @@ tasks:
       npm_command: initsync-fuzzer
       suite: initial_sync_fuzzer
       resmoke_args: "--mongodSetParameters='{logComponentVerbosity: {command: 2}}'"
+      use_xlarge_distro: "true"
 
 ## Standalone generational fuzzer for multiversion aggregation pipelines ##
 - <<: *jstestfuzz_template