test_basic.py 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. from __future__ import unicode_literals
  2. from django.apps import apps
  3. from django.contrib.auth import get_user_model
  4. from django.contrib.auth.models import User, AnonymousUser
  5. from django.contrib.auth.tests.custom_user import CustomUser
  6. from django.contrib.auth.tests.utils import skipIfCustomUser
  7. from django.core.exceptions import ImproperlyConfigured
  8. from django.dispatch import receiver
  9. from django.test import TestCase, override_settings
  10. from django.test.signals import setting_changed
  11. from django.utils import translation
  12. @receiver(setting_changed)
  13. def user_model_swapped(**kwargs):
  14. if kwargs['setting'] == 'AUTH_USER_MODEL':
  15. from django.db.models.manager import ensure_default_manager
  16. # Reset User manager
  17. setattr(User, 'objects', User._default_manager)
  18. ensure_default_manager(User)
  19. apps.clear_cache()
  20. @skipIfCustomUser
  21. class BasicTestCase(TestCase):
  22. def test_user(self):
  23. "Check that users can be created and can set their password"
  24. u = User.objects.create_user('testuser', 'test@example.com', 'testpw')
  25. self.assertTrue(u.has_usable_password())
  26. self.assertFalse(u.check_password('bad'))
  27. self.assertTrue(u.check_password('testpw'))
  28. # Check we can manually set an unusable password
  29. u.set_unusable_password()
  30. u.save()
  31. self.assertFalse(u.check_password('testpw'))
  32. self.assertFalse(u.has_usable_password())
  33. u.set_password('testpw')
  34. self.assertTrue(u.check_password('testpw'))
  35. u.set_password(None)
  36. self.assertFalse(u.has_usable_password())
  37. # Check authentication/permissions
  38. self.assertTrue(u.is_authenticated())
  39. self.assertFalse(u.is_staff)
  40. self.assertTrue(u.is_active)
  41. self.assertFalse(u.is_superuser)
  42. # Check API-based user creation with no password
  43. u2 = User.objects.create_user('testuser2', 'test2@example.com')
  44. self.assertFalse(u2.has_usable_password())
  45. def test_user_no_email(self):
  46. "Check that users can be created without an email"
  47. u = User.objects.create_user('testuser1')
  48. self.assertEqual(u.email, '')
  49. u2 = User.objects.create_user('testuser2', email='')
  50. self.assertEqual(u2.email, '')
  51. u3 = User.objects.create_user('testuser3', email=None)
  52. self.assertEqual(u3.email, '')
  53. def test_anonymous_user(self):
  54. "Check the properties of the anonymous user"
  55. a = AnonymousUser()
  56. self.assertEqual(a.pk, None)
  57. self.assertFalse(a.is_authenticated())
  58. self.assertFalse(a.is_staff)
  59. self.assertFalse(a.is_active)
  60. self.assertFalse(a.is_superuser)
  61. self.assertEqual(a.groups.all().count(), 0)
  62. self.assertEqual(a.user_permissions.all().count(), 0)
  63. def test_superuser(self):
  64. "Check the creation and properties of a superuser"
  65. super = User.objects.create_superuser('super', 'super@example.com', 'super')
  66. self.assertTrue(super.is_superuser)
  67. self.assertTrue(super.is_active)
  68. self.assertTrue(super.is_staff)
  69. def test_get_user_model(self):
  70. "The current user model can be retrieved"
  71. self.assertEqual(get_user_model(), User)
  72. @override_settings(AUTH_USER_MODEL='auth.CustomUser')
  73. def test_swappable_user(self):
  74. "The current user model can be swapped out for another"
  75. self.assertEqual(get_user_model(), CustomUser)
  76. with self.assertRaises(AttributeError):
  77. User.objects.all()
  78. @override_settings(AUTH_USER_MODEL='badsetting')
  79. def test_swappable_user_bad_setting(self):
  80. "The alternate user setting must point to something in the format app.model"
  81. with self.assertRaises(ImproperlyConfigured):
  82. get_user_model()
  83. @override_settings(AUTH_USER_MODEL='thismodel.doesntexist')
  84. def test_swappable_user_nonexistent_model(self):
  85. "The current user model must point to an installed model"
  86. with self.assertRaises(ImproperlyConfigured):
  87. get_user_model()
  88. @skipIfCustomUser
  89. def test_user_verbose_names_translatable(self):
  90. "Default User model verbose names are translatable (#19945)"
  91. with translation.override('en'):
  92. self.assertEqual(User._meta.verbose_name, 'user')
  93. self.assertEqual(User._meta.verbose_name_plural, 'users')
  94. with translation.override('es'):
  95. self.assertEqual(User._meta.verbose_name, 'usuario')
  96. self.assertEqual(User._meta.verbose_name_plural, 'usuarios')