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

fix teardown bug in FlaskClient

Browse source 
* Fixes pytest-dev/pytest-flask#42
This commit is contained in:
Fantix King 2019-04-16 17:29:17 -05:00 • committed by David Lord
parent 754b729578
commit a71c167836
No known key found for this signature in database
GPG key ID: 7A1C87E3F5BC42A8
3 changed files with 29 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

17
flask/testing.py
View file

@ -206,12 +206,17 @@ class FlaskClient(Client):
def __exit__(self, exc_type, exc_value, tb):
self.preserve_context = False
# on exit we want to clean up earlier. Normally the request context
# stays preserved until the next request in the same thread comes
# in. See RequestGlobals.push() for the general behavior.
top = _request_ctx_stack.top
if top is not None and top.preserved:
top.pop()
# Normally the request context is preserved until the next
# request in the same thread comes. When the client exits we
# want to clean up earlier. Pop request contexts until the stack
# is empty or a non-preserved one is found.
while True:
top = _request_ctx_stack.top
if top is not None and top.preserved:
top.pop()
else:
break
class FlaskCliRunner(CliRunner):