test_createsuperuser.py 922 B

12345678910111213141516171819202122232425262728
  1. from django.contrib.auth import get_user_model
  2. from django.core.management import call_command
  3. from django.test import TestCase
  4. from django.utils.six import StringIO
  5. class CreateSuperuserTests(TestCase):
  6. def test_create_superuser(self):
  7. """command creates superuser"""
  8. out = StringIO()
  9. call_command(
  10. "createsuperuser",
  11. interactive=False,
  12. username="joe",
  13. email="joe@somewhere.org",
  14. password="Pass.123",
  15. stdout=out
  16. )
  17. new_user = get_user_model().objects.order_by('-id')[:1][0]
  18. self.assertEqual(
  19. out.getvalue().splitlines()[-1].strip(),
  20. 'Superuser #%s has been created successfully.' % new_user.pk)
  21. self.assertEqual(new_user.username, 'joe')
  22. self.assertEqual(new_user.email, 'joe@somewhere.org')
  23. self.assertTrue(new_user.check_password("Pass.123"))