Skip to content

Commit 97ded5f

Browse files
fix: raise timeout
1 parent ee52317 commit 97ded5f

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

packages/celery-library/tests/conftest.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,15 @@
1010

1111
import pytest
1212
from 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+
)
1418
from celery.signals import worker_init
1519
from celery.worker.worker import WorkController
1620
from 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
1822
from celery_library.task_manager import CeleryTaskManager
1923
from celery_library.types import register_celery_types
2024
from 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

0 commit comments

Comments
 (0)