Consolidated all tests for createsuperuser in the same TestCase. (b78f9a12) · Commits · Dom Sekotill / django

django/contrib/auth/tests/test_basic.py

+0 −147

Original line number	Diff line number	Diff line
		# -- encoding: utf-8 --
		from __future__ import unicode_literals

		import locale

		from django.apps import apps
		from django.contrib.auth import get_user_model
		from django.contrib.auth.management.commands import createsuperuser
		from django.contrib.auth.models import User, AnonymousUser
		from django.contrib.auth.tests.custom_user import CustomUser
		from django.contrib.auth.tests.utils import skipIfCustomUser
		@@ -15,8 +11,6 @@ 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 django.utils.encoding import force_str
		from django.utils.six import binary_type, PY2, StringIO


		@receiver(setting_changed)
		@@ -29,46 +23,6 @@ def user_model_swapped(**kwargs):
		apps.clear_cache()


		def mock_inputs(inputs):
		"""
		Decorator to temporarily replace input/getpass to allow interactive
		createsuperuser.
		"""
		def inner(test_func):
		def wrapped(*args):
		class mock_getpass:
		@staticmethod
		def getpass(prompt=b'Password: ', stream=None):
		if PY2:
		# getpass on Windows only supports prompt as bytestring (#19807)
		assert isinstance(prompt, binary_type)
		return inputs['password']

		def mock_input(prompt):
		# prompt should be encoded in Python 2. This line will raise an
		# Exception if prompt contains unencoded non-ascii on Python 2.
		prompt = str(prompt)
		assert str('__proxy__') not in prompt
		response = ''
		for key, val in inputs.items():
		if force_str(key) in prompt.lower():
		response = val
		break
		return response

		old_getpass = createsuperuser.getpass
		old_input = createsuperuser.input
		createsuperuser.getpass = mock_getpass
		createsuperuser.input = mock_input
		try:
		test_func(*args)
		finally:
		createsuperuser.getpass = old_getpass
		createsuperuser.input = old_input
		return wrapped
		return inner


		@skipIfCustomUser
		class BasicTestCase(TestCase):
		def test_user(self):
		@@ -127,107 +81,6 @@ class BasicTestCase(TestCase):
		self.assertTrue(super.is_active)
		self.assertTrue(super.is_staff)

		def test_createsuperuser_management_command(self):
		"Check the operation of the createsuperuser management command"
		# We can use the management command to create a superuser
		new_io = StringIO()
		call_command(
		"createsuperuser",
		interactive=False,
		username="joe",
		email="joe@somewhere.org",
		stdout=new_io
		)
		command_output = new_io.getvalue().strip()
		self.assertEqual(command_output, 'Superuser created successfully.')
		u = User.objects.get(username="joe")
		self.assertEqual(u.email, 'joe@somewhere.org')

		# created password should be unusable
		self.assertFalse(u.has_usable_password())

		# We can supress output on the management command
		new_io = StringIO()
		call_command(
		"createsuperuser",
		interactive=False,
		username="joe2",
		email="joe2@somewhere.org",
		verbosity=0,
		stdout=new_io
		)
		command_output = new_io.getvalue().strip()
		self.assertEqual(command_output, '')
		u = User.objects.get(username="joe2")
		self.assertEqual(u.email, 'joe2@somewhere.org')
		self.assertFalse(u.has_usable_password())

		call_command(
		"createsuperuser",
		interactive=False,
		username="joe+admin@somewhere.org",
		email="joe@somewhere.org",
		verbosity=0
		)
		u = User.objects.get(username="joe+admin@somewhere.org")
		self.assertEqual(u.email, 'joe@somewhere.org')
		self.assertFalse(u.has_usable_password())

		@mock_inputs({'password': "nopasswd"})
		def test_createsuperuser_nolocale(self):
		"""
		Check that createsuperuser does not break when no locale is set. See
		ticket #16017.
		"""

		old_getdefaultlocale = locale.getdefaultlocale
		try:
		# Temporarily remove locale information
		locale.getdefaultlocale = lambda: (None, None)

		# Call the command in this new environment
		call_command(
		"createsuperuser",
		interactive=True,
		username="nolocale@somewhere.org",
		email="nolocale@somewhere.org",
		verbosity=0
		)

		except TypeError:
		self.fail("createsuperuser fails if the OS provides no information about the current locale")

		finally:
		# Re-apply locale information
		locale.getdefaultlocale = old_getdefaultlocale

		# If we were successful, a user should have been created
		u = User.objects.get(username="nolocale@somewhere.org")
		self.assertEqual(u.email, 'nolocale@somewhere.org')

		@mock_inputs({
		'password': "nopasswd",
		'uživatel': 'foo', # username (cz)
		'email': 'nolocale@somewhere.org'})
		def test_createsuperuser_non_ascii_verbose_name(self):
		# Aliased so the string doesn't get extracted
		from django.utils.translation import ugettext_lazy as ulazy
		username_field = User._meta.get_field('username')
		old_verbose_name = username_field.verbose_name
		username_field.verbose_name = ulazy('uživatel')
		new_io = StringIO()
		try:
		call_command(
		"createsuperuser",
		interactive=True,
		stdout=new_io
		)
		finally:
		username_field.verbose_name = old_verbose_name

		command_output = new_io.getvalue().strip()
		self.assertEqual(command_output, 'Superuser created successfully.')

		def test_get_user_model(self):
		"The current user model can be retrieved"
		self.assertEqual(get_user_model(), User)

django/contrib/auth/tests/test_management.py

+108 −11

Original line number	Diff line number	Diff line
		from __future__ import unicode_literals

		from datetime import date
		import locale

		from django.apps import apps
		from django.contrib.auth import models, management
		from django.contrib.auth.checks import check_user_model
		from django.contrib.auth.management import create_permissions
		from django.contrib.auth.management.commands import changepassword
		from django.contrib.auth.management.commands import changepassword, createsuperuser
		from django.contrib.auth.models import User
		from django.contrib.auth.tests.custom_user import CustomUser
		from django.contrib.auth.tests.utils import skipIfCustomUser
		@@ -16,7 +18,47 @@ from django.core.management import call_command
		from django.core.management.base import CommandError
		from django.test import TestCase, override_settings, override_system_checks
		from django.utils import six
		from django.utils.six import StringIO
		from django.utils.encoding import force_str


		def mock_inputs(inputs):
		"""
		Decorator to temporarily replace input/getpass to allow interactive
		createsuperuser.
		"""
		def inner(test_func):
		def wrapped(*args):
		class mock_getpass:
		@staticmethod
		def getpass(prompt=b'Password: ', stream=None):
		if six.PY2:
		# getpass on Windows only supports prompt as bytestring (#19807)
		assert isinstance(prompt, six.binary_type)
		return inputs['password']

		def mock_input(prompt):
		# prompt should be encoded in Python 2. This line will raise an
		# Exception if prompt contains unencoded non-ascii on Python 2.
		prompt = str(prompt)
		assert str('__proxy__') not in prompt
		response = ''
		for key, val in inputs.items():
		if force_str(key) in prompt.lower():
		response = val
		break
		return response

		old_getpass = createsuperuser.getpass
		old_input = createsuperuser.input
		createsuperuser.getpass = mock_getpass
		createsuperuser.input = mock_input
		try:
		test_func(*args)
		finally:
		createsuperuser.getpass = old_getpass
		createsuperuser.input = old_input
		return wrapped
		return inner


		@skipIfCustomUser
		@@ -53,8 +95,8 @@ class ChangepasswordManagementCommandTestCase(TestCase):

		def setUp(self):
		self.user = models.User.objects.create_user(username='joe', password='qwerty')
		self.stdout = StringIO()
		self.stderr = StringIO()
		self.stdout = six.StringIO()
		self.stderr = six.StringIO()

		def tearDown(self):
		self.stdout.close()
		@@ -100,10 +142,10 @@ class ChangepasswordManagementCommandTestCase(TestCase):
		@skipIfCustomUser
		class CreatesuperuserManagementCommandTestCase(TestCase):

		def test_createsuperuser(self):
		def test_basic_usage(self):
		"Check the operation of the createsuperuser management command"
		# We can use the management command to create a superuser
		new_io = StringIO()
		new_io = six.StringIO()
		call_command(
		"createsuperuser",
		interactive=False,
		@@ -119,9 +161,64 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
		# created password should be unusable
		self.assertFalse(u.has_usable_password())

		@mock_inputs({'password': "nopasswd"})
		def test_nolocale(self):
		"""
		Check that createsuperuser does not break when no locale is set. See
		ticket #16017.
		"""

		old_getdefaultlocale = locale.getdefaultlocale
		try:
		# Temporarily remove locale information
		locale.getdefaultlocale = lambda: (None, None)

		# Call the command in this new environment
		call_command(
		"createsuperuser",
		interactive=True,
		username="nolocale@somewhere.org",
		email="nolocale@somewhere.org",
		verbosity=0
		)

		except TypeError:
		self.fail("createsuperuser fails if the OS provides no information about the current locale")

		finally:
		# Re-apply locale information
		locale.getdefaultlocale = old_getdefaultlocale

		# If we were successful, a user should have been created
		u = User.objects.get(username="nolocale@somewhere.org")
		self.assertEqual(u.email, 'nolocale@somewhere.org')

		@mock_inputs({
		'password': "nopasswd",
		'u\u017eivatel': 'foo', # username (cz)
		'email': 'nolocale@somewhere.org'})
		def test_non_ascii_verbose_name(self):
		# Aliased so the string doesn't get extracted
		from django.utils.translation import ugettext_lazy as ulazy
		username_field = User._meta.get_field('username')
		old_verbose_name = username_field.verbose_name
		username_field.verbose_name = ulazy('u\u017eivatel')
		new_io = six.StringIO()
		try:
		call_command(
		"createsuperuser",
		interactive=True,
		stdout=new_io
		)
		finally:
		username_field.verbose_name = old_verbose_name

		command_output = new_io.getvalue().strip()
		self.assertEqual(command_output, 'Superuser created successfully.')

		def test_verbosity_zero(self):
		# We can supress output on the management command
		new_io = StringIO()
		new_io = six.StringIO()
		call_command(
		"createsuperuser",
		interactive=False,
		@@ -137,7 +234,7 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
		self.assertFalse(u.has_usable_password())

		def test_email_in_username(self):
		new_io = StringIO()
		new_io = six.StringIO()
		call_command(
		"createsuperuser",
		interactive=False,
		@@ -155,7 +252,7 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
		# We can use the management command to create a superuser
		# We skip validation because the temporary substitution of the
		# swappable User model messes with validation.
		new_io = StringIO()
		new_io = six.StringIO()
		call_command(
		"createsuperuser",
		interactive=False,
		@@ -178,7 +275,7 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
		# We can use the management command to create a superuser
		# We skip validation because the temporary substitution of the
		# swappable User model messes with validation.
		new_io = StringIO()
		new_io = six.StringIO()
		with self.assertRaises(CommandError):
		call_command(
		"createsuperuser",
		@@ -201,7 +298,7 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
		def isatty(self):
		return False

		out = StringIO()
		out = six.StringIO()
		call_command(
		"createsuperuser",
		stdin=FakeStdin(),