django-lambda-tasks 0.4.0__tar.gz → 0.4.1__tar.gz

{django_lambda_tasks-0.4.0 → django_lambda_tasks-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-lambda-tasks
-Version: 0.4.0
+Version: 0.4.1
 Summary: Run async tasks in a lambda function
 Requires-Python: >=3.10
 Requires-Dist: awslambdaric

{django_lambda_tasks-0.4.0 → django_lambda_tasks-0.4.1}/lambda_tasks/local_executor.py

@@ -1,10 +1,13 @@
 """Process pool executor for async local task execution."""
 
+import logging
 import uuid
-from concurrent.futures import ProcessPoolExecutor
+from concurrent.futures import Future, ProcessPoolExecutor
 
 from lambda_tasks.settings import LambdaTasksSettings
 
+logger = logging.getLogger(__name__)
+
 _pool: ProcessPoolExecutor | None = None
 
 
@@ -38,8 +41,18 @@ def _execute_in_worker(*, message_json: str, message_id: str) -> None:
     message.execute_immediately(message_id=message_id)
 
 
+def _log_worker_exception(future: Future) -> None:  # type: ignore[type-arg]
+    """Callback attached to each worker future. Logs unhandled exceptions."""
+    exception = future.exception()
+    if exception is not None:
+        logger.error("Worker process raised an exception", exc_info=exception)
+
+
 def submit_task(*, message_json: str) -> None:
     """Submit a task to the process pool. Fire-and-forget."""
     pool = get_pool()
     message_id = str(uuid.uuid4())
-    pool.submit(_execute_in_worker, message_json=message_json, message_id=message_id)
+    future = pool.submit(
+        _execute_in_worker, message_json=message_json, message_id=message_id
+    )
+    future.add_done_callback(_log_worker_exception)

{django_lambda_tasks-0.4.0 → django_lambda_tasks-0.4.1}/pyproject.toml

@@ -7,7 +7,7 @@ packages = ["lambda_tasks"]
 
 [project]
 name = "django-lambda-tasks"
-version = "0.4.0"
+version = "0.4.1"
 description = "Run async tasks in a lambda function"
 readme = "README.md"
 requires-python = ">=3.10"

{django_lambda_tasks-0.4.0 → django_lambda_tasks-0.4.1}/tests/test_local_executor.py

@@ -344,7 +344,7 @@ class TestSubmitTask:
         assert str(parsed) == message_id
 
     def test_submit_task_does_not_wait_on_future(self, settings):
-        """submit_task() does not call .result() or .add_done_callback() on the Future.
+        """submit_task() does not call .result() on the Future.
 
         Validates: Requirements 3.4, 5.3
         """
@@ -363,7 +363,24 @@ class TestSubmitTask:
             )
 
         mock_future.result.assert_not_called()
-        mock_future.add_done_callback.assert_not_called()
+
+    def test_submit_task_attaches_exception_logging_callback(self, settings):
+        """submit_task() attaches _log_worker_exception as a done callback."""
+        from lambda_tasks.local_executor import _log_worker_exception, submit_task
+
+        settings.LAMBDA_TASKS_LOCAL_WORKERS = 2
+        settings.LAMBDA_TASKS_EAGER = False
+
+        mock_pool = MagicMock()
+        mock_future = MagicMock()
+        mock_pool.submit.return_value = mock_future
+
+        with patch("lambda_tasks.local_executor.get_pool", return_value=mock_pool):
+            submit_task(
+                message_json='{"task_name": "myapp.tasks.foo", "kwargs": {}, "n_retries": 0}'
+            )
+
+        mock_future.add_done_callback.assert_called_once_with(_log_worker_exception)
 
 
 # ---------------------------------------------------------------------------