mirror of
https://github.com/wassname/flask-security.git
synced 2026-07-05 17:30:14 +08:00
Matt Wright
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
test_configuration
|
|
~~~~~~~~~~~~~~~~~~
|
|
|
|
Basic configuration tests
|
|
"""
|
|
|
|
import base64
|
|
|
|
from utils import authenticate, logout, init_app_with_options
|
|
|
|
|
|
def test_view_configuration(app, sqlalchemy_datastore):
|
|
init_app_with_options(app, sqlalchemy_datastore, **{
|
|
'SECURITY_LOGOUT_URL': '/custom_logout',
|
|
'SECURITY_LOGIN_URL': '/custom_login',
|
|
'SECURITY_POST_LOGIN_VIEW': '/post_login',
|
|
'SECURITY_POST_LOGOUT_VIEW': '/post_logout',
|
|
'SECURITY_DEFAULT_HTTP_AUTH_REALM': 'Custom Realm',
|
|
})
|
|
|
|
client = app.test_client()
|
|
|
|
response = client.get('/custom_login')
|
|
assert b"<h1>Login</h1>" in response.data
|
|
|
|
response = authenticate(client, endpoint='/custom_login', follow_redirects=True)
|
|
assert b'Post Login' in response.data
|
|
|
|
response = logout(client, endpoint='/custom_logout', follow_redirects=True)
|
|
assert b'Post Logout' in response.data
|
|
|
|
response = client.get('/http', headers={
|
|
'Authorization': 'Basic %s' % base64.b64encode(b"joe@lp.com:bogus")
|
|
})
|
|
assert b'<h1>Unauthorized</h1>' in response.data
|
|
assert 'WWW-Authenticate' in response.headers
|
|
assert 'Basic realm="Custom Realm"' == response.headers['WWW-Authenticate']
|
|
|
|
|
|
def test_template_configuration(app, sqlalchemy_datastore):
|
|
init_app_with_options(app, sqlalchemy_datastore, **{
|
|
'SECURITY_LOGIN_USER_TEMPLATE': 'custom_security/login_user.html',
|
|
})
|
|
client = app.test_client()
|
|
response = client.get('/login')
|
|
assert b'CUSTOM LOGIN USER' in response.data
|