celery task php,python - 如何对Celery任务进行单元测试？

在我的情况下(我假设许多其他人)，我想要的只是使用pytest测试任务的内部逻辑。

TL; DR; 最终嘲笑一切(选项2)

示例用例：

tests/test_tasks.py

@shared_task(bind=True)

def add_task(self, a, b):

return a+b;

tests/test_tasks.py

from proj import add_task

def test_add():

assert add_task(1, 2) == 3, '1 + 2 should equal 3'

但是，自从tests/test_tasks.py装饰器做了很多芹菜内部逻辑，它并不是真正的单元测试。

所以，对我来说，有两个选择：

选项1：单独的内部逻辑

tests/test_tasks.py

def internal_add(a, b):

return a + b;

tests/test_tasks.py

from .tasks_logic import internal_add

@shared_task(bind=True)

def add_task(self, a, b):

return internal_add(a, b);

这看起来非常奇怪，除了使其不太可读之外，它还需要手动提取和传递属于请求的属性，例如tests/test_tasks.py以备不时之需，这会使逻辑变得不那么纯净。

选项2：嘲笑

嘲笑芹菜内部

tests/test_tasks.py

# noinspection PyUnresolvedReferences

from celery import shared_task

from mock import patch

def mock_signature(**kwargs):

return {}

def mocked_shared_task(*decorator_args, **decorator_kwargs):

def mocked_shared_decorator(func):

func.signature = func.si = func.s = mock_signature

return func

return mocked_shared_decorator

patch('celery.shared_task', mocked_shared_task).start()

然后允许我模拟请求对象(同样，如果你需要来自请求的东西，比如id或重试计数器)。

tests/test_tasks.py

from proj import add_task

class MockedRequest:

def __init__(self, id=None):

self.id = id or 1

class MockedTask:

def __init__(self, id=None):

self.request = MockedRequest(id=id)

def test_add():

mocked_task = MockedTask(id=3)

assert add_task(mocked_task, 1, 2) == 3, '1 + 2 should equal 3'

这个解决方案更加手动，但是，它为我提供了实际单元测试所需的控制，无需重复自己，也不会失去芹菜范围。