diff --git a/docs/index.md b/docs/index.md index 33401d4c..b5300ec4 100644 --- a/docs/index.md +++ b/docs/index.md @@ -218,7 +218,7 @@ Defaults to return the JWT token. Example: ``` -def jwt_response_payload_handler(token, user=None): +def jwt_response_payload_handler(token, user=None, request=None): return { 'token': token, 'user': UserSerializer(user).data diff --git a/rest_framework_jwt/utils.py b/rest_framework_jwt/utils.py index 4e9ee042..74706d7c 100644 --- a/rest_framework_jwt/utils.py +++ b/rest_framework_jwt/utils.py @@ -67,7 +67,7 @@ def jwt_response_payload_handler(token, user=None, request=None): Example: - def jwt_response_payload_handler(token, user=None): + def jwt_response_payload_handler(token, user=None, request=None): return { 'token': token, 'user': UserSerializer(user).data diff --git a/tests/test_utils.py b/tests/test_utils.py index 421876fe..04f58efb 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -53,7 +53,7 @@ def test_jwt_decode(self): def test_jwt_response_payload(self): payload = utils.jwt_payload_handler(self.user) token = utils.jwt_encode_handler(payload) - response_data = utils.jwt_response_payload_handler(token, self.user) + response_data = utils.jwt_response_payload_handler(token) self.assertEqual(response_data, dict(token=token)) diff --git a/tests/utils.py b/tests/utils.py index 4f6c3dba..a6226569 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -6,7 +6,7 @@ def jwt_response_payload_handler(token, user=None, request=None): Example: - def jwt_response_payload_handler(token, user=None): + def jwt_response_payload_handler(token, user=None, request=None): return { 'token': token, 'user': UserSerializer(user).data