diff --git a/tests/__init__.py b/tests/__init__.py index 35ceb79..d5e7d56 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -30,7 +30,8 @@ class SecurityTest(TestCase): session['csrf'] = 'csrf_token' csrf_hmac = hmac.new(self.app.config['SECRET_KEY'], - 'csrf_token'.encode('utf8'), digestmod=sha1) + 'csrf_token'.encode('utf8'), + digestmod=sha1) self.csrf_token = '##' + csrf_hmac.hexdigest() def _create_app(self, auth_config, **kwargs): @@ -38,17 +39,17 @@ class SecurityTest(TestCase): def _get(self, route, content_type=None, follow_redirects=None, headers=None): return self.client.get(route, follow_redirects=follow_redirects, - content_type=content_type or 'text/html', - headers=headers) + content_type=content_type or 'text/html', + headers=headers) def _post(self, route, data=None, content_type=None, follow_redirects=True, headers=None): if isinstance(data, dict): data['csrf_token'] = self.csrf_token + content_type = content_type or 'application/x-www-form-urlencoded' return self.client.post(route, data=data, - follow_redirects=follow_redirects, - content_type=content_type or 'application/x-www-form-urlencoded', - headers=headers) + follow_redirects=follow_redirects, + content_type=content_type, headers=headers) def register(self, email, password='password'): data = dict(email=email, password=password, csrf_token=self.csrf_token) diff --git a/tests/configured_tests.py b/tests/configured_tests.py index 7cef0e4..af27be0 100644 --- a/tests/configured_tests.py +++ b/tests/configured_tests.py @@ -122,7 +122,6 @@ class DefaultTemplatePathTests(SecurityTest): 'SECURITY_LOGIN_USER_TEMPLATE': 'custom_security/login_user.html', } - def test_login_user_template(self): r = self._get('/login') @@ -155,9 +154,8 @@ class RecoverableTemplatePathTests(SecurityTest): def test_reset_password_template(self): with capture_reset_password_requests() as requests: - r = self._post('/reset', - data=dict(email='joe@lp.com'), - follow_redirects=True) + r = self._post('/reset', data=dict(email='joe@lp.com'), + follow_redirects=True) t = requests[0]['token'] @@ -365,9 +363,8 @@ class RecoverableTests(SecurityTest): def test_reset_view(self): with capture_reset_password_requests() as requests: - r = self._post('/reset', - data=dict(email='joe@lp.com'), - follow_redirects=True) + r = self._post('/reset', data=dict(email='joe@lp.com'), + follow_redirects=True) t = requests[0]['token'] r = self._get('/reset/' + t) self.assertIn('