test_createsuperuser.py 743 B

123456789101112131415161718192021222324252627282930
  1. from io import StringIO
  2. from django.contrib.auth import get_user_model
  3. from django.core.management import call_command
  4. User = get_user_model()
  5. def test_superuser_is_created_if_input_is_valid(db):
  6. out = StringIO()
  7. call_command(
  8. "createsuperuser",
  9. interactive=False,
  10. username="test",
  11. email="test@example.com",
  12. password="password",
  13. stdout=out,
  14. )
  15. command_output = out.getvalue().splitlines()[-1].strip()
  16. user = User.objects.order_by('-id')[:1][0]
  17. assert command_output == (
  18. "Superuser #%s has been created successfully." % user.pk
  19. )
  20. assert user.username == "test"
  21. assert user.email == "test@example.com"
  22. assert user.check_password("password")