test_createsuperuser.py 729 B

12345678910111213141516171819202122232425262728
  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 == ("Superuser #%s has been created successfully." % user.pk)
  18. assert user.username == "test"
  19. assert user.email == "test@example.com"
  20. assert user.check_password("password")