mirror of
https://github.com/django/django.git
synced 2025-10-24 06:06:09 +00:00
[1.8.x] Moved contrib.messages tests out of contrib.
Backport of b3cd9e0d07 from master
This commit is contained in:
54
tests/messages_tests/test_api.py
Normal file
54
tests/messages_tests/test_api.py
Normal file
@@ -0,0 +1,54 @@
|
||||
from django.contrib import messages
|
||||
from django.test import RequestFactory, TestCase
|
||||
|
||||
|
||||
class DummyStorage(object):
|
||||
"""
|
||||
dummy message-store to test the api methods
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.store = []
|
||||
|
||||
def add(self, level, message, extra_tags=''):
|
||||
self.store.append(message)
|
||||
|
||||
|
||||
class ApiTest(TestCase):
|
||||
def setUp(self):
|
||||
self.rf = RequestFactory()
|
||||
self.request = self.rf.request()
|
||||
self.storage = DummyStorage()
|
||||
|
||||
def test_ok(self):
|
||||
msg = 'some message'
|
||||
|
||||
self.request._messages = self.storage
|
||||
messages.add_message(self.request, messages.DEBUG, msg)
|
||||
self.assertIn(msg, self.storage.store)
|
||||
|
||||
def test_request_is_none(self):
|
||||
msg = 'some message'
|
||||
|
||||
self.request._messages = self.storage
|
||||
|
||||
with self.assertRaises(TypeError):
|
||||
messages.add_message(None, messages.DEBUG, msg)
|
||||
|
||||
self.assertEqual([], self.storage.store)
|
||||
|
||||
def test_middleware_missing(self):
|
||||
msg = 'some message'
|
||||
|
||||
with self.assertRaises(messages.MessageFailure):
|
||||
messages.add_message(self.request, messages.DEBUG, msg)
|
||||
|
||||
self.assertEqual([], self.storage.store)
|
||||
|
||||
def test_middleware_missing_silently(self):
|
||||
msg = 'some message'
|
||||
|
||||
messages.add_message(self.request, messages.DEBUG, msg,
|
||||
fail_silently=True)
|
||||
|
||||
self.assertEqual([], self.storage.store)
|
||||
Reference in New Issue
Block a user