fix bug

Author: khvn26

poetry.lock

@@ -19,7 +19,10 @@ dependencies = [
     "requests",
     "simplejson (>=3,<4)",
 ]
-optional-dependencies = { test-tools = ["pyfakefs (>=5,<6)"] }
+optional-dependencies = { test-tools = [
+    "pyfakefs (>=5,<6)",
+    "pytest-django (>=4,<5)",
+] }
 authors = [
     { name = "Matthew Elwell" },
     { name = "Gagan Trivedi" },

pyproject.toml

@@ -13,6 +13,7 @@
     Snapshot,
     SnapshotFixture,
 )
+from task_processor.task_run_method import TaskRunMethod
 
 
 def pytest_addoption(parser: pytest.Parser) -> None:
@@ -104,7 +105,7 @@ def run_tasks_impl(
     db: None,
     task_processor_mode: None,
 ) -> RunTasksFixture:
-    settings.TASK_RUN_METHOD = "TASK_PROCESSOR"
+    settings.TASK_RUN_METHOD = TaskRunMethod.TASK_PROCESSOR
 
     from task_processor.processor import run_tasks
 

src/common/test_tools/plugin.py

@@ -1,10 +1,13 @@
+from datetime import datetime, timedelta
+
 import prometheus_client
 import pytest
 
 from common.test_tools import AssertMetricFixture, RunTasksFixture
 from common.test_tools.plugin import assert_metric_impl
 from common.test_tools.utils import edition_printer
 from task_processor.decorators import register_task_handler
+from task_processor.models import Task
 
 
 def test_assert_metrics__asserts_expected(
@@ -75,3 +78,26 @@ def my_task() -> None:
     assert len(task_runs) == 1
     assert task_runs[0].task.task_identifier == "test_plugin.my_task"
     assert task_runs[0].result == "SUCCESS"
+
+
+def test_run_tasks__delay_inside_task__runs_expected(
+    run_tasks: RunTasksFixture,
+) -> None:
+    # Given
+    @register_task_handler()
+    def my_task() -> None:
+        return None
+
+    @register_task_handler()
+    def calling_task() -> None:
+        my_task.delay(delay_until=datetime.now() + timedelta(hours=1))
+
+    calling_task.delay()
+
+    # When
+    task_runs = run_tasks(num_tasks=1)
+
+    # Then
+    assert len(task_runs) == 1
+    assert Task.objects.count() == 2
+    assert Task.objects.latest("scheduled_for").task_identifier == "test_plugin.my_task"