diff --git a/iron_worker.py b/iron_worker.py index d2d6366..d28b947 100644 --- a/iron_worker.py +++ b/iron_worker.py @@ -440,6 +440,8 @@ def queue(self, task=None, tasks=None, retry=None, **kwargs): task.start_at = task.start_at.replace(tzinfo=tzlocal()) task_data["start_at"] = iron_core.IronClient.toRfc3339( task.start_at) + if task.timeout is not None: + task_data["timeout"] = task.timeout task_data["label"] = task.label task_data["cluster"] = task.cluster tasks_data.append(task_data) diff --git a/test.py b/test.py index 4ab9768..dadc0de 100644 --- a/test.py +++ b/test.py @@ -129,7 +129,7 @@ def test_postAdvancedSchedule(self): resp = self.worker.queue( code_name=self.code_name, payload={"schedule": "AWESOME SCHEDULE!"}, - start_at=start_at, run_every=3600, run_times=8) + start_at=start_at, run_every=3600, run_times=8, timeout=120) schedules = self.worker.tasks(scheduled=True) schedule_ids = []