diff options
Diffstat (limited to 'tests/auth_tests/test_models.py')
| -rw-r--r-- | tests/auth_tests/test_models.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tests/auth_tests/test_models.py b/tests/auth_tests/test_models.py index a92f882de0..c939f3437c 100644 --- a/tests/auth_tests/test_models.py +++ b/tests/auth_tests/test_models.py @@ -3,6 +3,7 @@ from __future__ import unicode_literals from django.conf.global_settings import PASSWORD_HASHERS from django.contrib.auth import get_user_model +from django.contrib.auth.base_user import AbstractBaseUser from django.contrib.auth.hashers import get_hasher from django.contrib.auth.models import ( AbstractUser, Group, Permission, User, UserManager, @@ -12,6 +13,8 @@ from django.core import mail from django.db.models.signals import post_save from django.test import TestCase, mock, override_settings +from .models.with_custom_email_field import CustomEmailField + class NaturalKeysTestCase(TestCase): @@ -160,6 +163,14 @@ class AbstractBaseUserTests(TestCase): self.assertNotEqual(username, ohm_username) self.assertEqual(username, 'iamtheΩ') # U+03A9 GREEK CAPITAL LETTER OMEGA + def test_default_email(self): + user = AbstractBaseUser() + self.assertEqual(user.get_email_field_name(), 'email') + + def test_custom_email(self): + user = CustomEmailField() + self.assertEqual(user.get_email_field_name(), 'email_address') + class AbstractUserTestCase(TestCase): def test_email_user(self): |
