Allow write custom testrun in scheduler PRODX-39509
Change-Id: Iccae2b41f52f2cfb83c11c4352347c2d96da7923
diff --git a/testrail_bot/control/models.py b/testrail_bot/control/models.py
index 2994890..02ff78b 100644
--- a/testrail_bot/control/models.py
+++ b/testrail_bot/control/models.py
@@ -131,22 +131,27 @@
auto_now=True)
-TASK_CHOICES = [
+TASKS = [
("control.celery_tasks.tasks.check_today_testplan",
"Check today testplan",
[]
),
- ("control.celery_tasks.tasks.task_to_check_testplan",
- "Check testplan",
- ["testplan_id"]
+ ("control.celery_tasks.tasks.check_specific_testplan",
+ "Check specific testplan",
+ ["testplan_id_arg"]
),
]
+TASK_CHOICES = list(map(lambda x: x[:-1], TASKS))
+
class CronPeriodicTask(PeriodicTask):
cron = models.CharField(default="",
max_length=300,
blank=False)
+ task_name = models.CharField(max_length=300, choices=TASK_CHOICES,
+ default=TASK_CHOICES[0][0])
+ testplan_id_arg = models.CharField(max_length=30, blank=True, null=True)
class Meta:
ordering = ["id"]