Added flask.stream_with_context

This commit is contained in:
Armin Ronacher 2012-06-27 15:06:39 +01:00
parent 2e816f554a
commit d5218997d9
8 changed files with 222 additions and 27 deletions

View file

@ -75,6 +75,26 @@ class AppContextTestCase(FlaskTestCase):
self.assert_equal(
flask.render_template_string('{{ g.spam }}'), 'eggs')
def test_context_refcounts(self):
called = []
app = flask.Flask(__name__)
@app.teardown_request
def teardown_req(error=None):
called.append('request')
@app.teardown_appcontext
def teardown_app(error=None):
called.append('app')
@app.route('/')
def index():
with flask._app_ctx_stack.top:
with flask._request_ctx_stack.top:
pass
self.assert_(flask._request_ctx_stack.request.environ
['werkzeug.request'] is not None)
c = app.test_client()
c.get('/')
self.assertEqual(called, ['request', 'app'])
def suite():
suite = unittest.TestSuite()