如何对运行芹菜任务的代码进行单元测试？

Question

我正在开发的应用程序是高度异步的。 Web 应用程序根据用户操作通过 celery 运行许多任务。 celery 任务本身能够启动进一步的任务。

如下所示的代码在我们的代码库中经常出现。

def do_sth():
    logic();
    if condition:
         function1.apply_async(*args)
    else:
         function2.apply_asynch(*args)

现在我们想开始对我们编写的任何新代码进行单元测试，但我们不确定如何执行此操作。 我们想在我们的pytest单元测试中断言我们想看看 function1 是否真的被调用了。 我们不希望一定运行function1本身，我们将会对单元测试function1 。

我不想将 celery 作为一个进程运行，我也不想在单元测试会话期间运行任何 AMQP 代理。

这是可以实现的吗？

编辑

有人指出，这是如何对 Celery 任务进行单元测试？

它不是。 想想看。 我要问的是如何测试函数是否通过 apply_async 调用了function1 。 这个问题是关于我如何测试function1本身。 有很大的不同。 在制定这个问题之前，我确实遇到了这个问题。

Answer 1

看看unittest.mock模块，它允许用“模拟”替换您不想调用的函数，这些“模拟”看起来与原始函数非常相似，以说服调用代码它正在调用“真实的东西”。 然后，您可以检查是否实际调用了模拟以及使用了哪些参数。 例子：

from unittest.mock import patch

def test_do_sth():
    with patch('function1.apply_async') as function1_mock:
        do_sth()
        assert function1_mock.called