From a690ae27a392cfa1474e062f7655250c416cd536 Mon Sep 17 00:00:00 2001
From: Randy Liou <dawran6@gmail.com>
Date: Mon, 22 May 2017 15:49:04 -0700
Subject: [PATCH] Add coverage for Blueprint request process methods

Add test to cover following methodss to the Blueprint object:
before_request, after_request, before_app_request,
before_app_first_request, after_app_request.
This PR increases the coverage of flask.blueprints by 6%.
---
 tests/test_blueprints.py | 62 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/tests/test_blueprints.py b/tests/test_blueprints.py
index 5f0d87da..78da0d2c 100644
--- a/tests/test_blueprints.py
+++ b/tests/test_blueprints.py
@@ -679,3 +679,65 @@ def test_template_global():
     with app.app_context():
         rv = flask.render_template_string('{{ get_answer() }}')
         assert rv == '42'
+
+def test_request_processing():
+    app = flask.Flask(__name__)
+    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
+
+    # Setup routes for testing
+    @bp.route('/bp')
+    def bp_endpoint():
+        return 'request'
+
+    app.register_blueprint(bp)
+
+    assert evts == []
+    rv = app.test_client().get('/bp')
+    assert rv.data == b'request|after'
+    assert evts == ['before', 'after']
+
+def test_app_request_processing():
+    app = flask.Flask(__name__)
+    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
+
+    app.register_blueprint(bp)
+
+    # Setup routes for testing
+    @app.route('/')
+    def bp_endpoint():
+        return 'request'
+
+    # before first request
+    assert evts == []
+
+    # first request
+    resp = app.test_client().get('/').data
+    assert resp == b'request|after'
+    assert evts == ['first', 'before', 'after']
+
+    # second request
+    resp = app.test_client().get('/').data
+    assert resp == b'request|after'
+    assert evts == ['first', 'before', 'after', 'before', 'after']