def test_get_legacy(self):
request = self.get_request()
storage = self.storage_class(request)
cookie_storage = self.get_cookie_storage(storage)
self.user.message_set.create(message='user message')
set_cookie_data(cookie_storage, ['cookie'])
# Test that the message actually contains what we expect.
self.assertEqual(len(storage), 2)
self.assertEqual(list(storage)[0].message, 'user message')
self.assertEqual(list(storage)[1], 'cookie')
def test_get_fallback_only(self):
request = self.get_request()
storage = self.storage_class(request)
cookie_storage, session_storage = storage.storages
# Set initial cookie and session data.
example_messages = [str(i) for i in range(5)]
set_cookie_data(cookie_storage, [], encode_empty=True)
set_session_data(session_storage,
example_messages + [EOFNotification()])
# Test that the message actually contains what we expect.
self.assertEqual(list(storage), example_messages)
def test_get_fallback_only(self):
request = self.get_request()
storage = self.storage_class(request)
cookie_storage, session_storage = storage.storages
# Set initial cookie and session data.
example_messages = [str(i) for i in range(5)]
set_cookie_data(cookie_storage, [], encode_empty=True)
set_session_data(session_storage,
example_messages + [EOFNotification()])
# Test that the message actually contains what we expect.
self.assertEqual(list(storage), example_messages)
def test_get(self):
request = self.get_request()
storage = self.storage_class(request)
cookie_storage = storage.storages[0]
# Set initial cookie data.
example_messages = [str(i) for i in range(5)]
set_cookie_data(cookie_storage, example_messages + [EOFNotification()])
# Overwrite the _get method of the second storage to prove it is not
# used (it would cause a TypeError: 'NoneType' object is not callable).
storage.storages[1]._get = None
# Test that the message actually contains what we expect.
self.assertEqual(list(storage), example_messages)
def test_get(self):
request = self.get_request()
storage = self.storage_class(request)
cookie_storage = storage.storages[0]
# Set initial cookie data.
example_messages = [str(i) for i in range(5)]
set_cookie_data(cookie_storage, example_messages + [EOFNotification()])
# Overwrite the _get method of the second storage to prove it is not
# used (it would cause a TypeError: 'NoneType' object is not callable).
storage.storages[1]._get = None
# Test that the message actually contains what we expect.
self.assertEqual(list(storage), example_messages)
def test_flush_used_backends(self):
request = self.get_request()
storage = self.storage_class(request)
cookie_storage = self.get_cookie_storage(storage)
session_storage = self.get_session_storage(storage)
# Set initial cookie and session data.
set_cookie_data(cookie_storage, ['cookie'])
set_session_data(session_storage, ['session'])
# When updating, previously used but no longer needed backends are
# flushed.
response = self.get_response()
list(storage)
storage.update(response)
session_storing = self.stored_session_messages_count(storage, response)
self.assertEqual(session_storing, 0)
