test improv

Signed-off-by: Javan Lacerda <[email protected]>

Author: javanlacerda

src/clusterfuzz/_internal/bot/tasks/utasks/corpus_pruning_task.py

@@ -515,8 +515,8 @@ def process_bad_units(self, bad_units_path, quarantine_corpus_path):
             crash_stacktrace=state.crash_stacktrace,
             unit_path=unit_path,
             security_flag=security_flag)
-    logs.info('Found %d bad units, %d unique crashes.' % (num_bad_units,
-                                                          len(crashes)))
+    logs.info(
+        f'Found {num_bad_units} bad units, {len(crashes)} unique crashes.')
     return crashes
 
 
@@ -617,8 +617,8 @@ def _record_cross_pollination_stats(output):
 
 def _get_pruner_and_runner(context):
   build_directory = environment.get_value('BUILD_DIR')
-  match context.fuzz_target.engine:
-    case 'libFuzzer':
+  match context.fuzz_target.engine.lower():
+    case 'libfuzzer':
       runner = LibFuzzerRunner(build_directory, context)
       pruner = LibFuzzerPruner(runner)
     case 'centipede':

src/clusterfuzz/_internal/tests/core/bot/tasks/utasks/corpus_pruning_task_test.py

@@ -402,8 +402,11 @@ def setUp(self):
         'clusterfuzz._internal.datastore.data_handler.update_task_status',
         'clusterfuzz._internal.datastore.data_handler.get_task_status',
         'clusterfuzz._internal.bot.fuzzers.centipede.engine.Engine.minimize_corpus',
+        'clusterfuzz._internal.bot.tasks.utasks.corpus_pruning_task.Context._create_temp_corpus_directory',
     ])
 
+    self.default_path = '/tmp/arbitrary/path'
+    self.mock._create_temp_corpus_directory.return_value = self.default_path
     self.engine = centipede_engine.Engine()
     self.mock.setup_build.side_effect = self._mock_setup_build
     self.mock.get_application_id.return_value = 'project'
@@ -437,7 +440,7 @@ def test_prune(self):
     # asserting here
     self.mock.minimize_corpus.assert_called_once_with(
         self.engine, os.path.join(TEST_DIR, 'build/clusterfuzz_format_target'),
-        [], [unittest.mock.ANY], unittest.mock.ANY, unittest.mock.ANY, 79200)
+        [], [self.default_path], self.default_path, self.default_path, 79200)
 
 
 class GetProtoTimestampTest(unittest.TestCase):