diff options
Diffstat (limited to 'lib/python2.7/site-packages/django/contrib/messages/tests/test_cookie.py')
| -rw-r--r-- | lib/python2.7/site-packages/django/contrib/messages/tests/test_cookie.py | 154 |
1 files changed, 0 insertions, 154 deletions
diff --git a/lib/python2.7/site-packages/django/contrib/messages/tests/test_cookie.py b/lib/python2.7/site-packages/django/contrib/messages/tests/test_cookie.py deleted file mode 100644 index 5871ecc..0000000 --- a/lib/python2.7/site-packages/django/contrib/messages/tests/test_cookie.py +++ /dev/null @@ -1,154 +0,0 @@ -import json - -from django.contrib.messages import constants -from django.contrib.messages.tests.base import BaseTests -from django.contrib.messages.storage.cookie import (CookieStorage, - MessageEncoder, MessageDecoder) -from django.contrib.messages.storage.base import Message -from django.test import TestCase -from django.test.utils import override_settings -from django.utils.safestring import SafeData, mark_safe - - -def set_cookie_data(storage, messages, invalid=False, encode_empty=False): - """ - Sets ``request.COOKIES`` with the encoded data and removes the storage - backend's loaded data cache. - """ - encoded_data = storage._encode(messages, encode_empty=encode_empty) - if invalid: - # Truncate the first character so that the hash is invalid. - encoded_data = encoded_data[1:] - storage.request.COOKIES = {CookieStorage.cookie_name: encoded_data} - if hasattr(storage, '_loaded_data'): - del storage._loaded_data - - -def stored_cookie_messages_count(storage, response): - """ - Returns an integer containing the number of messages stored. - """ - # Get a list of cookies, excluding ones with a max-age of 0 (because - # they have been marked for deletion). - cookie = response.cookies.get(storage.cookie_name) - if not cookie or cookie['max-age'] == 0: - return 0 - data = storage._decode(cookie.value) - if not data: - return 0 - if data[-1] == CookieStorage.not_finished: - data.pop() - return len(data) - - -@override_settings(SESSION_COOKIE_DOMAIN='.example.com') -class CookieTest(BaseTests, TestCase): - storage_class = CookieStorage - - def stored_messages_count(self, storage, response): - return stored_cookie_messages_count(storage, response) - - def test_get(self): - storage = self.storage_class(self.get_request()) - # Set initial data. - example_messages = ['test', 'me'] - set_cookie_data(storage, example_messages) - # Test that the message actually contains what we expect. - self.assertEqual(list(storage), example_messages) - - def test_domain(self): - """ - Ensure that CookieStorage honors SESSION_COOKIE_DOMAIN. - Refs #15618. - """ - # Test before the messages have been consumed - storage = self.get_storage() - response = self.get_response() - storage.add(constants.INFO, 'test') - storage.update(response) - self.assertTrue('test' in response.cookies['messages'].value) - self.assertEqual(response.cookies['messages']['domain'], '.example.com') - self.assertEqual(response.cookies['messages']['expires'], '') - - # Test after the messages have been consumed - storage = self.get_storage() - response = self.get_response() - storage.add(constants.INFO, 'test') - for m in storage: - pass # Iterate through the storage to simulate consumption of messages. - storage.update(response) - self.assertEqual(response.cookies['messages'].value, '') - self.assertEqual(response.cookies['messages']['domain'], '.example.com') - self.assertEqual(response.cookies['messages']['expires'], 'Thu, 01-Jan-1970 00:00:00 GMT') - - def test_get_bad_cookie(self): - request = self.get_request() - storage = self.storage_class(request) - # Set initial (invalid) data. - example_messages = ['test', 'me'] - set_cookie_data(storage, example_messages, invalid=True) - # Test that the message actually contains what we expect. - self.assertEqual(list(storage), []) - - def test_max_cookie_length(self): - """ - Tests that, if the data exceeds what is allowed in a cookie, older - messages are removed before saving (and returned by the ``update`` - method). - """ - storage = self.get_storage() - response = self.get_response() - - # When storing as a cookie, the cookie has constant overhead of approx - # 54 chars, and each message has a constant overhead of about 37 chars - # and a variable overhead of zero in the best case. We aim for a message - # size which will fit 4 messages into the cookie, but not 5. - # See also FallbackTest.test_session_fallback - msg_size = int((CookieStorage.max_cookie_size - 54) / 4.5 - 37) - for i in range(5): - storage.add(constants.INFO, str(i) * msg_size) - unstored_messages = storage.update(response) - - cookie_storing = self.stored_messages_count(storage, response) - self.assertEqual(cookie_storing, 4) - - self.assertEqual(len(unstored_messages), 1) - self.assertTrue(unstored_messages[0].message == '0' * msg_size) - - def test_json_encoder_decoder(self): - """ - Tests that a complex nested data structure containing Message - instances is properly encoded/decoded by the custom JSON - encoder/decoder classes. - """ - messages = [ - { - 'message': Message(constants.INFO, 'Test message'), - 'message_list': [Message(constants.INFO, 'message %s') \ - for x in range(5)] + [{'another-message': \ - Message(constants.ERROR, 'error')}], - }, - Message(constants.INFO, 'message %s'), - ] - encoder = MessageEncoder(separators=(',', ':')) - value = encoder.encode(messages) - decoded_messages = json.loads(value, cls=MessageDecoder) - self.assertEqual(messages, decoded_messages) - - def test_safedata(self): - """ - Tests that a message containing SafeData is keeping its safe status when - retrieved from the message storage. - """ - def encode_decode(data): - message = Message(constants.DEBUG, data) - encoded = storage._encode(message) - decoded = storage._decode(encoded) - return decoded.message - - storage = self.get_storage() - - self.assertIsInstance( - encode_decode(mark_safe("<b>Hello Django!</b>")), SafeData) - self.assertNotIsInstance( - encode_decode("<b>Hello Django!</b>"), SafeData) |