[1.8.x] Moved non-documented auth test models to the new test location. (2482c9dd) · Commits · Dom Sekotill / django

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