Browse Source

Add coverage for Blueprint.add_app_template_global

This tests the Blueprint.add_app_template_global mothod, which internally
calls the Blueprint.app_template_global method. The methods are used to
registering a function to the jinja template environment.
This PR increases the test coverage for module flask.blueprint by 4%.
pull/2299/head
Randy Liou 8 years ago
parent
commit
7ce01ab9b4
  1. 19
      tests/test_blueprints.py

19
tests/test_blueprints.py

@ -660,3 +660,22 @@ def test_context_processing():
assert b'42' in answer_page_bytes assert b'42' in answer_page_bytes
assert b'43' in answer_page_bytes assert b'43' in answer_page_bytes
def test_template_global():
app = flask.Flask(__name__)
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'

Loading…
Cancel
Save