diff --git a/mongoengine/django/mongo_auth/models.py b/mongoengine/django/mongo_auth/models.py index 71797185..d4947a23 100644 --- a/mongoengine/django/mongo_auth/models.py +++ b/mongoengine/django/mongo_auth/models.py @@ -15,8 +15,8 @@ MONGOENGINE_USER_DOCUMENT = getattr( settings, 'MONGOENGINE_USER_DOCUMENT', 'mongoengine.django.auth.User') -def get_user_document(self): - """Get the user docuemnt class user for authentcation. +def get_user_document(): + """Get the user document class used for authentication. This is the class defined in settings.MONGOENGINE_USER_DOCUMENT, which defaults to `mongoengine.django.auth.User`. diff --git a/tests/test_django.py b/tests/test_django.py index d67b126d..46568ac8 100644 --- a/tests/test_django.py +++ b/tests/test_django.py @@ -21,14 +21,16 @@ settings.configure( try: from django.contrib.auth import authenticate, get_user_model 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 except Exception: DJ15 = False from django.contrib.sessions.tests import SessionTestsMixin from mongoengine.django.sessions import SessionStore, MongoSession - - from datetime import tzinfo, timedelta ZERO = timedelta(0) @@ -165,6 +167,8 @@ class QuerySetTest(unittest.TestCase): class Note(Document): text = StringField() + Note.drop_collection() + for i in xrange(1, 101): Note(name="Note: %s" % i).save() @@ -258,9 +262,12 @@ class MongoAuthTest(unittest.TestCase): User.drop_collection() super(MongoAuthTest, self).setUp() - def test_user_model(self): + def test_get_user_model(self): self.assertEqual(get_user_model(), MongoUser) + def test_get_user_document(self): + self.assertEqual(get_user_document(), User) + def test_user_manager(self): manager = get_user_model()._default_manager self.assertTrue(isinstance(manager, MongoUserManager))