From 5935ee495c3e51d26fa2b33de09d532b310cd263 Mon Sep 17 00:00:00 2001
From: Mihir Singh <me@mihirsingh.com>
Date: Wed, 16 Apr 2014 17:39:11 -0400
Subject: [PATCH] Add tests for the `Vary: Cookie` header

---
 flask/testsuite/basic.py | 50 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/flask/testsuite/basic.py b/flask/testsuite/basic.py
index bc0838ad..4ab0bbb3 100644
--- a/flask/testsuite/basic.py
+++ b/flask/testsuite/basic.py
@@ -396,6 +396,56 @@ class BasicFunctionalityTestCase(FlaskTestCase):
         app.config['SESSION_REFRESH_EACH_REQUEST'] = False
         run_test(expect_header=False)
 
+    def test_session_vary_cookie(self):
+
+        app = flask.Flask(__name__)
+        app.secret_key = 'testkey'
+
+        @app.route('/set-session')
+        def set_session():
+
+            flask.session['test'] = 'test'
+            return ''
+
+        @app.route('/get-session')
+        def get_session():
+
+            s = flask.session.get('test')
+            return ''
+
+        @app.route('/get-session-with-dictionary')
+        def get_session_with_dictionary():
+
+            s = flask.session['test']
+            return ''
+
+        @app.route('/no-vary-header')
+        def no_vary_header():
+
+            return ''
+
+        c = app.test_client()
+
+        rv = c.get('/set-session')
+
+        self.assert_in('Vary', rv.headers)
+        self.assert_equal('Cookie', rv.headers['Vary'])
+
+        rv = c.get('/get-session')
+
+        self.assert_in('Vary', rv.headers)
+        self.assert_equal('Cookie', rv.headers['Vary'])
+
+        rv = c.get('/get-session-with-dictionary')
+
+        self.assert_in('Vary', rv.headers)
+        self.assert_equal('Cookie', rv.headers['Vary'])
+
+        rv = c.get('/no-vary-header')
+
+        self.assert_not_in('Vary', rv.headers)
+
+
     def test_flashes(self):
         app = flask.Flask(__name__)
         app.secret_key = 'testkey'