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