Merge remote-tracking branch 'origin/pr/432'

Conflicts:
	tests/test_django.py
This commit is contained in:
Ross Lawley 2013-08-06 11:05:52 +00:00
commit 88ca0f8196
2 changed files with 13 additions and 6 deletions

View File

@ -15,8 +15,8 @@ MONGOENGINE_USER_DOCUMENT = getattr(
settings, 'MONGOENGINE_USER_DOCUMENT', 'mongoengine.django.auth.User') settings, 'MONGOENGINE_USER_DOCUMENT', 'mongoengine.django.auth.User')
def get_user_document(self): def get_user_document():
"""Get the user docuemnt class user for authentcation. """Get the user document class used for authentication.
This is the class defined in settings.MONGOENGINE_USER_DOCUMENT, which This is the class defined in settings.MONGOENGINE_USER_DOCUMENT, which
defaults to `mongoengine.django.auth.User`. defaults to `mongoengine.django.auth.User`.

View File

@ -21,14 +21,16 @@ settings.configure(
try: try:
from django.contrib.auth import authenticate, get_user_model from django.contrib.auth import authenticate, get_user_model
from mongoengine.django.auth import User from mongoengine.django.auth import User
from mongoengine.django.mongo_auth.models import MongoUser, MongoUserManager from mongoengine.django.mongo_auth.models import (
MongoUser,
MongoUserManager,
get_user_document,
)
DJ15 = True DJ15 = True
except Exception: except Exception:
DJ15 = False DJ15 = False
from django.contrib.sessions.tests import SessionTestsMixin from django.contrib.sessions.tests import SessionTestsMixin
from mongoengine.django.sessions import SessionStore, MongoSession from mongoengine.django.sessions import SessionStore, MongoSession
from datetime import tzinfo, timedelta from datetime import tzinfo, timedelta
ZERO = timedelta(0) ZERO = timedelta(0)
@ -165,6 +167,8 @@ class QuerySetTest(unittest.TestCase):
class Note(Document): class Note(Document):
text = StringField() text = StringField()
Note.drop_collection()
for i in xrange(1, 101): for i in xrange(1, 101):
Note(name="Note: %s" % i).save() Note(name="Note: %s" % i).save()
@ -258,9 +262,12 @@ class MongoAuthTest(unittest.TestCase):
User.drop_collection() User.drop_collection()
super(MongoAuthTest, self).setUp() super(MongoAuthTest, self).setUp()
def test_user_model(self): def test_get_user_model(self):
self.assertEqual(get_user_model(), MongoUser) self.assertEqual(get_user_model(), MongoUser)
def test_get_user_document(self):
self.assertEqual(get_user_document(), User)
def test_user_manager(self): def test_user_manager(self):
manager = get_user_model()._default_manager manager = get_user_model()._default_manager
self.assertTrue(isinstance(manager, MongoUserManager)) self.assertTrue(isinstance(manager, MongoUserManager))