Refs #32800 -- Used the cookie argument to CsrfViewMiddlewareTestMixin._get_request() in more tests.

This commit is contained in:
Chris Jerdonek 2021-08-03 02:11:50 -04:00 committed by Carlton Gibson
parent c23aa73626
commit 26d8e3f302
1 changed files with 3 additions and 6 deletions

View File

@ -494,8 +494,7 @@ class CsrfViewMiddlewareTestMixin:
""" """
A new token is sent if the csrf_cookie is the empty string. A new token is sent if the csrf_cookie is the empty string.
""" """
req = self._get_request() req = self._get_request(cookie='')
self._set_csrf_cookie(req, '')
mw = CsrfViewMiddleware(token_view) mw = CsrfViewMiddleware(token_view)
mw.process_view(req, token_view, (), {}) mw.process_view(req, token_view, (), {})
resp = token_view(req) resp = token_view(req)
@ -1094,8 +1093,7 @@ class CsrfViewMiddlewareTests(CsrfViewMiddlewareTestMixin, SimpleTestCase):
If the token is longer than expected, it is ignored and a new token is If the token is longer than expected, it is ignored and a new token is
created. created.
""" """
req = self._get_request() req = self._get_request(cookie='x' * 100000)
self._set_csrf_cookie(req, 'x' * 100000)
mw = CsrfViewMiddleware(token_view) mw = CsrfViewMiddleware(token_view)
mw.process_view(req, token_view, (), {}) mw.process_view(req, token_view, (), {})
resp = mw(req) resp = mw(req)
@ -1108,8 +1106,7 @@ class CsrfViewMiddlewareTests(CsrfViewMiddlewareTestMixin, SimpleTestCase):
new token is created. new token is created.
""" """
token = ('!@#' + self._csrf_id_token)[:CSRF_TOKEN_LENGTH] token = ('!@#' + self._csrf_id_token)[:CSRF_TOKEN_LENGTH]
req = self._get_request() req = self._get_request(cookie=token)
self._set_csrf_cookie(req, token)
mw = CsrfViewMiddleware(token_view) mw = CsrfViewMiddleware(token_view)
mw.process_view(req, token_view, (), {}) mw.process_view(req, token_view, (), {})
resp = mw(req) resp = mw(req)