From 3ebafd1029d6e996b83ceb35ab3d399466ce5a69 Mon Sep 17 00:00:00 2001
From: Alex Neben <alex.neben@mongodb.com>
Date: Fri, 3 May 2024 10:37:30 -0600
Subject: [PATCH] DEVPROD-7042 Added tasks in generated tasks

---
 src/services/config_extraction.rs | 2 ++
 src/task_types/fuzzer_tasks.rs    | 3 +++
 src/task_types/resmoke_tasks.rs   | 3 +++
 3 files changed, 8 insertions(+)

diff --git a/src/services/config_extraction.rs b/src/services/config_extraction.rs
index c20d469..e27ffbc 100644
--- a/src/services/config_extraction.rs
+++ b/src/services/config_extraction.rs
@@ -212,6 +212,7 @@ impl ConfigExtractionService for ConfigExtractionServiceImpl {
             is_enterprise,
             platform: Some(evg_config_utils.infer_build_variant_platform(build_variant)),
             gen_task_suffix,
+            tags: task_def.tags.clone(),
         })
     }
 
@@ -289,6 +290,7 @@ impl ConfigExtractionService for ConfigExtractionServiceImpl {
             pass_through_vars: self.evg_config_utils.get_gen_task_vars(task_def),
             platform,
             gen_task_suffix,
+            tags: task_def.tags.clone(),
         })
     }
 
diff --git a/src/task_types/fuzzer_tasks.rs b/src/task_types/fuzzer_tasks.rs
index 17ba9ba..5f1647c 100644
--- a/src/task_types/fuzzer_tasks.rs
+++ b/src/task_types/fuzzer_tasks.rs
@@ -64,6 +64,8 @@ pub struct FuzzerGenTaskParams {
     pub platform: Option<String>,
     /// Name of variant specific suffix to add to tasks
     pub gen_task_suffix: Option<String>,
+    /// Tags to use in the generated tasks
+    pub tags: Option<Vec<String>>,
 }
 
 impl FuzzerGenTaskParams {
@@ -303,6 +305,7 @@ fn build_fuzzer_sub_task(
         name: formatted_name,
         commands: Some(commands),
         depends_on: params.get_dependencies(),
+        tags: params.tags.clone(),
         ..Default::default()
     }
 }
diff --git a/src/task_types/resmoke_tasks.rs b/src/task_types/resmoke_tasks.rs
index 4b90660..77ba900 100644
--- a/src/task_types/resmoke_tasks.rs
+++ b/src/task_types/resmoke_tasks.rs
@@ -76,6 +76,8 @@ pub struct ResmokeGenParams {
     pub platform: Option<String>,
     /// Name of variant specific suffix to add to tasks
     pub gen_task_suffix: Option<String>,
+    /// Tags to use in the generated tasks
+    pub tags: Option<Vec<String>>,
 }
 
 impl ResmokeGenParams {
@@ -782,6 +784,7 @@ impl GenResmokeTaskService for GenResmokeTaskServiceImpl {
                     params.require_multiversion_setup,
                 )),
                 depends_on: params.get_dependencies(),
+                tags: params.tags.clone(),
                 ..Default::default()
             },
             use_large_distro: params.use_large_distro,