Prefer retry method from the task decorator if no retry method is giv…

…en to RetryException (#356)

This matches the already documented behavior.

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## Version 0.21.0
+
+* When raising `RetryException` with no `method`, use task decorator retry method if set ([356](https://github.com/closeio/tasktiger/pull/356))
+
 ## Version 0.20.0
 
 * Add `tiger.get_sizes_for_queues_and_states` ([352](https://github.com/closeio/tasktiger/pull/352))

tasktiger/__init__.py

@@ -12,7 +12,7 @@
 from .tasktiger import TaskTiger, run_worker
 from .worker import Worker
 
-__version__ = "0.20.0"
+__version__ = "0.21.0"
 __all__ = [
     "TaskTiger",
     "Worker",

tasktiger/worker.py

@@ -771,8 +771,12 @@ def _mark_done() -> None:
                 has_job_timeout = True
 
             if execution and execution.get("retry"):
+                # Prefer retry method from the execution, then the task, then
+                # default.
                 if "retry_method" in execution:
                     retry_func, retry_args = execution["retry_method"]
+                elif task.retry_method:
+                    retry_func, retry_args = task.retry_method
                 else:
                     # We expect the serialized method here.
                     retry_func, retry_args = serialize_retry_method(

tests/tasks.py

@@ -135,6 +135,11 @@ def retry_task_2():
     raise RetryException(method=fixed(DELAY, 1), log_error=False)
 
 
+@tiger.task(retry_method=fixed(DELAY, 1))
+def retry_task_3():
+    raise RetryException(log_error=False)
+
+
 def verify_current_task():
     with redis.Redis(
         host=REDIS_HOST, db=TEST_DB, decode_responses=True

tests/test_base.py

@@ -39,6 +39,7 @@
     non_batch_task,
     retry_task,
     retry_task_2,
+    retry_task_3,
     simple_task,
     sleep_task,
     task_on_other_queue,
@@ -617,6 +618,23 @@ def test_retry_exception_2(self):
 
         pytest.raises(TaskNotFound, task.n_executions)
 
+    def test_retry_exception_3(self):
+        task = self.tiger.delay(retry_task_3)
+        self._ensure_queues(queued={"default": 1})
+        assert task.n_executions() == 0
+
+        Worker(self.tiger).run(once=True)
+        self._ensure_queues(scheduled={"default": 1})
+        assert task.n_executions() == 1
+
+        time.sleep(DELAY)
+
+        Worker(self.tiger).run(once=True)
+        Worker(self.tiger).run(once=True)
+        self._ensure_queues()
+
+        pytest.raises(TaskNotFound, task.n_executions)
+
     @pytest.mark.parametrize("count", [1, 3, 7])
     def test_retry_executions_count(self, count):
         task = self.tiger.delay(exception_task, retry_method=fixed(DELAY, 20))