|
|
|
@ -353,7 +353,7 @@ class TestSendfile(object):
|
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
|
|
|
|
|
with app.test_request_context(): |
|
|
|
|
f = open(os.path.join(app.root_path, 'static/index.html'), mode='rb') |
|
|
|
|
with open(os.path.join(app.root_path, 'static/index.html'), mode='rb') as f: |
|
|
|
|
rv = flask.send_file(f) |
|
|
|
|
rv.direct_passthrough = False |
|
|
|
|
with app.open_resource('static/index.html') as f: |
|
|
|
@ -368,7 +368,7 @@ class TestSendfile(object):
|
|
|
|
|
app.use_x_sendfile = True |
|
|
|
|
|
|
|
|
|
with app.test_request_context(): |
|
|
|
|
f = open(os.path.join(app.root_path, 'static/index.html')) |
|
|
|
|
with open(os.path.join(app.root_path, 'static/index.html')) as f: |
|
|
|
|
rv = flask.send_file(f) |
|
|
|
|
assert rv.mimetype == 'text/html' |
|
|
|
|
assert 'x-sendfile' in rv.headers |
|
|
|
@ -434,9 +434,10 @@ class TestSendfile(object):
|
|
|
|
|
def test_attachment(self, recwarn): |
|
|
|
|
app = flask.Flask(__name__) |
|
|
|
|
with app.test_request_context(): |
|
|
|
|
f = open(os.path.join(app.root_path, 'static/index.html')) |
|
|
|
|
with open(os.path.join(app.root_path, 'static/index.html')) as f: |
|
|
|
|
rv = flask.send_file(f, as_attachment=True) |
|
|
|
|
value, options = parse_options_header(rv.headers['Content-Disposition']) |
|
|
|
|
value, options = \ |
|
|
|
|
parse_options_header(rv.headers['Content-Disposition']) |
|
|
|
|
assert value == 'attachment' |
|
|
|
|
rv.close() |
|
|
|
|
|
|
|
|
|