SERVER-59450 Add a resmoke.py subcommand to invoke minimizer runner [evergreen task updates]

2 files changed, 24 insertions, 4 deletions

diff --git a/buildscripts/task_generation/task_types/fuzzer_tasks.py b/buildscripts/task_generation/task_types/fuzzer_tasks.py
index 1e62f668d4c..9292108f040 100644
--- a/buildscripts/task_generation/task_types/fuzzer_tasks.py
+++ b/buildscripts/task_generation/task_types/fuzzer_tasks.py
@@ -9,6 +9,8 @@ from buildscripts.task_generation.constants import ARCHIVE_DIST_TEST_DEBUG_TASK,
 from buildscripts.task_generation.task_types.multiversion_decorator import MultiversionGenTaskDecorator
 from buildscripts.util import taskname
 
+_MINIMIZABLE_FUZZERS = ["agg-fuzzer", "query-fuzzer"]
+
 
 class FuzzerTask(NamedTuple):
     """
@@ -59,6 +61,11 @@ class FuzzerGenTaskParams(NamedTuple):
     large_distro_name: Optional[str]
     config_location: str
 
+    @property
+    def is_minimizable(self) -> bool:
+        """Return whether this fuzzer in minimizable."""
+        return self.npm_command in _MINIMIZABLE_FUZZERS
+
     def jstestfuzz_params(self) -> Dict[str, str]:
         """Build a dictionary of parameters to pass to jstestfuzz."""
         return {
@@ -129,5 +136,7 @@ class FuzzerGenTaskService:
             FunctionCall("run jstestfuzz", params.jstestfuzz_params()),
             FunctionCall(RUN_GENERATED_TESTS, run_tests_vars)
         ]
+        if params.is_minimizable:
+            commands.append(FunctionCall("minimize jstestfuzz", params.jstestfuzz_params()))
 
         return Task(sub_task_name, commands, {TaskDependency(ARCHIVE_DIST_TEST_DEBUG_TASK)})
diff --git a/buildscripts/tests/task_generation/task_types/test_fuzzer_tasks.py b/buildscripts/tests/task_generation/task_types/test_fuzzer_tasks.py
index 155d5d59edc..89f5a8a3330 100644
--- a/buildscripts/tests/task_generation/task_types/test_fuzzer_tasks.py
+++ b/buildscripts/tests/task_generation/task_types/test_fuzzer_tasks.py
@@ -7,7 +7,7 @@ import buildscripts.task_generation.task_types.fuzzer_tasks as under_test
 # pylint: disable=missing-docstring,invalid-name,unused-argument,no-self-use,protected-access
 
 
-def build_mock_fuzzer_params(jstestfuzz_vars="vars for jstestfuzz"):
+def build_mock_fuzzer_params(jstestfuzz_vars="vars for jstestfuzz", npm_command="jstestfuzz"):
     return under_test.FuzzerGenTaskParams(
         task_name="task name",
         variant="build variant",
@@ -15,7 +15,7 @@ def build_mock_fuzzer_params(jstestfuzz_vars="vars for jstestfuzz"):
         num_files=10,
         num_tasks=5,
         resmoke_args="args for resmoke",
-        npm_command="jstestfuzz",
+        npm_command=npm_command,
         jstestfuzz_vars=jstestfuzz_vars,
         continue_on_failure=True,
         resmoke_jobs_max=5,
@@ -66,14 +66,25 @@ class TestGenerateTasks(unittest.TestCase):
 
 
 class TestBuildFuzzerSubTask(unittest.TestCase):
-    def test_sub_task_should_be_built_correct(self):
-        mock_params = build_mock_fuzzer_params()
+    def test_sub_task_should_be_built_correct_no_minimize_command(self):
+        mock_params = build_mock_fuzzer_params(npm_command="jstestfuzz")
         fuzzer_service = under_test.FuzzerGenTaskService()
 
         sub_task = fuzzer_service.build_fuzzer_sub_task(3, mock_params)
 
         self.assertEqual(sub_task.name, f"{mock_params.task_name}_3_{mock_params.variant}")
         self.assertEqual(len(sub_task.commands), 6)
+        self.assertNotEqual(sub_task.commands[-1].as_dict()["func"], "minimize jstestfuzz")
+
+    def test_sub_task_should_be_built_correct_with_minimize_command(self):
+        mock_params = build_mock_fuzzer_params(npm_command="agg-fuzzer")
+        fuzzer_service = under_test.FuzzerGenTaskService()
+
+        sub_task = fuzzer_service.build_fuzzer_sub_task(3, mock_params)
+
+        self.assertEqual(sub_task.name, f"{mock_params.task_name}_3_{mock_params.variant}")
+        self.assertEqual(len(sub_task.commands), 7)
+        self.assertEqual(sub_task.commands[-1].as_dict()["func"], "minimize jstestfuzz")
 
     def test_sub_task_should_include_timeout_info(self):
         mock_params = build_mock_fuzzer_params()