summaryrefslogtreecommitdiff
path: root/tests/auth_tests/test_models.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/auth_tests/test_models.py')
-rw-r--r--tests/auth_tests/test_models.py11
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):