maya-chen/flask
1
0
Fork 0
forked from orbit-oss/flask
Code Pull requests Activity

Add coverage for Blueprint teardown request method

Browse source 
Test the following methods in the Blueprint object: teardown_request,
and teardown_app_request.
This PR increases the coverage of blueprint module by 3%.
This commit is contained in:
Randy Liou 2017-05-22 16:54:52 -07:00
parent a690ae27a3
commit d8d712a0de
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_blueprints.py
View file

@ -692,6 +692,9 @@ def test_request_processing():
response.data += b'|after' response.data += b'|after'
evts.append('after') evts.append('after')
return response return response
@bp.teardown_request
def teardown_bp(exc):
evts.append('teardown')
# Setup routes for testing # Setup routes for testing
@bp.route('/bp') @bp.route('/bp')
@ -703,7 +706,7 @@ def test_request_processing():
assert evts == [] assert evts == []
rv = app.test_client().get('/bp') rv = app.test_client().get('/bp')
assert rv.data == b'request|after' assert rv.data == b'request|after'
assert evts == ['before', 'after'] assert evts == ['before', 'after', 'teardown']
def test_app_request_processing(): def test_app_request_processing():
app = flask.Flask(__name__) app = flask.Flask(__name__)
@ -721,6 +724,9 @@ def test_app_request_processing():
response.data += b'|after' response.data += b'|after'
evts.append('after') evts.append('after')
return response return response
@bp.teardown_app_request
def teardown_app(exc):
evts.append('teardown')
app.register_blueprint(bp) app.register_blueprint(bp)
@ -735,9 +741,9 @@ def test_app_request_processing():
# first request # first request
resp = app.test_client().get('/').data resp = app.test_client().get('/').data
assert resp == b'request|after' assert resp == b'request|after'
assert evts == ['first', 'before', 'after'] assert evts == ['first', 'before', 'after', 'teardown']
# second request # second request
resp = app.test_client().get('/').data resp = app.test_client().get('/').data
assert resp == b'request|after' assert resp == b'request|after'
assert evts == ['first', 'before', 'after', 'before', 'after'] assert evts == ['first'] + ['before', 'after', 'teardown'] * 2