def test_delay_scheduled_3(self): task = Task(self.tiger, simple_task, unique=True) task.delay(when=datetime.timedelta(minutes=5)) self._ensure_queues(scheduled={'default': 1}) # We can look up a unique task by recreating it. task = Task(self.tiger, simple_task, unique=True) task.cancel() self._ensure_queues()
def test_delay_scheduled(self): task = Task(self.tiger, simple_task, queue='a') task.delay(when=datetime.timedelta(minutes=5)) self._ensure_queues(scheduled={'a': 1}) # Test canceling a scheduled task. task.cancel() self._ensure_queues() # Canceling again raises an error pytest.raises(TaskNotFound, task.cancel)
def test_delay_scheduled_2(self): task = Task(self.tiger, simple_task, queue='a') task.delay(when=datetime.timedelta(minutes=5)) self._ensure_queues(scheduled={'a': 1}) task_id = task.id # We can't look up a non-unique task by recreating it. task = Task(self.tiger, simple_task, queue='a') pytest.raises(TaskNotFound, task.cancel) # We can look up a task by its ID. fetch_task = lambda: Task.from_id(self.tiger, 'a', 'scheduled', task_id) task = fetch_task() task.cancel() self._ensure_queues() # Task.from_id raises if it doesn't exist. pytest.raises(TaskNotFound, fetch_task)