Page MenuHome

test_flask_extra.py
No OneTemporary

test_flask_extra.py

import unittest
import flask
class FlaskExtraTest(unittest.TestCase):
def test_vary_xhr(self):
import pillar.flask_extra
class TestApp(flask.Flask):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.add_url_rule('/must-vary', 'must-vary', self.must_vary)
self.add_url_rule('/no-vary', 'no-vary', self.no_vary)
@pillar.flask_extra.vary_xhr()
def must_vary(self):
return 'yay'
def no_vary(self):
return 'nah', 201
app = TestApp(__name__)
client = app.test_client()
resp = client.get('/must-vary')
self.assertEqual(200, resp.status_code)
self.assertEqual('X-Requested-With', resp.headers['Vary'])
self.assertEqual('yay', resp.data.decode())
resp = client.get('/no-vary')
self.assertEqual(201, resp.status_code)
self.assertNotIn('Vary', resp.headers)
self.assertEqual('nah', resp.data.decode())

File Metadata

Mime Type
text/x-python
Expires
Sat, Jul 11, 1:23 AM (1 d, 23 h)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
b3/57/ac8075437cee42642598e9872a03

Event Timeline