|
|
|
|
@ -18,7 +18,7 @@ from werkzeug.http import parse_cache_control_header
|
|
|
|
|
from jinja2 import TemplateNotFound |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_blueprint_specific_error_handling(): |
|
|
|
|
def test_blueprint_specific_error_handling(app, client): |
|
|
|
|
frontend = flask.Blueprint('frontend', __name__) |
|
|
|
|
backend = flask.Blueprint('backend', __name__) |
|
|
|
|
sideend = flask.Blueprint('sideend', __name__) |
|
|
|
|
@ -43,7 +43,6 @@ def test_blueprint_specific_error_handling():
|
|
|
|
|
def sideend_no(): |
|
|
|
|
flask.abort(403) |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(frontend) |
|
|
|
|
app.register_blueprint(backend) |
|
|
|
|
app.register_blueprint(sideend) |
|
|
|
|
@ -52,15 +51,15 @@ def test_blueprint_specific_error_handling():
|
|
|
|
|
def app_forbidden(e): |
|
|
|
|
return 'application itself says no', 403 |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert client.get('/frontend-no').data == b'frontend says no' |
|
|
|
|
assert client.get('/backend-no').data == b'backend says no' |
|
|
|
|
assert client.get('/what-is-a-sideend').data == b'application itself says no' |
|
|
|
|
|
|
|
|
|
assert c.get('/frontend-no').data == b'frontend says no' |
|
|
|
|
assert c.get('/backend-no').data == b'backend says no' |
|
|
|
|
assert c.get('/what-is-a-sideend').data == b'application itself says no' |
|
|
|
|
|
|
|
|
|
def test_blueprint_specific_user_error_handling(): |
|
|
|
|
def test_blueprint_specific_user_error_handling(app, client): |
|
|
|
|
class MyDecoratorException(Exception): |
|
|
|
|
pass |
|
|
|
|
|
|
|
|
|
class MyFunctionException(Exception): |
|
|
|
|
pass |
|
|
|
|
|
|
|
|
|
@ -74,24 +73,48 @@ def test_blueprint_specific_user_error_handling():
|
|
|
|
|
def my_function_exception_handler(e): |
|
|
|
|
assert isinstance(e, MyFunctionException) |
|
|
|
|
return 'bam' |
|
|
|
|
|
|
|
|
|
blue.register_error_handler(MyFunctionException, my_function_exception_handler) |
|
|
|
|
|
|
|
|
|
@blue.route('/decorator') |
|
|
|
|
def blue_deco_test(): |
|
|
|
|
raise MyDecoratorException() |
|
|
|
|
|
|
|
|
|
@blue.route('/function') |
|
|
|
|
def blue_func_test(): |
|
|
|
|
raise MyFunctionException() |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(blue) |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert client.get('/decorator').data == b'boom' |
|
|
|
|
assert client.get('/function').data == b'bam' |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_blueprint_app_error_handling(app, client): |
|
|
|
|
errors = flask.Blueprint('errors', __name__) |
|
|
|
|
|
|
|
|
|
@errors.app_errorhandler(403) |
|
|
|
|
def forbidden_handler(e): |
|
|
|
|
return 'you shall not pass', 403 |
|
|
|
|
|
|
|
|
|
@app.route('/forbidden') |
|
|
|
|
def app_forbidden(): |
|
|
|
|
flask.abort(403) |
|
|
|
|
|
|
|
|
|
forbidden_bp = flask.Blueprint('forbidden_bp', __name__) |
|
|
|
|
|
|
|
|
|
@forbidden_bp.route('/nope') |
|
|
|
|
def bp_forbidden(): |
|
|
|
|
flask.abort(403) |
|
|
|
|
|
|
|
|
|
assert c.get('/decorator').data == b'boom' |
|
|
|
|
assert c.get('/function').data == b'bam' |
|
|
|
|
app.register_blueprint(errors) |
|
|
|
|
app.register_blueprint(forbidden_bp) |
|
|
|
|
|
|
|
|
|
def test_blueprint_url_definitions(): |
|
|
|
|
assert client.get('/forbidden').data == b'you shall not pass' |
|
|
|
|
assert client.get('/nope').data == b'you shall not pass' |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_blueprint_url_definitions(app, client): |
|
|
|
|
bp = flask.Blueprint('test', __name__) |
|
|
|
|
|
|
|
|
|
@bp.route('/foo', defaults={'baz': 42}) |
|
|
|
|
@ -102,17 +125,16 @@ def test_blueprint_url_definitions():
|
|
|
|
|
def bar(bar): |
|
|
|
|
return text_type(bar) |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/1', url_defaults={'bar': 23}) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/2', url_defaults={'bar': 19}) |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert c.get('/1/foo').data == b'23/42' |
|
|
|
|
assert c.get('/2/foo').data == b'19/42' |
|
|
|
|
assert c.get('/1/bar').data == b'23' |
|
|
|
|
assert c.get('/2/bar').data == b'19' |
|
|
|
|
assert client.get('/1/foo').data == b'23/42' |
|
|
|
|
assert client.get('/2/foo').data == b'19/42' |
|
|
|
|
assert client.get('/1/bar').data == b'23' |
|
|
|
|
assert client.get('/2/bar').data == b'19' |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_blueprint_url_processors(): |
|
|
|
|
def test_blueprint_url_processors(app, client): |
|
|
|
|
bp = flask.Blueprint('frontend', __name__, url_prefix='/<lang_code>') |
|
|
|
|
|
|
|
|
|
@bp.url_defaults |
|
|
|
|
@ -131,28 +153,26 @@ def test_blueprint_url_processors():
|
|
|
|
|
def about(): |
|
|
|
|
return flask.url_for('.index') |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp) |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert client.get('/de/').data == b'/de/about' |
|
|
|
|
assert client.get('/de/about').data == b'/de/' |
|
|
|
|
|
|
|
|
|
assert c.get('/de/').data == b'/de/about' |
|
|
|
|
assert c.get('/de/about').data == b'/de/' |
|
|
|
|
|
|
|
|
|
def test_templates_and_static(test_apps): |
|
|
|
|
from blueprintapp import app |
|
|
|
|
c = app.test_client() |
|
|
|
|
client = app.test_client() |
|
|
|
|
|
|
|
|
|
rv = c.get('/') |
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert rv.data == b'Hello from the Frontend' |
|
|
|
|
rv = c.get('/admin/') |
|
|
|
|
rv = client.get('/admin/') |
|
|
|
|
assert rv.data == b'Hello from the Admin' |
|
|
|
|
rv = c.get('/admin/index2') |
|
|
|
|
rv = client.get('/admin/index2') |
|
|
|
|
assert rv.data == b'Hello from the Admin' |
|
|
|
|
rv = c.get('/admin/static/test.txt') |
|
|
|
|
rv = client.get('/admin/static/test.txt') |
|
|
|
|
assert rv.data.strip() == b'Admin File' |
|
|
|
|
rv.close() |
|
|
|
|
rv = c.get('/admin/static/css/test.css') |
|
|
|
|
rv = client.get('/admin/static/css/test.css') |
|
|
|
|
assert rv.data.strip() == b'/* nested file */' |
|
|
|
|
rv.close() |
|
|
|
|
|
|
|
|
|
@ -163,7 +183,7 @@ def test_templates_and_static(test_apps):
|
|
|
|
|
if app.config['SEND_FILE_MAX_AGE_DEFAULT'] == expected_max_age: |
|
|
|
|
expected_max_age = 7200 |
|
|
|
|
app.config['SEND_FILE_MAX_AGE_DEFAULT'] = expected_max_age |
|
|
|
|
rv = c.get('/admin/static/css/test.css') |
|
|
|
|
rv = client.get('/admin/static/css/test.css') |
|
|
|
|
cc = parse_cache_control_header(rv.headers['Cache-Control']) |
|
|
|
|
assert cc.max_age == expected_max_age |
|
|
|
|
rv.close() |
|
|
|
|
@ -181,8 +201,8 @@ def test_templates_and_static(test_apps):
|
|
|
|
|
with flask.Flask(__name__).test_request_context(): |
|
|
|
|
assert flask.render_template('nested/nested.txt') == 'I\'m nested' |
|
|
|
|
|
|
|
|
|
def test_default_static_cache_timeout(): |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
def test_default_static_cache_timeout(app): |
|
|
|
|
class MyBlueprint(flask.Blueprint): |
|
|
|
|
def get_send_file_max_age(self, filename): |
|
|
|
|
return 100 |
|
|
|
|
@ -205,12 +225,14 @@ def test_default_static_cache_timeout():
|
|
|
|
|
finally: |
|
|
|
|
app.config['SEND_FILE_MAX_AGE_DEFAULT'] = max_age_default |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_templates_list(test_apps): |
|
|
|
|
from blueprintapp import app |
|
|
|
|
templates = sorted(app.jinja_env.list_templates()) |
|
|
|
|
assert templates == ['admin/index.html', 'frontend/index.html'] |
|
|
|
|
|
|
|
|
|
def test_dotted_names(): |
|
|
|
|
|
|
|
|
|
def test_dotted_names(app, client): |
|
|
|
|
frontend = flask.Blueprint('myapp.frontend', __name__) |
|
|
|
|
backend = flask.Blueprint('myapp.backend', __name__) |
|
|
|
|
|
|
|
|
|
@ -226,18 +248,15 @@ def test_dotted_names():
|
|
|
|
|
def backend_index(): |
|
|
|
|
return flask.url_for('myapp.frontend.frontend_index') |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(frontend) |
|
|
|
|
app.register_blueprint(backend) |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert c.get('/fe').data.strip() == b'/be' |
|
|
|
|
assert c.get('/fe2').data.strip() == b'/fe' |
|
|
|
|
assert c.get('/be').data.strip() == b'/fe' |
|
|
|
|
assert client.get('/fe').data.strip() == b'/be' |
|
|
|
|
assert client.get('/fe2').data.strip() == b'/fe' |
|
|
|
|
assert client.get('/be').data.strip() == b'/fe' |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_dotted_names_from_app(): |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.testing = True |
|
|
|
|
def test_dotted_names_from_app(app, client): |
|
|
|
|
test = flask.Blueprint('test', __name__) |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
@ -250,11 +269,11 @@ def test_dotted_names_from_app():
|
|
|
|
|
|
|
|
|
|
app.register_blueprint(test) |
|
|
|
|
|
|
|
|
|
with app.test_client() as c: |
|
|
|
|
rv = c.get('/') |
|
|
|
|
assert rv.data == b'/test/' |
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert rv.data == b'/test/' |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_empty_url_defaults(): |
|
|
|
|
def test_empty_url_defaults(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.route('/', defaults={'page': 1}) |
|
|
|
|
@ -262,15 +281,13 @@ def test_empty_url_defaults():
|
|
|
|
|
def something(page): |
|
|
|
|
return str(page) |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp) |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert c.get('/').data == b'1' |
|
|
|
|
assert c.get('/page/2').data == b'2' |
|
|
|
|
assert client.get('/').data == b'1' |
|
|
|
|
assert client.get('/page/2').data == b'2' |
|
|
|
|
|
|
|
|
|
def test_route_decorator_custom_endpoint(): |
|
|
|
|
|
|
|
|
|
def test_route_decorator_custom_endpoint(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.route('/foo') |
|
|
|
|
@ -289,21 +306,20 @@ def test_route_decorator_custom_endpoint():
|
|
|
|
|
def bar_foo(): |
|
|
|
|
return flask.request.endpoint |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.request.endpoint |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert c.get('/').data == b'index' |
|
|
|
|
assert c.get('/py/foo').data == b'bp.foo' |
|
|
|
|
assert c.get('/py/bar').data == b'bp.bar' |
|
|
|
|
assert c.get('/py/bar/123').data == b'bp.123' |
|
|
|
|
assert c.get('/py/bar/foo').data == b'bp.bar_foo' |
|
|
|
|
assert client.get('/').data == b'index' |
|
|
|
|
assert client.get('/py/foo').data == b'bp.foo' |
|
|
|
|
assert client.get('/py/bar').data == b'bp.bar' |
|
|
|
|
assert client.get('/py/bar/123').data == b'bp.123' |
|
|
|
|
assert client.get('/py/bar/foo').data == b'bp.bar_foo' |
|
|
|
|
|
|
|
|
|
def test_route_decorator_custom_endpoint_with_dots(): |
|
|
|
|
|
|
|
|
|
def test_route_decorator_custom_endpoint_with_dots(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.route('/foo') |
|
|
|
|
@ -344,231 +360,461 @@ def test_route_decorator_custom_endpoint_with_dots():
|
|
|
|
|
lambda: None |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
c = app.test_client() |
|
|
|
|
assert c.get('/py/foo').data == b'bp.foo' |
|
|
|
|
assert client.get('/py/foo').data == b'bp.foo' |
|
|
|
|
# The rule's didn't actually made it through |
|
|
|
|
rv = c.get('/py/bar') |
|
|
|
|
rv = client.get('/py/bar') |
|
|
|
|
assert rv.status_code == 404 |
|
|
|
|
rv = c.get('/py/bar/123') |
|
|
|
|
rv = client.get('/py/bar/123') |
|
|
|
|
assert rv.status_code == 404 |
|
|
|
|
|
|
|
|
|
def test_template_filter(): |
|
|
|
|
|
|
|
|
|
def test_endpoint_decorator(app, client): |
|
|
|
|
from werkzeug.routing import Rule |
|
|
|
|
app.url_map.add(Rule('/foo', endpoint='bar')) |
|
|
|
|
|
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.endpoint('bar') |
|
|
|
|
def foobar(): |
|
|
|
|
return flask.request.endpoint |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/bp_prefix') |
|
|
|
|
|
|
|
|
|
assert client.get('/foo').data == b'bar' |
|
|
|
|
assert client.get('/bp_prefix/bar').status_code == 404 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_template_filter(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_filter() |
|
|
|
|
def my_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'my_reverse' in app.jinja_env.filters.keys() |
|
|
|
|
assert app.jinja_env.filters['my_reverse'] == my_reverse |
|
|
|
|
assert app.jinja_env.filters['my_reverse']('abcd') == 'dcba' |
|
|
|
|
|
|
|
|
|
def test_add_template_filter(): |
|
|
|
|
|
|
|
|
|
def test_add_template_filter(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def my_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
|
|
|
|
|
bp.add_app_template_filter(my_reverse) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'my_reverse' in app.jinja_env.filters.keys() |
|
|
|
|
assert app.jinja_env.filters['my_reverse'] == my_reverse |
|
|
|
|
assert app.jinja_env.filters['my_reverse']('abcd') == 'dcba' |
|
|
|
|
|
|
|
|
|
def test_template_filter_with_name(): |
|
|
|
|
|
|
|
|
|
def test_template_filter_with_name(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_filter('strrev') |
|
|
|
|
def my_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'strrev' in app.jinja_env.filters.keys() |
|
|
|
|
assert app.jinja_env.filters['strrev'] == my_reverse |
|
|
|
|
assert app.jinja_env.filters['strrev']('abcd') == 'dcba' |
|
|
|
|
|
|
|
|
|
def test_add_template_filter_with_name(): |
|
|
|
|
|
|
|
|
|
def test_add_template_filter_with_name(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def my_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
|
|
|
|
|
bp.add_app_template_filter(my_reverse, 'strrev') |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'strrev' in app.jinja_env.filters.keys() |
|
|
|
|
assert app.jinja_env.filters['strrev'] == my_reverse |
|
|
|
|
assert app.jinja_env.filters['strrev']('abcd') == 'dcba' |
|
|
|
|
|
|
|
|
|
def test_template_filter_with_template(): |
|
|
|
|
|
|
|
|
|
def test_template_filter_with_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_filter() |
|
|
|
|
def super_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_filter.html', value='abcd') |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert rv.data == b'dcba' |
|
|
|
|
|
|
|
|
|
def test_template_filter_after_route_with_template(): |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
def test_template_filter_after_route_with_template(app, client): |
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_filter.html', value='abcd') |
|
|
|
|
|
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_filter() |
|
|
|
|
def super_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert rv.data == b'dcba' |
|
|
|
|
|
|
|
|
|
def test_add_template_filter_with_template(): |
|
|
|
|
|
|
|
|
|
def test_add_template_filter_with_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def super_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
|
|
|
|
|
bp.add_app_template_filter(super_reverse) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_filter.html', value='abcd') |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert rv.data == b'dcba' |
|
|
|
|
|
|
|
|
|
def test_template_filter_with_name_and_template(): |
|
|
|
|
|
|
|
|
|
def test_template_filter_with_name_and_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_filter('super_reverse') |
|
|
|
|
def my_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_filter.html', value='abcd') |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert rv.data == b'dcba' |
|
|
|
|
|
|
|
|
|
def test_add_template_filter_with_name_and_template(): |
|
|
|
|
|
|
|
|
|
def test_add_template_filter_with_name_and_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def my_reverse(s): |
|
|
|
|
return s[::-1] |
|
|
|
|
|
|
|
|
|
bp.add_app_template_filter(my_reverse, 'super_reverse') |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_filter.html', value='abcd') |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert rv.data == b'dcba' |
|
|
|
|
|
|
|
|
|
def test_template_test(): |
|
|
|
|
|
|
|
|
|
def test_template_test(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_test() |
|
|
|
|
def is_boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'is_boolean' in app.jinja_env.tests.keys() |
|
|
|
|
assert app.jinja_env.tests['is_boolean'] == is_boolean |
|
|
|
|
assert app.jinja_env.tests['is_boolean'](False) |
|
|
|
|
|
|
|
|
|
def test_add_template_test(): |
|
|
|
|
|
|
|
|
|
def test_add_template_test(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def is_boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
|
|
|
|
|
bp.add_app_template_test(is_boolean) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'is_boolean' in app.jinja_env.tests.keys() |
|
|
|
|
assert app.jinja_env.tests['is_boolean'] == is_boolean |
|
|
|
|
assert app.jinja_env.tests['is_boolean'](False) |
|
|
|
|
|
|
|
|
|
def test_template_test_with_name(): |
|
|
|
|
|
|
|
|
|
def test_template_test_with_name(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_test('boolean') |
|
|
|
|
def is_boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'boolean' in app.jinja_env.tests.keys() |
|
|
|
|
assert app.jinja_env.tests['boolean'] == is_boolean |
|
|
|
|
assert app.jinja_env.tests['boolean'](False) |
|
|
|
|
|
|
|
|
|
def test_add_template_test_with_name(): |
|
|
|
|
|
|
|
|
|
def test_add_template_test_with_name(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def is_boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
|
|
|
|
|
bp.add_app_template_test(is_boolean, 'boolean') |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
assert 'boolean' in app.jinja_env.tests.keys() |
|
|
|
|
assert app.jinja_env.tests['boolean'] == is_boolean |
|
|
|
|
assert app.jinja_env.tests['boolean'](False) |
|
|
|
|
|
|
|
|
|
def test_template_test_with_template(): |
|
|
|
|
|
|
|
|
|
def test_template_test_with_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_test() |
|
|
|
|
def boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_test.html', value=False) |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert b'Success!' in rv.data |
|
|
|
|
|
|
|
|
|
def test_template_test_after_route_with_template(): |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
def test_template_test_after_route_with_template(app, client): |
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_test.html', value=False) |
|
|
|
|
|
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_test() |
|
|
|
|
def boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert b'Success!' in rv.data |
|
|
|
|
|
|
|
|
|
def test_add_template_test_with_template(): |
|
|
|
|
|
|
|
|
|
def test_add_template_test_with_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
|
|
|
|
|
bp.add_app_template_test(boolean) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_test.html', value=False) |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert b'Success!' in rv.data |
|
|
|
|
|
|
|
|
|
def test_template_test_with_name_and_template(): |
|
|
|
|
|
|
|
|
|
def test_template_test_with_name_and_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_test('boolean') |
|
|
|
|
def is_boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_test.html', value=False) |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert b'Success!' in rv.data |
|
|
|
|
|
|
|
|
|
def test_add_template_test_with_name_and_template(): |
|
|
|
|
|
|
|
|
|
def test_add_template_test_with_name_and_template(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
def is_boolean(value): |
|
|
|
|
return isinstance(value, bool) |
|
|
|
|
|
|
|
|
|
bp.add_app_template_test(is_boolean, 'boolean') |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
app.register_blueprint(bp, url_prefix='/py') |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.render_template('template_test.html', value=False) |
|
|
|
|
rv = app.test_client().get('/') |
|
|
|
|
|
|
|
|
|
rv = client.get('/') |
|
|
|
|
assert b'Success!' in rv.data |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_context_processing(app, client): |
|
|
|
|
answer_bp = flask.Blueprint('answer_bp', __name__) |
|
|
|
|
|
|
|
|
|
template_string = lambda: flask.render_template_string( |
|
|
|
|
'{% if notanswer %}{{ notanswer }} is not the answer. {% endif %}' |
|
|
|
|
'{% if answer %}{{ answer }} is the answer.{% endif %}' |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
# App global context processor |
|
|
|
|
@answer_bp.app_context_processor |
|
|
|
|
def not_answer_context_processor(): |
|
|
|
|
return {'notanswer': 43} |
|
|
|
|
|
|
|
|
|
# Blueprint local context processor |
|
|
|
|
@answer_bp.context_processor |
|
|
|
|
def answer_context_processor(): |
|
|
|
|
return {'answer': 42} |
|
|
|
|
|
|
|
|
|
# Setup endpoints for testing |
|
|
|
|
@answer_bp.route('/bp') |
|
|
|
|
def bp_page(): |
|
|
|
|
return template_string() |
|
|
|
|
|
|
|
|
|
@app.route('/') |
|
|
|
|
def app_page(): |
|
|
|
|
return template_string() |
|
|
|
|
|
|
|
|
|
# Register the blueprint |
|
|
|
|
app.register_blueprint(answer_bp) |
|
|
|
|
|
|
|
|
|
app_page_bytes = client.get('/').data |
|
|
|
|
answer_page_bytes = client.get('/bp').data |
|
|
|
|
|
|
|
|
|
assert b'43' in app_page_bytes |
|
|
|
|
assert b'42' not in app_page_bytes |
|
|
|
|
|
|
|
|
|
assert b'42' in answer_page_bytes |
|
|
|
|
assert b'43' in answer_page_bytes |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_template_global(app): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
@bp.app_template_global() |
|
|
|
|
def get_answer(): |
|
|
|
|
return 42 |
|
|
|
|
|
|
|
|
|
# Make sure the function is not in the jinja_env already |
|
|
|
|
assert 'get_answer' not in app.jinja_env.globals.keys() |
|
|
|
|
app.register_blueprint(bp) |
|
|
|
|
|
|
|
|
|
# Tests |
|
|
|
|
assert 'get_answer' in app.jinja_env.globals.keys() |
|
|
|
|
assert app.jinja_env.globals['get_answer'] is get_answer |
|
|
|
|
assert app.jinja_env.globals['get_answer']() == 42 |
|
|
|
|
|
|
|
|
|
with app.app_context(): |
|
|
|
|
rv = flask.render_template_string('{{ get_answer() }}') |
|
|
|
|
assert rv == '42' |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_request_processing(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
evts = [] |
|
|
|
|
|
|
|
|
|
@bp.before_request |
|
|
|
|
def before_bp(): |
|
|
|
|
evts.append('before') |
|
|
|
|
|
|
|
|
|
@bp.after_request |
|
|
|
|
def after_bp(response): |
|
|
|
|
response.data += b'|after' |
|
|
|
|
evts.append('after') |
|
|
|
|
return response |
|
|
|
|
|
|
|
|
|
@bp.teardown_request |
|
|
|
|
def teardown_bp(exc): |
|
|
|
|
evts.append('teardown') |
|
|
|
|
|
|
|
|
|
# Setup routes for testing |
|
|
|
|
@bp.route('/bp') |
|
|
|
|
def bp_endpoint(): |
|
|
|
|
return 'request' |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp) |
|
|
|
|
|
|
|
|
|
assert evts == [] |
|
|
|
|
rv = client.get('/bp') |
|
|
|
|
assert rv.data == b'request|after' |
|
|
|
|
assert evts == ['before', 'after', 'teardown'] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_app_request_processing(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
evts = [] |
|
|
|
|
|
|
|
|
|
@bp.before_app_first_request |
|
|
|
|
def before_first_request(): |
|
|
|
|
evts.append('first') |
|
|
|
|
|
|
|
|
|
@bp.before_app_request |
|
|
|
|
def before_app(): |
|
|
|
|
evts.append('before') |
|
|
|
|
|
|
|
|
|
@bp.after_app_request |
|
|
|
|
def after_app(response): |
|
|
|
|
response.data += b'|after' |
|
|
|
|
evts.append('after') |
|
|
|
|
return response |
|
|
|
|
|
|
|
|
|
@bp.teardown_app_request |
|
|
|
|
def teardown_app(exc): |
|
|
|
|
evts.append('teardown') |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp) |
|
|
|
|
|
|
|
|
|
# Setup routes for testing |
|
|
|
|
@app.route('/') |
|
|
|
|
def bp_endpoint(): |
|
|
|
|
return 'request' |
|
|
|
|
|
|
|
|
|
# before first request |
|
|
|
|
assert evts == [] |
|
|
|
|
|
|
|
|
|
# first request |
|
|
|
|
resp = client.get('/').data |
|
|
|
|
assert resp == b'request|after' |
|
|
|
|
assert evts == ['first', 'before', 'after', 'teardown'] |
|
|
|
|
|
|
|
|
|
# second request |
|
|
|
|
resp = client.get('/').data |
|
|
|
|
assert resp == b'request|after' |
|
|
|
|
assert evts == ['first'] + ['before', 'after', 'teardown'] * 2 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_app_url_processors(app, client): |
|
|
|
|
bp = flask.Blueprint('bp', __name__) |
|
|
|
|
|
|
|
|
|
# Register app-wide url defaults and preprocessor on blueprint |
|
|
|
|
@bp.app_url_defaults |
|
|
|
|
def add_language_code(endpoint, values): |
|
|
|
|
values.setdefault('lang_code', flask.g.lang_code) |
|
|
|
|
|
|
|
|
|
@bp.app_url_value_preprocessor |
|
|
|
|
def pull_lang_code(endpoint, values): |
|
|
|
|
flask.g.lang_code = values.pop('lang_code') |
|
|
|
|
|
|
|
|
|
# Register route rules at the app level |
|
|
|
|
@app.route('/<lang_code>/') |
|
|
|
|
def index(): |
|
|
|
|
return flask.url_for('about') |
|
|
|
|
|
|
|
|
|
@app.route('/<lang_code>/about') |
|
|
|
|
def about(): |
|
|
|
|
return flask.url_for('index') |
|
|
|
|
|
|
|
|
|
app.register_blueprint(bp) |
|
|
|
|
|
|
|
|
|
assert client.get('/de/').data == b'/de/about' |
|
|
|
|
assert client.get('/de/about').data == b'/de/' |
|
|
|
|
|
Kenneth Reitz
7 years ago
committed by
GitHub
