Commit 5ab327a3 authored by Tim Graham's avatar Tim Graham
Browse files

Moved non-documented auth test models to the new test location.

parent 2d7aca3d

django/contrib/auth/tests/custom_user.py

+25 −133
Original line number Diff line number Diff line
@@ -8,7 +8,6 @@ from django.db import models 
# The custom User uses email as the unique identifier, and requires

# that every user provide a date of birth. This lets us test

# changes in username datatype, and non-text required fields.



class CustomUserManager(BaseUserManager):

    def create_user(self, email, date_of_birth, password=None):

        """
@@ -33,18 +32,6 @@ class CustomUserManager(BaseUserManager): 
        return u





class CustomUserWithFKManager(BaseUserManager):

    def create_superuser(self, username, email, group, password):

        user = self.model(username_id=username, email_id=email, group_id=group)

        user.set_password(password)

        user.save(using=self._db)

        return user





class Email(models.Model):

    email = models.EmailField(verbose_name='email address', max_length=255, unique=True)





class CustomUser(AbstractBaseUser):

    email = models.EmailField(verbose_name='email address', max_length=255, unique=True)

    is_active = models.BooleanField(default=True)
@@ -90,26 +77,15 @@ class CustomUser(AbstractBaseUser): 
        return self.is_admin





class CustomUserWithFK(AbstractBaseUser):

    username = models.ForeignKey(Email, related_name='primary')

    email = models.ForeignKey(Email, to_field='email', related_name='secondary')

    group = models.ForeignKey(Group)



    custom_objects = CustomUserWithFKManager()



    USERNAME_FIELD = 'username'

    REQUIRED_FIELDS = ['email', 'group']



    class Meta:

        app_label = 'auth'





# At this point, temporarily remove the groups and user_permissions M2M

# fields from the AbstractUser class, so they don't clash with the related_name

# that sets.



old_au_local_m2m = AbstractUser._meta.local_many_to_many

old_pm_local_m2m = PermissionsMixin._meta.local_many_to_many

class RemoveGroupsAndPermissions(object):

    """

    A context manager to temporarily remove the groups and user_permissions M2M

    fields from the AbstractUser class, so they don't clash with the

    related_name sets.

    """

    def __enter__(self):

        self._old_au_local_m2m = AbstractUser._meta.local_many_to_many

        self._old_pm_local_m2m = PermissionsMixin._meta.local_many_to_many

        groups = models.ManyToManyField(Group, blank=True)

        groups.contribute_to_class(PermissionsMixin, "groups")

        user_permissions = models.ManyToManyField(Permission, blank=True)
@@ -117,11 +93,15 @@ user_permissions.contribute_to_class(PermissionsMixin, "user_permissions") 
        PermissionsMixin._meta.local_many_to_many = [groups, user_permissions]

        AbstractUser._meta.local_many_to_many = [groups, user_permissions]



    def __exit__(self, exc_type, exc_value, traceback):

        AbstractUser._meta.local_many_to_many = self._old_au_local_m2m

        PermissionsMixin._meta.local_many_to_many = self._old_pm_local_m2m





# The extension user is a simple extension of the built-in user class,

# adding a required date_of_birth field. This allows us to check for

# any hard references to the name "User" in forms/handlers etc.



with RemoveGroupsAndPermissions():

    class ExtensionUser(AbstractUser):

        date_of_birth = models.DateField()
@@ -131,91 +111,3 @@ class ExtensionUser(AbstractUser): 


        class Meta:

            app_label = 'auth'
 




# The CustomPermissionsUser users email as the identifier, but uses the normal

# Django permissions model. This allows us to check that the PermissionsMixin

# includes everything that is needed to interact with the ModelBackend.



class CustomPermissionsUserManager(CustomUserManager):

    def create_superuser(self, email, password, date_of_birth):

        u = self.create_user(email, password=password, date_of_birth=date_of_birth)

        u.is_superuser = True

        u.save(using=self._db)

        return u





class CustomPermissionsUser(AbstractBaseUser, PermissionsMixin):

    email = models.EmailField(verbose_name='email address', max_length=255, unique=True)

    date_of_birth = models.DateField()



    custom_objects = CustomPermissionsUserManager()



    USERNAME_FIELD = 'email'

    REQUIRED_FIELDS = ['date_of_birth']



    class Meta:

        app_label = 'auth'



    def get_full_name(self):

        return self.email



    def get_short_name(self):

        return self.email



    def __unicode__(self):

        return self.email





class IsActiveTestUser1(AbstractBaseUser):

    """

    This test user class and derivatives test the default is_active behavior

    """

    username = models.CharField(max_length=30, unique=True)



    custom_objects = BaseUserManager()



    USERNAME_FIELD = 'username'



    class Meta:

        app_label = 'auth'



    # the is_active attr is provided by AbstractBaseUser





class CustomUserNonUniqueUsername(AbstractBaseUser):

    "A user with a non-unique username"

    username = models.CharField(max_length=30)



    USERNAME_FIELD = 'username'



    class Meta:

        app_label = 'auth'





class CustomUserNonListRequiredFields(AbstractBaseUser):

    "A user with a non-list REQUIRED_FIELDS"

    username = models.CharField(max_length=30, unique=True)

    date_of_birth = models.DateField()



    USERNAME_FIELD = 'username'

    REQUIRED_FIELDS = 'date_of_birth'



    class Meta:

        app_label = 'auth'





class CustomUserBadRequiredFields(AbstractBaseUser):

    "A user with a USERNAME_FIELD that appears in REQUIRED_FIELDS (invalid)"

    username = models.CharField(max_length=30, unique=True)

    date_of_birth = models.DateField()



    USERNAME_FIELD = 'username'

    REQUIRED_FIELDS = ['username', 'date_of_birth']



    class Meta:

        app_label = 'auth'



# Undo swap hack

AbstractUser._meta.local_many_to_many = old_au_local_m2m

PermissionsMixin._meta.local_many_to_many = old_pm_local_m2m

tests/auth_tests/models/__init__.py

0 → 100644
+13 −0
Original line number Diff line number Diff line 
from .custom_permissions import CustomPermissionsUser

from .is_active import IsActiveTestUser1

from .invalid_models import (

    CustomUserNonUniqueUsername, CustomUserNonListRequiredFields,

    CustomUserBadRequiredFields,

)

from .with_foreign_key import CustomUserWithFK, Email



__all__ = (

    'CustomPermissionsUser', 'CustomUserNonUniqueUsername',

    'CustomUserNonListRequiredFields', 'CustomUserBadRequiredFields',

    'CustomUserWithFK', 'Email', 'IsActiveTestUser1',

)

tests/auth_tests/models/custom_permissions.py

0 → 100644
+43 −0
Original line number Diff line number Diff line 
"""

The CustomPermissionsUser users email as the identifier, but uses the normal

Django permissions model. This allows us to check that the PermissionsMixin

includes everything that is needed to interact with the ModelBackend.

"""

from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin

from django.contrib.auth.tests.custom_user import (

    CustomUserManager, RemoveGroupsAndPermissions,

)

from django.db import models

from django.utils.encoding import python_2_unicode_compatible





class CustomPermissionsUserManager(CustomUserManager):

    def create_superuser(self, email, password, date_of_birth):

        u = self.create_user(email, password=password, date_of_birth=date_of_birth)

        u.is_superuser = True

        u.save(using=self._db)

        return u





with RemoveGroupsAndPermissions():

    @python_2_unicode_compatible

    class CustomPermissionsUser(AbstractBaseUser, PermissionsMixin):

        email = models.EmailField(verbose_name='email address', max_length=255, unique=True)

        date_of_birth = models.DateField()



        custom_objects = CustomPermissionsUserManager()



        USERNAME_FIELD = 'email'

        REQUIRED_FIELDS = ['date_of_birth']



        class Meta:

            app_label = 'auth'



        def get_full_name(self):

            return self.email



        def get_short_name(self):

            return self.email



        def __str__(self):

            return self.email

tests/auth_tests/models/invalid_models.py

0 → 100644
+36 −0
Original line number Diff line number Diff line 
from django.contrib.auth.models import AbstractBaseUser

from django.db import models





class CustomUserNonUniqueUsername(AbstractBaseUser):

    "A user with a non-unique username"

    username = models.CharField(max_length=30)



    USERNAME_FIELD = 'username'



    class Meta:

        app_label = 'auth'





class CustomUserNonListRequiredFields(AbstractBaseUser):

    "A user with a non-list REQUIRED_FIELDS"

    username = models.CharField(max_length=30, unique=True)

    date_of_birth = models.DateField()



    USERNAME_FIELD = 'username'

    REQUIRED_FIELDS = 'date_of_birth'



    class Meta:

        app_label = 'auth'





class CustomUserBadRequiredFields(AbstractBaseUser):

    "A user with a USERNAME_FIELD that appears in REQUIRED_FIELDS (invalid)"

    username = models.CharField(max_length=30, unique=True)

    date_of_birth = models.DateField()



    USERNAME_FIELD = 'username'

    REQUIRED_FIELDS = ['username', 'date_of_birth']



    class Meta:

        app_label = 'auth'

tests/auth_tests/models/is_active.py

0 → 100644
+18 −0
Original line number Diff line number Diff line 
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager

from django.db import models





class IsActiveTestUser1(AbstractBaseUser):

    """

    This test user class and derivatives test the default is_active behavior

    """

    username = models.CharField(max_length=30, unique=True)



    custom_objects = BaseUserManager()



    USERNAME_FIELD = 'username'



    class Meta:

        app_label = 'auth'



    # the is_active attr is provided by AbstractBaseUser