Advertisement
Guest User

Untitled

a guest
Apr 6th, 2016
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.17 KB | None | 0 0
  1. class User(AbstractBaseUser, PermissionsMixin):
  2. username = models.CharField(max_length=25, unique=True, error_messages={
  3. 'unique': 'The username is taken'
  4. })
  5. first_name = models.CharField(max_length=60, blank=True, null=True)
  6. last_name = models.CharField(max_length=60, blank=True, null=True)
  7. email = models.EmailField(unique=True, db_index=True, error_messages={
  8. 'unique': 'This email id is already registered!'
  9. })
  10.  
  11. is_active = models.BooleanField(default=True)
  12. is_staff = models.BooleanField(default=False)
  13.  
  14. date_joined = models.DateTimeField(auto_now_add=True)
  15.  
  16. USERNAME_FIELD = 'email'
  17. REQUIRED_FIELDS = ['username',]
  18.  
  19.  
  20. objects = UserManager()
  21.  
  22. def get_full_name(self):
  23. return ' '.join([self.first_name, self.last_name])
  24.  
  25. def get_short_name(self):
  26. return self.email
  27.  
  28. def __unicode__(self):
  29. return self.username
  30.  
  31. class SettingsTest(TestCase):
  32. def test_account_is_configured(self):
  33. self.assertTrue('accounts' in INSTALLED_APPS)
  34. self.assertTrue('accounts.User' == AUTH_USER_MODEL)
  35.  
  36.  
  37. class UserTest(TestCase):
  38. def setUp(self):
  39. self.username = "testuser"
  40. self.email = "testuser@testbase.com"
  41. self.first_name = "Test"
  42. self.last_name = "User"
  43. self.password = "z"
  44.  
  45. self.test_user = User.objects.create_user(
  46. username=self.username,
  47. email=self.email,
  48. first_name=self.first_name,
  49. last_name=self.last_name
  50. )
  51.  
  52. def test_create_user(self):
  53. self.assertIsInstance(self.test_user, User)
  54.  
  55. def test_default_user_is_active(self):
  56. self.assertTrue(self.test_user.is_active)
  57.  
  58. def test_default_user_is_staff(self):
  59. self.assertFalse(self.test_user.is_staff)
  60.  
  61. def test_default_user_is_superuser(self):
  62. self.assertFalse(self.test_user.is_superuser)
  63.  
  64. def test_get_full_name(self):
  65. self.assertEqual('Test User', self.test_user.get_full_name())
  66.  
  67. def test_get_short_name(self):
  68. self.assertEqual(self.email, self.test_user.get_short_name())
  69.  
  70. def test_unicode(self):
  71. self.assertEqual(self.username, self.test_user.__unicode__())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement