Question

我想使用python mock库测试我的Django应用程序发送电子邮件。

测试代码：

# tests.py
from django.test import TestCase

class MyTestCase(TestCase):

    @mock.patch('django.core.mail.mail_managers')
    def test_canceled_wo_claiming(self, mocked_mail_managers):
        client = Client()
        client.get('/')
        print(mocked_mail_managers.called)
        mocked_mail_managers.assert_called_with('Hi, managers!', 'Message Body')

第一个例子 - 没有任务

# views.py
from django.views.generic import View
from django.core.mail import mail_managers

class MyView(View):

    def get(self, request):
        mail_managers('Hi, managers!', 'Message Body')
        return HttpResponse('Hello!')

第二个例子 - 有任务

# views.py
from django.views.generic import View
from . import tasks

class MyView(View):
    def get(self, request):
        tasks.notify.apply_async()
        return HttpResponse('Hello!')


# tasks.py
from celery import shared_task
from django.core.mail import mail_managers

@shared_task
def notify():
    mail_managers('Hi, managers!', 'Message Body')

第一个示例正常，第二个示例失败，Not called异常。

我的设置：

# Celery
BROKEN_URL = 'memory://'
BROKER_BACKEND = 'memory'

CELERY_ALWAYS_EAGER = True
CELERY_EAGER_PROPAGATES_EXCEPTIONS = True
TEST_RUNNER = 'djcelery.contrib.test_runner.CeleryTestSuiteRunner'

是否有可能执行此类集成测试或解决此问题的唯一方法是将测试分成两个？

Answer 1

直接从使用它们的代码测试异步任务的行为可能很棘手。一个原因是测试可能甚至在任务实际运行之前执行断言，这可能会给你误报。在这样的情况下我做的是将测试分为两个步骤：

模拟任务并测试在必须调用任务时调用它并使用预期的参数。
将任务作为独立功能进行测试，并将其作为普通功能执行，即不需要芹菜服务器。

为了说明，这可能是：

# views.py
from path.to.tasks import my_task


def my_view(requtest):
    # do stuff
    my_task.delay('foo', 'bar')
    return HttpResponse('whatever')


# test_my_task.py
from views import my_view
from path.to.tasks import my_task


class MyTest(TestCase):
    @mock.patch('path.to.tasks.my_task')
    def test_my_task_is_called(self, mocked_task):
        client = Client()
        client.get('/')
        my_task.assert_called_with('foo', 'bar')

    def test_my_task_works(self):
        my_task('foo', 'bar')  # note I don't use .delay(...), .apply_async(...), etc
        assert 'my task did what I expected it to do'

通过这种方式，您可以测试您的实现代码在您的任务方面是否正确运行，并且一旦按预期调用该任务就会正常运行。

我希望它有用！：）

Answer 2

我发现了一个问题而且非常愚蠢。 Described here和Here：

基本原则是你可以在查找对象的位置进行修补，这不一定与定义对象的位置相同。

我需要改变：

@mock.patch('django.core.mail.mail_managers')

带

@mock.patch('path.to.tasks.mail_managers')

Python：Mock在芹菜任务中不起作用

2 个答案: