Fixing sessions for django 1.3 and django 1.4

This commit is contained in:
Ross Lawley 2012-04-24 21:00:30 +01:00
parent 88dc64653e
commit a928047147
2 changed files with 16 additions and 1 deletions

View File

@ -44,7 +44,7 @@ class SessionStore(SessionBase):
def create(self): def create(self):
while True: while True:
self.session_key = self._get_new_session_key() self._session_key = self._get_new_session_key()
try: try:
self.save(must_create=True) self.save(must_create=True)
except CreateError: except CreateError:

View File

@ -12,6 +12,10 @@ from django.core.paginator import Paginator
settings.configure() settings.configure()
from django.contrib.sessions.tests import SessionTestsMixin
from mongoengine.django.sessions import SessionStore, MongoSession
class QuerySetTest(unittest.TestCase): class QuerySetTest(unittest.TestCase):
def setUp(self): def setUp(self):
@ -88,3 +92,14 @@ class QuerySetTest(unittest.TestCase):
end = p * 2 end = p * 2
start = end - 1 start = end - 1
self.assertEqual(t.render(Context(d)), u'%d:%d:' % (start, end)) self.assertEqual(t.render(Context(d)), u'%d:%d:' % (start, end))
class MongoDBSessionTest(SessionTestsMixin, unittest.TestCase):
backend = SessionStore
def setUp(self):
connect(db='mongoenginetest')
MongoSession.drop_collection()
super(MongoDBSessionTest, self).setUp()