Merge remote-tracking branch 'origin/pr/432'
Conflicts: tests/test_django.py
This commit is contained in:
commit
88ca0f8196
@ -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`.
|
||||||
|
@ -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))
|
||||||
|
Loading…
x
Reference in New Issue
Block a user