diff --git a/tests/test_django.py b/tests/test_django.py index 56a61640..1741e0f8 100644 --- a/tests/test_django.py +++ b/tests/test_django.py @@ -7,6 +7,7 @@ from mongoengine import * from mongoengine.django.shortcuts import get_document_or_404 +import django from django.http import Http404 from django.template import Context, Template from django.conf import settings @@ -19,6 +20,10 @@ settings.configure( AUTHENTICATION_BACKENDS = ('mongoengine.django.auth.MongoEngineBackend',) ) +# For Django >= 1.7 +if hasattr(django, 'setup'): + django.setup() + try: from django.contrib.auth import authenticate, get_user_model from mongoengine.django.auth import User