Fixed session loading in flask.testing.TestClient.session_transaction()

This commit is contained in:
Michał Bartoszkiewicz 2011-08-29 21:37:52 +02:00
parent 87f50fdc6f
commit 4dc1796b1c
2 changed files with 4 additions and 1 deletions

View file

@ -57,7 +57,7 @@ class FlaskClient(Client):
raise RuntimeError('Session transactions only make sense '
'with cookies enabled.')
app = self.application
environ_overrides = kwargs.pop('environ_overrides', {})
environ_overrides = kwargs.setdefault('environ_overrides', {})
self.cookie_jar.inject_wsgi(environ_overrides)
outer_reqctx = _request_ctx_stack.top
with app.test_request_context(*args, **kwargs) as c:

View file

@ -59,6 +59,9 @@ class TestToolsTestCase(FlaskTestCase):
self.assert_equal(len(sess), 1)
rv = c.get('/')
self.assert_equal(rv.data, '[42]')
with c.session_transaction() as sess:
self.assert_equal(len(sess), 1)
self.assert_equal(sess['foo'], [42])
def test_session_transactions_no_null_sessions(self):
app = flask.Flask(__name__)