summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorTim Graham <timograham@gmail.com>2016-03-08 13:06:58 -0500
committerTim Graham <timograham@gmail.com>2016-03-08 13:06:58 -0500
commit5ca045cd25be692e5908550e9ae903da1925f60e (patch)
tree2a197d936ce6ed54e5d4926f747b71d8ff4de925 /tests
parent96ec67a7cf89a136e793305343c5bba8521cdb47 (diff)
Moved createsuperuser test app into auth_tests.
Diffstat (limited to 'tests')
-rw-r--r--tests/auth_tests/test_management.py44
-rw-r--r--tests/createsuperuser/__init__.py0
-rw-r--r--tests/createsuperuser/tests.py55
3 files changed, 44 insertions, 55 deletions
diff --git a/tests/auth_tests/test_management.py b/tests/auth_tests/test_management.py
index fe3963e221..1584ca4d46 100644
--- a/tests/auth_tests/test_management.py
+++ b/tests/auth_tests/test_management.py
@@ -169,6 +169,28 @@ class ChangepasswordManagementCommandTestCase(TestCase):
call_command('changepassword', username='J\xfalia', stdout=self.stdout)
+class MultiDBChangepasswordManagementCommandTestCase(TestCase):
+ multi_db = True
+
+ @mock.patch.object(changepassword.Command, '_get_pass', return_value='not qwerty')
+ def test_that_changepassword_command_with_database_option_uses_given_db(self, mock_get_pass):
+ """
+ changepassword --database should operate on the specified DB.
+ """
+ user = User.objects.db_manager('other').create_user(username='joe', password='qwerty')
+ self.assertTrue(user.check_password('qwerty'))
+
+ out = six.StringIO()
+ call_command('changepassword', username='joe', database='other', stdout=out)
+ command_output = out.getvalue().strip()
+
+ self.assertEqual(
+ command_output,
+ "Changing password for user 'joe'\nPassword changed successfully for user 'joe'"
+ )
+ self.assertTrue(User.objects.using('other').get(username="joe").check_password('not qwerty'))
+
+
@override_settings(
SILENCED_SYSTEM_CHECKS=['fields.W342'], # ForeignKey(unique=True)
AUTH_PASSWORD_VALIDATORS=[{'NAME': 'django.contrib.auth.password_validation.NumericPasswordValidator'}],
@@ -525,6 +547,28 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
test(self)
+class MultiDBCreatesuperuserTestCase(TestCase):
+ multi_db = True
+
+ def test_createsuperuser_command_with_database_option(self):
+ """
+ changepassword --database should operate on the specified DB.
+ """
+ new_io = six.StringIO()
+ call_command(
+ 'createsuperuser',
+ interactive=False,
+ username='joe',
+ email='joe@somewhere.org',
+ database='other',
+ stdout=new_io,
+ )
+ command_output = new_io.getvalue().strip()
+ self.assertEqual(command_output, 'Superuser created successfully.')
+ user = User.objects.using('other').get(username='joe')
+ self.assertEqual(user.email, 'joe@somewhere.org')
+
+
class CustomUserModelValidationTestCase(SimpleTestCase):
@override_settings(AUTH_USER_MODEL='auth_tests.CustomUserNonListRequiredFields')
@override_system_checks([check_user_model])
diff --git a/tests/createsuperuser/__init__.py b/tests/createsuperuser/__init__.py
deleted file mode 100644
index e69de29bb2..0000000000
--- a/tests/createsuperuser/__init__.py
+++ /dev/null
diff --git a/tests/createsuperuser/tests.py b/tests/createsuperuser/tests.py
deleted file mode 100644
index 3187e9b54d..0000000000
--- a/tests/createsuperuser/tests.py
+++ /dev/null
@@ -1,55 +0,0 @@
-from django.contrib.auth import models
-from django.contrib.auth.management.commands import changepassword
-from django.core.management import call_command
-from django.test import TestCase, mock
-from django.utils.six import StringIO
-
-
-class MultiDBChangepasswordManagementCommandTestCase(TestCase):
- multi_db = True
-
- def setUp(self):
- self.user = models.User.objects.db_manager('other').create_user(username='joe', password='qwerty')
-
- @mock.patch.object(changepassword.Command, '_get_pass', return_value='not qwerty')
- def test_that_changepassword_command_with_database_option_uses_given_db(self, mock_get_pass):
- """
- Executing the changepassword management command with a database option
- should operate on the specified DB
- """
- self.assertTrue(self.user.check_password('qwerty'))
-
- out = StringIO()
- call_command('changepassword', username='joe', database='other', stdout=out)
- command_output = out.getvalue().strip()
-
- self.assertEqual(
- command_output,
- "Changing password for user 'joe'\nPassword changed successfully for user 'joe'"
- )
- self.assertTrue(models.User.objects.using('other').get(username="joe").check_password("not qwerty"))
-
-
-class MultiDBCreatesuperuserTestCase(TestCase):
- multi_db = True
-
- def test_createsuperuser_command_with_database_option(self):
- " createsuperuser command should operate on specified DB"
- new_io = StringIO()
-
- call_command(
- "createsuperuser",
- interactive=False,
- username="joe",
- email="joe@somewhere.org",
- database='other',
- stdout=new_io
- )
- command_output = new_io.getvalue().strip()
-
- self.assertEqual(command_output, 'Superuser created successfully.')
-
- u = models.User.objects.using('other').get(username="joe")
- self.assertEqual(u.email, 'joe@somewhere.org')
-
- new_io.close()