dc11cdb4a4
The implementations were moved to Werkzeug, Flask's functions become wrappers around Werkzeug to pass some Flask-specific values. cache_timeout is renamed to max_age. SEND_FILE_MAX_AGE_DEFAULT, app.send_file_max_age_default, and app.get_send_file_max_age defaults to None. This tells the browser to use conditional requests rather than a 12 hour cache. attachment_filename is renamed to download_name, and is always sent if a name is known. Deprecate helpers.safe_join in favor of werkzeug.utils.safe_join. Removed most of the send_file tests, they're tested in Werkzeug. In the file upload example, renamed the uploaded_file view to download_file to avoid a common source of confusion.
78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
import gc
|
|
import platform
|
|
import threading
|
|
|
|
import pytest
|
|
|
|
import flask
|
|
|
|
_gc_lock = threading.Lock()
|
|
|
|
|
|
class assert_no_leak:
|
|
def __enter__(self):
|
|
gc.disable()
|
|
_gc_lock.acquire()
|
|
loc = flask._request_ctx_stack._local
|
|
|
|
# Force Python to track this dictionary at all times.
|
|
# This is necessary since Python only starts tracking
|
|
# dicts if they contain mutable objects. It's a horrible,
|
|
# horrible hack but makes this kinda testable.
|
|
loc.__storage__["FOOO"] = [1, 2, 3]
|
|
|
|
gc.collect()
|
|
self.old_objects = len(gc.get_objects())
|
|
|
|
def __exit__(self, exc_type, exc_value, tb):
|
|
gc.collect()
|
|
new_objects = len(gc.get_objects())
|
|
if new_objects > self.old_objects:
|
|
pytest.fail("Example code leaked")
|
|
_gc_lock.release()
|
|
gc.enable()
|
|
|
|
|
|
@pytest.mark.skipif(platform.python_implementation() == "PyPy", reason="CPython only")
|
|
def test_memory_consumption():
|
|
app = flask.Flask(__name__)
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return flask.render_template("simple_template.html", whiskey=42)
|
|
|
|
def fire():
|
|
with app.test_client() as c:
|
|
rv = c.get("/")
|
|
assert rv.status_code == 200
|
|
assert rv.data == b"<h1>42</h1>"
|
|
|
|
# Trigger caches
|
|
fire()
|
|
|
|
with assert_no_leak():
|
|
for _x in range(10):
|
|
fire()
|
|
|
|
|
|
def test_aborting(app):
|
|
class Foo(Exception):
|
|
whatever = 42
|
|
|
|
@app.errorhandler(Foo)
|
|
def handle_foo(e):
|
|
return str(e.whatever)
|
|
|
|
@app.route("/")
|
|
def index():
|
|
raise flask.abort(flask.redirect(flask.url_for("test")))
|
|
|
|
@app.route("/test")
|
|
def test():
|
|
raise Foo()
|
|
|
|
with app.test_client() as c:
|
|
rv = c.get("/")
|
|
assert rv.headers["Location"] == "http://localhost/test"
|
|
rv = c.get("/test")
|
|
assert rv.data == b"42"
|