diff --git a/tests/conftest.py b/tests/conftest.py index 852aa4178..22b851da5 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -22,8 +22,7 @@ from unittest.mock import MagicMock, patch def _mock_celery_task(): """全局 mock Celery 任务的 delay/apply_async/send_task 方法。""" - from celery import Task - from celery import Celery + from celery import Celery, Task # 保存原始方法 _orig_delay = Task.delay