From 6eb3ce11e4d6be237ad2e4895d8ae0b7bc1803dc Mon Sep 17 00:00:00 2001 From: Simon Charette Date: Thu, 4 Feb 2016 11:47:51 -0500 Subject: Fixed #26089 -- Removed custom user test models from public API. Thanks to Tim Graham for the review. --- tests/auth_tests/test_basic.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tests/auth_tests/test_basic.py') 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) -- cgit v1.3