Merge pull request #26 from cheerzorpo/user-pk

User ID property in payloads (and tests)

Author: jpadilla

rest_framework_jwt/tests/test_utils.py

@@ -16,7 +16,7 @@ def test_jwt_payload_handler(self):
         payload = utils.jwt_payload_handler(self.user)
 
         self.assertTrue(isinstance(payload, dict))
-        self.assertEqual(payload['user_id'], self.user.id)
+        self.assertEqual(payload['user_id'], self.user.pk)
         self.assertEqual(payload['email'], self.email)
         self.assertEqual(payload['username'], self.username)
         self.assertTrue('exp' in payload)

rest_framework_jwt/utils.py

@@ -6,7 +6,7 @@
 
 def jwt_payload_handler(user):
     return {
-        'user_id': user.id,
+        'user_id': user.pk,
         'email': user.email,
         'username': user.get_username(),
         'exp': datetime.datetime.utcnow() + api_settings.JWT_EXPIRATION_DELTA