maya-chen/flask
1
0
Fork 0
forked from orbit-oss/flask
Code Pull requests Activity

Fixed PyStringIO in tests for 2.x

Browse source
This commit is contained in:
Armin Ronacher 2014-02-09 13:13:31 +00:00
parent 8cb79fd026
commit 001100bc0b
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
flask/testsuite/helpers.py
View file

@ -265,8 +265,11 @@ class SendfileTestCase(FlaskTestCase):
# etags # etags
self.assert_equal(len(captured), 1) self.assert_equal(len(captured), 1)
with catch_warnings() as captured: with catch_warnings() as captured:
class PyStringIO(StringIO): class PyStringIO(object):
pass def __init__(self, *args, **kwargs):
self._io = StringIO(*args, **kwargs)
def __getattr__(self, name):
return getattr(self._io, name)
f = PyStringIO('Test') f = PyStringIO('Test')
f.name = 'test.txt' f.name = 'test.txt'
rv = flask.send_file(f) rv = flask.send_file(f)