[fuzzing-decision] Limit deadline to 5 days (TC API limit)

MozillaSecurity · Nov 20, 2024 · 880a392 · 880a392
1 parent ef31ee4
commit 880a392
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/services/fuzzing-decision/src/fuzzing_decision/decision/pool.py b/services/fuzzing-decision/src/fuzzing_decision/decision/pool.py
@@ -436,7 +436,12 @@ def build_tasks(
             task = yaml.safe_load(
                 FUZZING_TASK.substitute(
                     created=stringDate(now),
-                    deadline=stringDate(now + timedelta(seconds=preprocess.cycle_time)),
+                    deadline=stringDate(
+                        now
+                        + min(
+                            timedelta(days=5), timedelta(seconds=preprocess.cycle_time)
+                        )
+                    ),
                     description=DESCRIPTION.replace("\n", "\\n"),
                     expires=stringDate(fromNow("4 weeks", now)),
                     max_run_time=preprocess.max_run_time,
@@ -462,7 +467,9 @@ def build_tasks(
             task = yaml.safe_load(
                 FUZZING_TASK.substitute(
                     created=stringDate(now),
-                    deadline=stringDate(now + timedelta(seconds=self.cycle_time)),
+                    deadline=stringDate(
+                        now + min(timedelta(days=5), timedelta(seconds=self.cycle_time))
+                    ),
                     description=DESCRIPTION.replace("\n", "\\n"),
                     expires=stringDate(fromNow("4 weeks", now)),
                     max_run_time=self.max_run_time,
@@ -616,7 +623,10 @@ def build_tasks(
                 task = yaml.safe_load(
                     FUZZING_TASK.substitute(
                         created=stringDate(now),
-                        deadline=stringDate(now + timedelta(seconds=pool.cycle_time)),
+                        deadline=stringDate(
+                            now
+                            + min(timedelta(days=5), timedelta(seconds=pool.cycle_time))
+                        ),
                         description=DESCRIPTION.replace("\n", "\\n"),
                         expires=stringDate(fromNow("4 weeks", now)),
                         max_run_time=pool.max_run_time,