def _assert_recreate_jwt_from_cookies(self, response, can_recreate):
"""
Verifies that a JWT can be properly recreated from the 2 separate
JWT-related cookies using the JwtAuthCookieMiddleware middleware.
"""
self.request.COOKIES = response.cookies
JwtAuthCookieMiddleware().process_request(self.request)
self.assertEqual(
cookies_api.jwt_cookies.jwt_cookie_name() in self.request.COOKIES,
can_recreate,
)
def _assert_recreate_jwt_from_cookies(self, response, can_recreate):
"""
If can_recreate is True, verifies that a JWT can be properly recreated
from the 2 separate JWT-related cookies using the
JwtAuthCookieMiddleware middleware and returns the recreated JWT.
If can_recreate is False, verifies that a JWT cannot be recreated.
"""
self._copy_cookies_to_request(response, self.request)
JwtAuthCookieMiddleware().process_view(self.request, None, None, None)
assert (cookies_api.jwt_cookies.jwt_cookie_name()
in self.request.COOKIES) == can_recreate
if can_recreate:
jwt_string = self.request.COOKIES[
cookies_api.jwt_cookies.jwt_cookie_name()]
jwt = jwt_decode_handler(jwt_string)
assert jwt['scopes'] == ['user_id', 'email', 'profile']
def setUp(self):
super(TestJwtAuthCookieMiddleware, self).setUp()
self.request = RequestFactory().get('/')
self.middleware = JwtAuthCookieMiddleware()
def setUp(self):
super().setUp()
self.request = RequestFactory().get('/')
self.request.session = 'mock session'
self.middleware = JwtAuthCookieMiddleware() # pylint: disable=no-value-for-parameter