From 5978a0f55f8ffd723d552645d70ac92e4e1148e4 Mon Sep 17 00:00:00 2001
From: David Lord <davidism@gmail.com>
Date: Fri, 2 Jun 2017 11:07:53 -0700
Subject: [PATCH] failing test for streaming session

---
 tests/test_helpers.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tests/test_helpers.py b/tests/test_helpers.py
index a2d22dd6..c66e650b 100644
--- a/tests/test_helpers.py
+++ b/tests/test_helpers.py
@@ -861,6 +861,20 @@ class TestStreaming(object):
         assert rv.data == b'Hello World!'
         assert called == [42]
 
+    def test_stream_keeps_session(self, app, client):
+        @app.route('/')
+        def index():
+            flask.session['test'] = 'flask'
+
+            @flask.stream_with_context
+            def gen():
+                yield flask.session['test']
+
+            return flask.Response(gen())
+
+        rv = client.get('/')
+        assert rv.data == b'flask'
+
 
 class TestSafeJoin(object):
     def test_safe_join(self):