summaryrefslogtreecommitdiff
path: root/tests/auth_tests/test_basic.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/auth_tests/test_basic.py')
-rw-r--r--tests/auth_tests/test_basic.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/auth_tests/test_basic.py b/tests/auth_tests/test_basic.py
index 6fe88f769f..81b7ac9c56 100644
--- a/tests/auth_tests/test_basic.py
+++ b/tests/auth_tests/test_basic.py
@@ -3,13 +3,14 @@ from __future__ import unicode_literals
from django.apps import apps
from django.contrib.auth import get_user_model
from django.contrib.auth.models import AnonymousUser, User
-from django.contrib.auth.tests.custom_user import CustomUser
from django.core.exceptions import ImproperlyConfigured
from django.dispatch import receiver
from django.test import TestCase, override_settings
from django.test.signals import setting_changed
from django.utils import translation
+from .models import CustomUser
+
@receiver(setting_changed)
def user_model_swapped(**kwargs):
@@ -87,7 +88,7 @@ class BasicTestCase(TestCase):
"The current user model can be retrieved"
self.assertEqual(get_user_model(), User)
- @override_settings(AUTH_USER_MODEL='auth.CustomUser')
+ @override_settings(AUTH_USER_MODEL='auth_tests.CustomUser')
def test_swappable_user(self):
"The current user model can be swapped out for another"
self.assertEqual(get_user_model(), CustomUser)