mirror of
https://github.com/wassname/flask-security.git
synced 2026-07-14 01:10:34 +08:00
Fix passing category to login required message
This commit is contained in:
@@ -157,10 +157,8 @@ def _get_login_manager(app):
|
||||
lm.login_view = '%s.login' % cv('BLUEPRINT_NAME', app=app)
|
||||
lm.user_loader(_user_loader)
|
||||
lm.token_loader(_token_loader)
|
||||
lm.login_message = cv('MSG_LOGIN', app=app)
|
||||
lm.login_message_category = 'info'
|
||||
lm.needs_refresh_message = cv('MSG_REFRESH', app=app)
|
||||
lm.needs_refresh_message_category = 'info'
|
||||
lm.login_message, lm.login_message_category = cv('MSG_LOGIN', app=app)
|
||||
lm.needs_refresh_message, lm.needs_refresh_message_category = cv('MSG_REFRESH', app=app)
|
||||
lm.init_app(app)
|
||||
return lm
|
||||
|
||||
|
||||
@@ -60,7 +60,7 @@ class DefaultSecurityTests(SecurityTest):
|
||||
|
||||
def test_unauthorized_access(self):
|
||||
r = self._get('/profile', follow_redirects=True)
|
||||
self.assertIn('Please log in to access this page', r.data)
|
||||
self.assertIn('<li class="message">Please log in to access this page.</li>', r.data)
|
||||
|
||||
def test_authorized_access(self):
|
||||
self.authenticate()
|
||||
|
||||
Reference in New Issue
Block a user