diff options
| author | Chris Jerdonek <chris.jerdonek@gmail.com> | 2021-06-23 10:16:40 -0400 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2021-06-29 08:56:05 +0200 |
| commit | 6bccb643476a880d35a6af83aefd903f0a803eff (patch) | |
| tree | 88dcb0cc689f61f09b813ef15dcaa72d6e82dd82 /tests/csrf_tests | |
| parent | 4397d2bd6be1e0eb264b3807f8ae8c0d3cbc116e (diff) | |
Refs #32843 -- Moved _get_GET_csrf_cookie_request() to CsrfViewMiddlewareTestMixin.
Diffstat (limited to 'tests/csrf_tests')
| -rw-r--r-- | tests/csrf_tests/tests.py | 24 |
1 files changed, 11 insertions, 13 deletions
diff --git a/tests/csrf_tests/tests.py b/tests/csrf_tests/tests.py index f3ee9cdedd..65e4a4acfd 100644 --- a/tests/csrf_tests/tests.py +++ b/tests/csrf_tests/tests.py @@ -90,6 +90,9 @@ class CsrfViewMiddlewareTestMixin: _csrf_id_cookie = MASKED_TEST_SECRET1 _csrf_id_token = MASKED_TEST_SECRET2 + def _set_csrf_cookie(self, req, cookie): + raise NotImplementedError('This method must be implemented by a subclass.') + def _get_request(self): req = TestingHttpRequest() req.method = 'GET' @@ -99,7 +102,12 @@ class CsrfViewMiddlewareTestMixin: return self._get_request() def _get_GET_csrf_cookie_request(self, cookie=None): - raise NotImplementedError('This method must be implemented by a subclass.') + """The cookie argument defaults to the valid test cookie.""" + if cookie is None: + cookie = self._csrf_id_cookie + req = self._get_request() + self._set_csrf_cookie(req, cookie) + return req def _get_POST_csrf_cookie_request( self, cookie=None, post_token=None, meta_token=None, token_header=None, @@ -858,13 +866,8 @@ class CsrfViewMiddlewareTestMixin: class CsrfViewMiddlewareTests(CsrfViewMiddlewareTestMixin, SimpleTestCase): - def _get_GET_csrf_cookie_request(self, cookie=None): - """The cookie argument defaults to the valid test cookie.""" - if cookie is None: - cookie = self._csrf_id_cookie - req = self._get_request() + def _set_csrf_cookie(self, req, cookie): req.COOKIES[settings.CSRF_COOKIE_NAME] = cookie - return req def test_ensures_csrf_cookie_no_middleware(self): """ @@ -1068,13 +1071,8 @@ class CsrfViewMiddlewareUseSessionsTests(CsrfViewMiddlewareTestMixin, SimpleTest CSRF tests with CSRF_USE_SESSIONS=True. """ - def _get_GET_csrf_cookie_request(self, cookie=None): - """The cookie argument defaults to the valid test cookie.""" - if cookie is None: - cookie = self._csrf_id_cookie - req = self._get_request() + def _set_csrf_cookie(self, req, cookie): req.session[CSRF_SESSION_KEY] = cookie - return req def test_no_session_on_request(self): msg = ( |
