File tree Expand file tree Collapse file tree 1 file changed +13
-3
lines changed
packages/celery-library/tests Expand file tree Collapse file tree 1 file changed +13
-3
lines changed Original file line number Diff line number Diff line change 1010
1111import pytest
1212from celery import Celery # type: ignore[import-untyped]
13- from celery .contrib .testing .worker import TestWorkController , start_worker
13+ from celery .contrib .testing .worker import (
14+ TestWorkController ,
15+ start_worker ,
16+ test_worker_stopped ,
17+ )
1418from celery .signals import worker_init
1519from celery .worker .worker import WorkController
1620from celery_library .backends ._redis import RedisTaskInfoStore
17- from celery_library .signals import on_worker_init
21+ from celery_library .signals import on_worker_init , on_worker_shutdown
1822from celery_library .task_manager import CeleryTaskManager
1923from celery_library .types import register_celery_types
2024from pytest_simcore .helpers .monkeypatch_envs import setenvs_from_dict
@@ -132,15 +136,21 @@ def _on_worker_init_wrapper(sender: WorkController, **_kwargs):
132136
133137 worker_init .connect (_on_worker_init_wrapper )
134138
139+ def _on_worker_stopped_wrapper (_ : Celery , worker : WorkController , ** _kwargs ):
140+ return on_worker_shutdown (sender = worker , ** _kwargs )
141+
142+ test_worker_stopped .connect (_on_worker_stopped_wrapper )
143+
135144 register_celery_tasks (celery_app )
136145
137146 with start_worker (
138147 celery_app ,
139148 concurrency = 1 ,
140149 pool = "threads" ,
141- loglevel = "info " ,
150+ loglevel = "debug " ,
142151 perform_ping_check = False ,
143152 queues = "default" ,
153+ shutdown_timeout = 15.0 ,
144154 ) as worker :
145155 yield worker
146156
You can’t perform that action at this time.
0 commit comments