pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add a test case for get_user_document

Browse Source
This commit is contained in:
Nicolas Cortot 2013-08-04 11:08:11 +02:00
parent 40b0a15b35
commit a0d255369a
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_django.py
View File

@ -22,7 +22,11 @@ try:
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
@ -270,9 +274,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))