add get_decoded_jwt_from_cookie

ARCH-603

edx_rest_framework_extensions/__init__.py

@@ -1,3 +1,3 @@
 """ edx Django REST Framework extensions. """
 
-__version__ = '2.1.0'  # pragma: no cover
+__version__ = '2.2.0'  # pragma: no cover

edx_rest_framework_extensions/auth/jwt/cookies.py

@@ -3,6 +3,7 @@
 """
 
 from django.conf import settings
+from edx_rest_framework_extensions.auth.jwt.decoder import jwt_decode_handler
 
 
 def jwt_cookie_name():
@@ -19,3 +20,17 @@ def jwt_cookie_signature_name():
 
 def jwt_refresh_cookie_name():
     return settings.JWT_AUTH.get('JWT_AUTH_REFRESH_COOKIE') or 'edx-jwt-refresh-cookie'
+
+
+def get_decoded_jwt(request):
+    """
+    Grab jwt from jwt cookie in request if possible.
+
+    Returns a decoded jwt dict if it can be found.
+    Returns None if the jwt is not found.
+    """
+    jwt_cookie = request.COOKIES.get(jwt_cookie_name(), None)
+
+    if not jwt_cookie:
+        return None
+    return jwt_decode_handler(jwt_cookie)

edx_rest_framework_extensions/auth/jwt/tests/test_cookies.py

@@ -2,8 +2,12 @@
 Unit tests for jwt cookies module.
 """
 import ddt
+import mock
 from django.test import override_settings, TestCase
 
+from edx_rest_framework_extensions.auth.jwt.decoder import jwt_decode_handler
+from edx_rest_framework_extensions.auth.jwt.tests.utils import generate_latest_version_payload, generate_jwt_token
+from edx_rest_framework_extensions.tests.factories import UserFactory
 from .. import cookies
 
 
@@ -29,3 +33,19 @@ def test_get_setting_value(self, jwt_cookie_func, setting_name, setting_value):
     @ddt.unpack
     def test_get_default_value(self, jwt_cookie_func, expected_default_value):
         self.assertEqual(jwt_cookie_func(), expected_default_value)
+
+    def test_get_decoded_jwt_from_existing_cookie(self):
+        user = UserFactory()
+        payload = generate_latest_version_payload(user)
+        jwt = generate_jwt_token(payload)
+        expected_decoded_jwt = jwt_decode_handler(jwt)
+
+        mock_request_with_cookie = mock.Mock(COOKIES={'edx-jwt-cookie': jwt})
+
+        decoded_jwt = cookies.get_decoded_jwt(mock_request_with_cookie)
+        self.assertEquals(expected_decoded_jwt, decoded_jwt)
+
+    def test_get_decoded_jwt_when_no_cookie(self):
+        mock_request = mock.Mock(COOKIES={})
+
+        self.assertIsNone(cookies.get_decoded_jwt(mock_request))