我正在研究一个Django项目并正在为它编写单元测试。但是,在测试中,当我尝试登录用户时,我收到此错误:
MessageFailure: You cannot add messages without installing django.contrib.messages.middleware.MessageMiddleware
登录实际站点工作正常 - 并使用MessageMiddleware显示登录消息。
在我的测试中,如果我这样做:
from django.conf import settings
print settings.MIDDLEWARE_CLASSES
然后它输出:
('django.middleware.cache.UpdateCacheMiddleware',
'django.middleware.common.CommonMiddleware',
'django.contrib.sessions.middleware.SessionMiddleware',
'django.middleware.csrf.CsrfViewMiddleware',
'django.contrib.auth.middleware.AuthenticationMiddleware',
'django.contrib.messages.middleware.MessageMiddleware',
'django.middleware.clickjacking.XFrameOptionsMiddleware',
'django.middleware.cache.FetchFromCacheMiddleware',
'debug_toolbar.middleware.DebugToolbarMiddleware')
这似乎表明在运行测试时安装了MessageMiddleware。
我有一个明显的步骤吗?
UPDATE
根据下面的建议,它确实看起来像是一个设置的东西。
我目前有这样的settings/__init__.py
:
try:
from settings.development import *
except ImportError:
pass
和settings/defaults.py
包含大部分标准设置(包括MIDDLEWARE_CLASSES
)。然后settings.development.py
覆盖了一些这样的默认值:
from defaults import *
DEBUG = True
# etc
看起来我的开发站点本身工作正常,使用开发设置。但是,虽然测试似乎加载设置好(默认和开发)settings.DEBUG
设置为False
。我不知道为什么,或者这是否是问题的原因。
使用RequestFactory创建请求时,Django 1.4 has a bug。
要解决此问题,请使用RequestFactory创建请求并执行以下操作:
from django.contrib.messages.storage.fallback import FallbackStorage
setattr(request, 'session', 'session')
messages = FallbackStorage(request)
setattr(request, '_messages', messages)
适合我!
解决这个问题非常优雅的方法是使用mock模拟消息模块
假设您在名为FooView
的app中有一个名为myapp
的基于类的视图
from django.contrib import messages
from django.views.generic import TemplateView
class FooView(TemplateView):
def post(self, request, *args, **kwargs):
...
messages.add_message(request, messages.SUCCESS, '\o/ Profit \o/')
...
你现在可以测试它
def test_successful_post(self):
mock_messages = patch('myapp.views.FooView.messages').start()
mock_messages.SUCCESS = success = 'super duper'
request = self.rf.post('/', {})
view = FooView.as_view()
response = view(request)
msg = _(u'\o/ Profit \o/')
mock_messages.add_message.assert_called_with(request, success, msg)
在我的情况下(django 1.8)这个问题发生在单元测试调用user_logged_in
信号的信号处理程序时,看起来消息app尚未被调用,即request._messages
尚未设置。这失败了:
from django.contrib.auth.signals import user_logged_in
...
@receiver(user_logged_in)
def user_logged_in_handler(sender, user, request, **kwargs):
...
messages.warning(request, "user has logged in")
在普通视图函数(后面称为)中对messages.warning
的相同调用没有任何问题。
一个解决方法我基于https://code.djangoproject.com/ticket/17971的一个建议,只在信号处理函数中使用fail_silently
参数,即这解决了我的问题:
messages.warning(request, "user has logged in",
fail_silently=True )
你只有一个settings.py吗?
测试创建自定义(测试)数据库。也许你没有任何消息或某些东西......也许你需要setUp()灯具或什么?
需要更多信息才能正确回答。
为什么不简单地做一些像?你确定在调试模式下运行测试吗?
# settings.py
DEBUG = True
from django.conf import settings
# where message is sent:
if not settings.DEBUG:
# send your message ...
这建立在Tarsis Azevedo's answer的基础上,创建一个MessagingRequest
助手类。
鉴于一个KittenAdmin
,我想获得100%的测试覆盖率:
from django.contrib import admin, messages
class KittenAdmin(admin.ModelAdmin):
def warm_fuzzy_method(self, request):
messages.warning(request, 'Can I haz cheezburger?')
我创建了一个MessagingRequest
助手类,用于说一个test_helpers.py
文件:
from django.contrib.messages.storage.fallback import FallbackStorage
from django.http import HttpRequest
class MessagingRequest(HttpRequest):
session = 'session'
def __init__(self):
super(MessagingRequest, self).__init__()
self._messages = FallbackStorage(self)
def get_messages(self):
return getattr(self._messages, '_queued_messages')
def get_message_strings(self):
return [str(m) for m in self.get_messages()]
然后在标准Django tests.py
:
from django.contrib.admin.sites import AdminSite
from django.test import TestCase
from cats.kitten.admin import KittenAdmin
from cats.kitten.models import Kitten
from cats.kitten.test_helpers import MessagingRequest
class KittenAdminTest(TestCase):
def test_kitten_admin_message(self):
admin = KittenAdmin(model=Kitten, admin_site=AdminSite())
expect = ['Can I haz cheezburger?']
request = MessagingRequest()
admin.warm_fuzzy_method(request)
self.assertEqual(request.get_message_strings(), expect)
结果:
coverage run --include='cats/kitten/*' manage.py test; coverage report -m
Creating test database for alias 'default'...
.
----------------------------------------------------------------------
Ran 1 test in 0.001s
OK
Destroying test database for alias 'default'...
Name Stmts Miss Cover Missing
----------------------------------------------------------------------
cats/kitten/__init__.py 0 0 100%
cats/kitten/admin.py 4 0 100%
cats/kitten/migrations/0001_initial.py 5 0 100%
cats/kitten/migrations/__init__.py 0 0 100%
cats/kitten/models.py 3 0 100%
cats/kitten/test_helpers.py 11 0 100%
cats/kitten/tests.py 12 0 100%
----------------------------------------------------------------------
TOTAL 35 0 100%
当从单元测试调用时,在login_callback信号接收器函数中发生了这种情况,问题的解决方法是:
from django.contrib.messages.storage import default_storage
@receiver(user_logged_in)
def login_callback(sender, user, request, **kwargs):
if not hasattr(request, '_messages'): # fails for tests
request._messages = default_storage(request)
Django 2.0.x
如果您在中间件中发现问题,那么您不会进行“单元测试”。单元测试测试一个功能单元。如果您与系统的其他部分进行交互,那么您正在进行一种称为“集成”测试的操作。
你应该尝试编写更好的测试,不应该出现这种问题。试试RequestFactory。 ;)
def test_some_view(self):
factory = RequestFactory()
user = get_mock_user()
request = factory.get("/my/view")
request.user = user
response = my_view(request)
self.asssertEqual(status_code, 200)