Merge branch 'master' into vary-cookies

tests/conftest.py

@@ -0,0 +1,134 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.conftest
+    ~~~~~~~~~~~~~~
+
+    :copyright: (c) 2015 by the Flask Team, see AUTHORS for more details.
+    :license: BSD, see LICENSE for more details.
+"""
+import flask
+import gc
+import os
+import sys
+import pkgutil
+import pytest
+import textwrap
+
+
+@pytest.fixture
+def test_apps(monkeypatch):
+    monkeypatch.syspath_prepend(
+        os.path.abspath(os.path.join(
+            os.path.dirname(__file__), 'test_apps'))
+    )
+
+@pytest.fixture(autouse=True)
+def leak_detector(request):
+    def ensure_clean_request_context():
+        # make sure we're not leaking a request context since we are
+        # testing flask internally in debug mode in a few cases
+        leaks = []
+        while flask._request_ctx_stack.top is not None:
+            leaks.append(flask._request_ctx_stack.pop())
+        assert leaks == []
+    request.addfinalizer(ensure_clean_request_context)
+
+
+@pytest.fixture(params=(True, False))
+def limit_loader(request, monkeypatch):
+    """Patch pkgutil.get_loader to give loader without get_filename or archive.
+
+    This provides for tests where a system has custom loaders, e.g. Google App
+    Engine's HardenedModulesHook, which have neither the `get_filename` method
+    nor the `archive` attribute.
+
+    This fixture will run the testcase twice, once with and once without the
+    limitation/mock.
+    """
+    if not request.param:
+        return
+
+    class LimitedLoader(object):
+        def __init__(self, loader):
+            self.loader = loader
+
+        def __getattr__(self, name):
+            if name in ('archive', 'get_filename'):
+                msg = 'Mocking a loader which does not have `%s.`' % name
+                raise AttributeError(msg)
+            return getattr(self.loader, name)
+
+    old_get_loader = pkgutil.get_loader
+
+    def get_loader(*args, **kwargs):
+        return LimitedLoader(old_get_loader(*args, **kwargs))
+    monkeypatch.setattr(pkgutil, 'get_loader', get_loader)
+
+
+@pytest.fixture
+def modules_tmpdir(tmpdir, monkeypatch):
+    '''A tmpdir added to sys.path'''
+    rv = tmpdir.mkdir('modules_tmpdir')
+    monkeypatch.syspath_prepend(str(rv))
+    return rv
+
+
+@pytest.fixture
+def modules_tmpdir_prefix(modules_tmpdir, monkeypatch):
+    monkeypatch.setattr(sys, 'prefix', str(modules_tmpdir))
+    return modules_tmpdir
+
+
+@pytest.fixture
+def site_packages(modules_tmpdir, monkeypatch):
+    '''Create a fake site-packages'''
+    rv = modules_tmpdir \
+        .mkdir('lib')\
+        .mkdir('python{x[0]}.{x[1]}'.format(x=sys.version_info))\
+        .mkdir('site-packages')
+    monkeypatch.syspath_prepend(str(rv))
+    return rv
+
+
+@pytest.fixture
+def install_egg(modules_tmpdir, monkeypatch):
+    '''Generate egg from package name inside base and put the egg into
+    sys.path'''
+    def inner(name, base=modules_tmpdir):
+        if not isinstance(name, str):
+            raise ValueError(name)
+        base.join(name).ensure_dir()
+        base.join(name).join('__init__.py').ensure()
+
+        egg_setup = base.join('setup.py')
+        egg_setup.write(textwrap.dedent("""
+        from setuptools import setup
+        setup(name='{0}',
+              version='1.0',
+              packages=['site_egg'],
+              zip_safe=True)
+        """.format(name)))
+
+        import subprocess
+        subprocess.check_call(
+            [sys.executable, 'setup.py', 'bdist_egg'],
+            cwd=str(modules_tmpdir)
+        )
+        egg_path, = modules_tmpdir.join('dist/').listdir()
+        monkeypatch.syspath_prepend(str(egg_path))
+        return egg_path
+    return inner
+
+
+@pytest.fixture
+def purge_module(request):
+    def inner(name):
+        request.addfinalizer(lambda: sys.modules.pop(name, None))
+    return inner
+
+
+@pytest.yield_fixture(autouse=True)
+def catch_deprecation_warnings(recwarn):
+    yield
+    gc.collect()
+    assert not recwarn.list

tests/static/config.json

@@ -0,0 +1,4 @@
+{
+    "TEST_KEY": "foo",
+    "SECRET_KEY": "devkey"
+}

tests/static/index.html

@@ -0,0 +1 @@
+<h1>Hello World!</h1>

tests/templates/_macro.html

@@ -0,0 +1 @@
+{% macro hello(name) %}Hello {{ name }}!{% endmacro %}

tests/templates/context_template.html

@@ -0,0 +1 @@
+<p>{{ value }}|{{ injected_value }}

tests/templates/escaping_template.html

@@ -0,0 +1,6 @@
+{{ text }}
+{{ html }}
+{% autoescape false %}{{ text }}
+{{ html }}{% endautoescape %}
+{% autoescape true %}{{ text }}
+{{ html }}{% endautoescape %}

tests/templates/mail.txt

@@ -0,0 +1 @@
+{{ foo}} Mail

tests/templates/nested/nested.txt

@@ -0,0 +1 @@
+I'm nested

tests/templates/non_escaping_template.txt

@@ -0,0 +1,8 @@
+{{ text }}
+{{ html }}
+{% autoescape false %}{{ text }}
+{{ html }}{% endautoescape %}
+{% autoescape true %}{{ text }}
+{{ html }}{% endautoescape %}
+{{ text }}
+{{ html }}

tests/templates/simple_template.html

@@ -0,0 +1 @@
+<h1>{{ whiskey }}</h1>

tests/templates/template_filter.html

@@ -0,0 +1 @@
+{{ value|super_reverse }}

tests/templates/template_test.html

@@ -0,0 +1,3 @@
+{% if value is boolean %}
+    Success!
+{% endif %}

tests/test_appctx.py

@@ -0,0 +1,170 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.appctx
+    ~~~~~~~~~~~~
+
+    Tests the application context.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import flask
+
+
+def test_basic_url_generation():
+    app = flask.Flask(__name__)
+    app.config['SERVER_NAME'] = 'localhost'
+    app.config['PREFERRED_URL_SCHEME'] = 'https'
+
+    @app.route('/')
+    def index():
+        pass
+
+    with app.app_context():
+        rv = flask.url_for('index')
+        assert rv == 'https://localhost/'
+
+def test_url_generation_requires_server_name():
+    app = flask.Flask(__name__)
+    with app.app_context():
+        with pytest.raises(RuntimeError):
+            flask.url_for('index')
+
+def test_url_generation_without_context_fails():
+    with pytest.raises(RuntimeError):
+        flask.url_for('index')
+
+def test_request_context_means_app_context():
+    app = flask.Flask(__name__)
+    with app.test_request_context():
+        assert flask.current_app._get_current_object() == app
+    assert flask._app_ctx_stack.top is None
+
+def test_app_context_provides_current_app():
+    app = flask.Flask(__name__)
+    with app.app_context():
+        assert flask.current_app._get_current_object() == app
+    assert flask._app_ctx_stack.top is None
+
+def test_app_tearing_down():
+    cleanup_stuff = []
+    app = flask.Flask(__name__)
+    @app.teardown_appcontext
+    def cleanup(exception):
+        cleanup_stuff.append(exception)
+
+    with app.app_context():
+        pass
+
+    assert cleanup_stuff == [None]
+
+def test_app_tearing_down_with_previous_exception():
+    cleanup_stuff = []
+    app = flask.Flask(__name__)
+    @app.teardown_appcontext
+    def cleanup(exception):
+        cleanup_stuff.append(exception)
+
+    try:
+        raise Exception('dummy')
+    except Exception:
+        pass
+
+    with app.app_context():
+        pass
+
+    assert cleanup_stuff == [None]
+
+def test_app_tearing_down_with_handled_exception():
+    cleanup_stuff = []
+    app = flask.Flask(__name__)
+    @app.teardown_appcontext
+    def cleanup(exception):
+        cleanup_stuff.append(exception)
+
+    with app.app_context():
+        try:
+            raise Exception('dummy')
+        except Exception:
+            pass
+
+    assert cleanup_stuff == [None]
+
+def test_app_ctx_globals_methods():
+    app = flask.Flask(__name__)
+    with app.app_context():
+        # get
+        assert flask.g.get('foo') is None
+        assert flask.g.get('foo', 'bar') == 'bar'
+        # __contains__
+        assert 'foo' not in flask.g
+        flask.g.foo = 'bar'
+        assert 'foo' in flask.g
+        # setdefault
+        flask.g.setdefault('bar', 'the cake is a lie')
+        flask.g.setdefault('bar', 'hello world')
+        assert flask.g.bar == 'the cake is a lie'
+        # pop
+        assert flask.g.pop('bar') == 'the cake is a lie'
+        with pytest.raises(KeyError):
+            flask.g.pop('bar')
+        assert flask.g.pop('bar', 'more cake') == 'more cake'
+        # __iter__
+        assert list(flask.g) == ['foo']
+
+def test_custom_app_ctx_globals_class():
+    class CustomRequestGlobals(object):
+        def __init__(self):
+            self.spam = 'eggs'
+    app = flask.Flask(__name__)
+    app.app_ctx_globals_class = CustomRequestGlobals
+    with app.app_context():
+        assert flask.render_template_string('{{ g.spam }}') == 'eggs'
+
+def test_context_refcounts():
+    called = []
+    app = flask.Flask(__name__)
+    @app.teardown_request
+    def teardown_req(error=None):
+        called.append('request')
+    @app.teardown_appcontext
+    def teardown_app(error=None):
+        called.append('app')
+    @app.route('/')
+    def index():
+        with flask._app_ctx_stack.top:
+            with flask._request_ctx_stack.top:
+                pass
+        env = flask._request_ctx_stack.top.request.environ
+        assert env['werkzeug.request'] is not None
+        return u''
+    c = app.test_client()
+    res = c.get('/')
+    assert res.status_code == 200
+    assert res.data == b''
+    assert called == ['request', 'app']
+
+
+def test_clean_pop():
+    called = []
+    app = flask.Flask(__name__)
+
+    @app.teardown_request
+    def teardown_req(error=None):
+        1 / 0
+
+    @app.teardown_appcontext
+    def teardown_app(error=None):
+        called.append('TEARDOWN')
+
+    try:
+        with app.test_request_context():
+            called.append(flask.current_app.name)
+    except ZeroDivisionError:
+        pass
+
+    assert called == ['test_appctx', 'TEARDOWN']
+    assert not flask.current_app

tests/test_apps/blueprintapp/__init__.py

@@ -0,0 +1,8 @@
+from flask import Flask
+
+app = Flask(__name__)
+app.config['DEBUG'] = True
+from blueprintapp.apps.admin import admin
+from blueprintapp.apps.frontend import frontend
+app.register_blueprint(admin)
+app.register_blueprint(frontend)

tests/test_apps/blueprintapp/apps/admin/__init__.py

@@ -0,0 +1,15 @@
+from flask import Blueprint, render_template
+
+admin = Blueprint('admin', __name__, url_prefix='/admin',
+                  template_folder='templates',
+                  static_folder='static')
+
+
+@admin.route('/')
+def index():
+    return render_template('admin/index.html')
+
+
+@admin.route('/index2')
+def index2():
+    return render_template('./admin/index.html')

tests/test_apps/blueprintapp/apps/admin/static/css/test.css

@@ -0,0 +1 @@
+/* nested file */

tests/test_apps/blueprintapp/apps/admin/static/test.txt

@@ -0,0 +1 @@
+Admin File

tests/test_apps/blueprintapp/apps/admin/templates/admin/index.html

@@ -0,0 +1 @@
+Hello from the Admin

tests/test_apps/blueprintapp/apps/frontend/__init__.py

@@ -0,0 +1,13 @@
+from flask import Blueprint, render_template
+
+frontend = Blueprint('frontend', __name__, template_folder='templates')
+
+
+@frontend.route('/')
+def index():
+    return render_template('frontend/index.html')
+
+
+@frontend.route('/missing')
+def missing_template():
+    return render_template('missing_template.html')

tests/test_apps/blueprintapp/apps/frontend/templates/frontend/index.html

@@ -0,0 +1 @@
+Hello from the Frontend

tests/test_apps/cliapp/app.py

@@ -0,0 +1,5 @@
+from __future__ import absolute_import, print_function
+
+from flask import Flask
+
+testapp = Flask('testapp')

tests/test_apps/cliapp/importerrorapp.py

@@ -0,0 +1,7 @@
+from __future__ import absolute_import, print_function
+
+from flask import Flask
+
+raise ImportError()
+
+testapp = Flask('testapp')

tests/test_apps/cliapp/multiapp.py

@@ -0,0 +1,6 @@
+from __future__ import absolute_import, print_function
+
+from flask import Flask
+
+app1 = Flask('app1')
+app2 = Flask('app2')

tests/test_apps/subdomaintestmodule/__init__.py

@@ -0,0 +1,4 @@
+from flask import Module
+
+
+mod = Module(__name__, 'foo', subdomain='foo')

tests/test_apps/subdomaintestmodule/static/hello.txt

@@ -0,0 +1 @@
+Hello Subdomain

tests/test_blueprints.py

@@ -0,0 +1,593 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.blueprints
+    ~~~~~~~~~~~~~~~~
+
+    Blueprints (and currently modules)
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import flask
+
+from flask._compat import text_type
+from werkzeug.http import parse_cache_control_header
+from jinja2 import TemplateNotFound
+
+
+def test_blueprint_specific_error_handling():
+    frontend = flask.Blueprint('frontend', __name__)
+    backend = flask.Blueprint('backend', __name__)
+    sideend = flask.Blueprint('sideend', __name__)
+
+    @frontend.errorhandler(403)
+    def frontend_forbidden(e):
+        return 'frontend says no', 403
+
+    @frontend.route('/frontend-no')
+    def frontend_no():
+        flask.abort(403)
+
+    @backend.errorhandler(403)
+    def backend_forbidden(e):
+        return 'backend says no', 403
+
+    @backend.route('/backend-no')
+    def backend_no():
+        flask.abort(403)
+
+    @sideend.route('/what-is-a-sideend')
+    def sideend_no():
+        flask.abort(403)
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(frontend)
+    app.register_blueprint(backend)
+    app.register_blueprint(sideend)
+
+    @app.errorhandler(403)
+    def app_forbidden(e):
+        return 'application itself says no', 403
+
+    c = app.test_client()
+
+    assert c.get('/frontend-no').data == b'frontend says no'
+    assert c.get('/backend-no').data == b'backend says no'
+    assert c.get('/what-is-a-sideend').data == b'application itself says no'
+
+def test_blueprint_specific_user_error_handling():
+    class MyDecoratorException(Exception):
+        pass
+    class MyFunctionException(Exception):
+        pass
+
+    blue = flask.Blueprint('blue', __name__)
+
+    @blue.errorhandler(MyDecoratorException)
+    def my_decorator_exception_handler(e):
+        assert isinstance(e, MyDecoratorException)
+        return 'boom'
+
+    def my_function_exception_handler(e):
+        assert isinstance(e, MyFunctionException)
+        return 'bam'
+    blue.register_error_handler(MyFunctionException, my_function_exception_handler)
+
+    @blue.route('/decorator')
+    def blue_deco_test():
+        raise MyDecoratorException()
+    @blue.route('/function')
+    def blue_func_test():
+        raise MyFunctionException()
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(blue)
+
+    c = app.test_client()
+
+    assert c.get('/decorator').data == b'boom'
+    assert c.get('/function').data == b'bam'
+
+def test_blueprint_url_definitions():
+    bp = flask.Blueprint('test', __name__)
+
+    @bp.route('/foo', defaults={'baz': 42})
+    def foo(bar, baz):
+        return '%s/%d' % (bar, baz)
+
+    @bp.route('/bar')
+    def bar(bar):
+        return text_type(bar)
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/1', url_defaults={'bar': 23})
+    app.register_blueprint(bp, url_prefix='/2', url_defaults={'bar': 19})
+
+    c = app.test_client()
+    assert c.get('/1/foo').data == b'23/42'
+    assert c.get('/2/foo').data == b'19/42'
+    assert c.get('/1/bar').data == b'23'
+    assert c.get('/2/bar').data == b'19'
+
+def test_blueprint_url_processors():
+    bp = flask.Blueprint('frontend', __name__, url_prefix='/<lang_code>')
+
+    @bp.url_defaults
+    def add_language_code(endpoint, values):
+        values.setdefault('lang_code', flask.g.lang_code)
+
+    @bp.url_value_preprocessor
+    def pull_lang_code(endpoint, values):
+        flask.g.lang_code = values.pop('lang_code')
+
+    @bp.route('/')
+    def index():
+        return flask.url_for('.about')
+
+    @bp.route('/about')
+    def about():
+        return flask.url_for('.index')
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp)
+
+    c = app.test_client()
+
+    assert c.get('/de/').data == b'/de/about'
+    assert c.get('/de/about').data == b'/de/'
+
+def test_templates_and_static(test_apps):
+    from blueprintapp import app
+    c = app.test_client()
+
+    rv = c.get('/')
+    assert rv.data == b'Hello from the Frontend'
+    rv = c.get('/admin/')
+    assert rv.data == b'Hello from the Admin'
+    rv = c.get('/admin/index2')
+    assert rv.data == b'Hello from the Admin'
+    rv = c.get('/admin/static/test.txt')
+    assert rv.data.strip() == b'Admin File'
+    rv.close()
+    rv = c.get('/admin/static/css/test.css')
+    assert rv.data.strip() == b'/* nested file */'
+    rv.close()
+
+    # try/finally, in case other tests use this app for Blueprint tests.
+    max_age_default = app.config['SEND_FILE_MAX_AGE_DEFAULT']
+    try:
+        expected_max_age = 3600
+        if app.config['SEND_FILE_MAX_AGE_DEFAULT'] == expected_max_age:
+            expected_max_age = 7200
+        app.config['SEND_FILE_MAX_AGE_DEFAULT'] = expected_max_age
+        rv = c.get('/admin/static/css/test.css')
+        cc = parse_cache_control_header(rv.headers['Cache-Control'])
+        assert cc.max_age == expected_max_age
+        rv.close()
+    finally:
+        app.config['SEND_FILE_MAX_AGE_DEFAULT'] = max_age_default
+
+    with app.test_request_context():
+        assert flask.url_for('admin.static', filename='test.txt') == '/admin/static/test.txt'
+
+    with app.test_request_context():
+        with pytest.raises(TemplateNotFound) as e:
+            flask.render_template('missing.html')
+        assert e.value.name == 'missing.html'
+
+    with flask.Flask(__name__).test_request_context():
+        assert flask.render_template('nested/nested.txt') == 'I\'m nested'
+
+def test_default_static_cache_timeout():
+    app = flask.Flask(__name__)
+    class MyBlueprint(flask.Blueprint):
+        def get_send_file_max_age(self, filename):
+            return 100
+
+    blueprint = MyBlueprint('blueprint', __name__, static_folder='static')
+    app.register_blueprint(blueprint)
+
+    # try/finally, in case other tests use this app for Blueprint tests.
+    max_age_default = app.config['SEND_FILE_MAX_AGE_DEFAULT']
+    try:
+        with app.test_request_context():
+            unexpected_max_age = 3600
+            if app.config['SEND_FILE_MAX_AGE_DEFAULT'] == unexpected_max_age:
+                unexpected_max_age = 7200
+            app.config['SEND_FILE_MAX_AGE_DEFAULT'] = unexpected_max_age
+            rv = blueprint.send_static_file('index.html')
+            cc = parse_cache_control_header(rv.headers['Cache-Control'])
+            assert cc.max_age == 100
+            rv.close()
+    finally:
+        app.config['SEND_FILE_MAX_AGE_DEFAULT'] = max_age_default
+
+def test_templates_list(test_apps):
+    from blueprintapp import app
+    templates = sorted(app.jinja_env.list_templates())
+    assert templates == ['admin/index.html', 'frontend/index.html']
+
+def test_dotted_names():
+    frontend = flask.Blueprint('myapp.frontend', __name__)
+    backend = flask.Blueprint('myapp.backend', __name__)
+
+    @frontend.route('/fe')
+    def frontend_index():
+        return flask.url_for('myapp.backend.backend_index')
+
+    @frontend.route('/fe2')
+    def frontend_page2():
+        return flask.url_for('.frontend_index')
+
+    @backend.route('/be')
+    def backend_index():
+        return flask.url_for('myapp.frontend.frontend_index')
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(frontend)
+    app.register_blueprint(backend)
+
+    c = app.test_client()
+    assert c.get('/fe').data.strip() == b'/be'
+    assert c.get('/fe2').data.strip() == b'/fe'
+    assert c.get('/be').data.strip() == b'/fe'
+
+def test_dotted_names_from_app():
+    app = flask.Flask(__name__)
+    app.testing = True
+    test = flask.Blueprint('test', __name__)
+
+    @app.route('/')
+    def app_index():
+        return flask.url_for('test.index')
+
+    @test.route('/test/')
+    def index():
+        return flask.url_for('app_index')
+
+    app.register_blueprint(test)
+
+    with app.test_client() as c:
+        rv = c.get('/')
+        assert rv.data == b'/test/'
+
+def test_empty_url_defaults():
+    bp = flask.Blueprint('bp', __name__)
+
+    @bp.route('/', defaults={'page': 1})
+    @bp.route('/page/<int:page>')
+    def something(page):
+        return str(page)
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp)
+
+    c = app.test_client()
+    assert c.get('/').data == b'1'
+    assert c.get('/page/2').data == b'2'
+
+def test_route_decorator_custom_endpoint():
+
+    bp = flask.Blueprint('bp', __name__)
+
+    @bp.route('/foo')
+    def foo():
+        return flask.request.endpoint
+
+    @bp.route('/bar', endpoint='bar')
+    def foo_bar():
+        return flask.request.endpoint
+
+    @bp.route('/bar/123', endpoint='123')
+    def foo_bar_foo():
+        return flask.request.endpoint
+
+    @bp.route('/bar/foo')
+    def bar_foo():
+        return flask.request.endpoint
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+
+    @app.route('/')
+    def index():
+        return flask.request.endpoint
+
+    c = app.test_client()
+    assert c.get('/').data == b'index'
+    assert c.get('/py/foo').data == b'bp.foo'
+    assert c.get('/py/bar').data == b'bp.bar'
+    assert c.get('/py/bar/123').data == b'bp.123'
+    assert c.get('/py/bar/foo').data == b'bp.bar_foo'
+
+def test_route_decorator_custom_endpoint_with_dots():
+    bp = flask.Blueprint('bp', __name__)
+
+    @bp.route('/foo')
+    def foo():
+        return flask.request.endpoint
+
+    try:
+        @bp.route('/bar', endpoint='bar.bar')
+        def foo_bar():
+            return flask.request.endpoint
+    except AssertionError:
+        pass
+    else:
+        raise AssertionError('expected AssertionError not raised')
+
+    try:
+        @bp.route('/bar/123', endpoint='bar.123')
+        def foo_bar_foo():
+            return flask.request.endpoint
+    except AssertionError:
+        pass
+    else:
+        raise AssertionError('expected AssertionError not raised')
+
+    def foo_foo_foo():
+        pass
+
+    pytest.raises(
+        AssertionError,
+        lambda: bp.add_url_rule(
+            '/bar/123', endpoint='bar.123', view_func=foo_foo_foo
+        )
+    )
+
+    pytest.raises(
+        AssertionError,
+        bp.route('/bar/123', endpoint='bar.123'),
+        lambda: None
+    )
+
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+
+    c = app.test_client()
+    assert c.get('/py/foo').data == b'bp.foo'
+    # The rule's didn't actually made it through
+    rv = c.get('/py/bar')
+    assert rv.status_code == 404
+    rv = c.get('/py/bar/123')
+    assert rv.status_code == 404
+
+
+def test_endpoint_decorator():
+    from werkzeug.routing import Rule
+    app = flask.Flask(__name__)
+    app.url_map.add(Rule('/foo', endpoint='bar'))
+
+    bp = flask.Blueprint('bp', __name__)
+
+    @bp.endpoint('bar')
+    def foobar():
+        return flask.request.endpoint
+
+    app.register_blueprint(bp, url_prefix='/bp_prefix')
+
+    c = app.test_client()
+    assert c.get('/foo').data == b'bar'
+    assert c.get('/bp_prefix/bar').status_code == 404
+
+
+def test_template_filter():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_filter()
+    def my_reverse(s):
+        return s[::-1]
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'my_reverse' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['my_reverse'] == my_reverse
+    assert app.jinja_env.filters['my_reverse']('abcd') == 'dcba'
+
+def test_add_template_filter():
+    bp = flask.Blueprint('bp', __name__)
+    def my_reverse(s):
+        return s[::-1]
+    bp.add_app_template_filter(my_reverse)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'my_reverse' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['my_reverse'] == my_reverse
+    assert app.jinja_env.filters['my_reverse']('abcd') == 'dcba'
+
+def test_template_filter_with_name():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_filter('strrev')
+    def my_reverse(s):
+        return s[::-1]
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'strrev' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['strrev'] == my_reverse
+    assert app.jinja_env.filters['strrev']('abcd') == 'dcba'
+
+def test_add_template_filter_with_name():
+    bp = flask.Blueprint('bp', __name__)
+    def my_reverse(s):
+        return s[::-1]
+    bp.add_app_template_filter(my_reverse, 'strrev')
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'strrev' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['strrev'] == my_reverse
+    assert app.jinja_env.filters['strrev']('abcd') == 'dcba'
+
+def test_template_filter_with_template():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_filter()
+    def super_reverse(s):
+        return s[::-1]
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_template_filter_after_route_with_template():
+    app = flask.Flask(__name__)
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_filter()
+    def super_reverse(s):
+        return s[::-1]
+    app.register_blueprint(bp, url_prefix='/py')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_add_template_filter_with_template():
+    bp = flask.Blueprint('bp', __name__)
+    def super_reverse(s):
+        return s[::-1]
+    bp.add_app_template_filter(super_reverse)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_template_filter_with_name_and_template():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_filter('super_reverse')
+    def my_reverse(s):
+        return s[::-1]
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_add_template_filter_with_name_and_template():
+    bp = flask.Blueprint('bp', __name__)
+    def my_reverse(s):
+        return s[::-1]
+    bp.add_app_template_filter(my_reverse, 'super_reverse')
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_template_test():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_test()
+    def is_boolean(value):
+        return isinstance(value, bool)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'is_boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['is_boolean'] == is_boolean
+    assert app.jinja_env.tests['is_boolean'](False)
+
+def test_add_template_test():
+    bp = flask.Blueprint('bp', __name__)
+    def is_boolean(value):
+        return isinstance(value, bool)
+    bp.add_app_template_test(is_boolean)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'is_boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['is_boolean'] == is_boolean
+    assert app.jinja_env.tests['is_boolean'](False)
+
+def test_template_test_with_name():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_test('boolean')
+    def is_boolean(value):
+        return isinstance(value, bool)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['boolean'] == is_boolean
+    assert app.jinja_env.tests['boolean'](False)
+
+def test_add_template_test_with_name():
+    bp = flask.Blueprint('bp', __name__)
+    def is_boolean(value):
+        return isinstance(value, bool)
+    bp.add_app_template_test(is_boolean, 'boolean')
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    assert 'boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['boolean'] == is_boolean
+    assert app.jinja_env.tests['boolean'](False)
+
+def test_template_test_with_template():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_test()
+    def boolean(value):
+        return isinstance(value, bool)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_template_test_after_route_with_template():
+    app = flask.Flask(__name__)
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_test()
+    def boolean(value):
+        return isinstance(value, bool)
+    app.register_blueprint(bp, url_prefix='/py')
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_add_template_test_with_template():
+    bp = flask.Blueprint('bp', __name__)
+    def boolean(value):
+        return isinstance(value, bool)
+    bp.add_app_template_test(boolean)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_template_test_with_name_and_template():
+    bp = flask.Blueprint('bp', __name__)
+    @bp.app_template_test('boolean')
+    def is_boolean(value):
+        return isinstance(value, bool)
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_add_template_test_with_name_and_template():
+    bp = flask.Blueprint('bp', __name__)
+    def is_boolean(value):
+        return isinstance(value, bool)
+    bp.add_app_template_test(is_boolean, 'boolean')
+    app = flask.Flask(__name__)
+    app.register_blueprint(bp, url_prefix='/py')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data

tests/test_cli.py

@@ -0,0 +1,270 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.test_cli
+    ~~~~~~~~~~~~~~
+
+    :copyright: (c) 2016 by the Flask Team, see AUTHORS for more details.
+    :license: BSD, see LICENSE for more details.
+"""
+#
+# This file was part of Flask-CLI and was modified under the terms its license,
+# the Revised BSD License.
+# Copyright (C) 2015 CERN.
+#
+from __future__ import absolute_import, print_function
+import os
+import sys
+from functools import partial
+
+import click
+import pytest
+from click.testing import CliRunner
+from flask import Flask, current_app
+
+from flask.cli import cli, AppGroup, FlaskGroup, NoAppException, ScriptInfo, \
+    find_best_app, locate_app, with_appcontext, prepare_exec_for_file, \
+    find_default_import_path, get_version
+
+
+@pytest.fixture
+def runner():
+    return CliRunner()
+
+
+def test_cli_name(test_apps):
+    """Make sure the CLI object's name is the app's name and not the app itself"""
+    from cliapp.app import testapp
+    assert testapp.cli.name == testapp.name
+
+
+def test_find_best_app(test_apps):
+    """Test if `find_best_app` behaves as expected with different combinations of input."""
+    class Module:
+        app = Flask('appname')
+    assert find_best_app(Module) == Module.app
+
+    class Module:
+        application = Flask('appname')
+    assert find_best_app(Module) == Module.application
+
+    class Module:
+        myapp = Flask('appname')
+    assert find_best_app(Module) == Module.myapp
+
+    class Module:
+        pass
+    pytest.raises(NoAppException, find_best_app, Module)
+
+    class Module:
+        myapp1 = Flask('appname1')
+        myapp2 = Flask('appname2')
+    pytest.raises(NoAppException, find_best_app, Module)
+
+
+def test_prepare_exec_for_file(test_apps):
+    """Expect the correct path to be set and the correct module name to be returned.
+
+    :func:`prepare_exec_for_file` has a side effect, where
+    the parent directory of given file is added to `sys.path`.
+    """
+    realpath = os.path.realpath('/tmp/share/test.py')
+    dirname = os.path.dirname(realpath)
+    assert prepare_exec_for_file('/tmp/share/test.py') == 'test'
+    assert dirname in sys.path
+
+    realpath = os.path.realpath('/tmp/share/__init__.py')
+    dirname = os.path.dirname(os.path.dirname(realpath))
+    assert prepare_exec_for_file('/tmp/share/__init__.py') == 'share'
+    assert dirname in sys.path
+
+    with pytest.raises(NoAppException):
+        prepare_exec_for_file('/tmp/share/test.txt')
+
+
+def test_locate_app(test_apps):
+    """Test of locate_app."""
+    assert locate_app("cliapp.app").name == "testapp"
+    assert locate_app("cliapp.app:testapp").name == "testapp"
+    assert locate_app("cliapp.multiapp:app1").name == "app1"
+    pytest.raises(NoAppException, locate_app, "notanpp.py")
+    pytest.raises(NoAppException, locate_app, "cliapp/app")
+    pytest.raises(RuntimeError, locate_app, "cliapp.app:notanapp")
+    pytest.raises(NoAppException, locate_app, "cliapp.importerrorapp")
+
+
+def test_find_default_import_path(test_apps, monkeypatch, tmpdir):
+    """Test of find_default_import_path."""
+    monkeypatch.delitem(os.environ, 'FLASK_APP', raising=False)
+    assert find_default_import_path() == None
+    monkeypatch.setitem(os.environ, 'FLASK_APP', 'notanapp')
+    assert find_default_import_path() == 'notanapp'
+    tmpfile = tmpdir.join('testapp.py')
+    tmpfile.write('')
+    monkeypatch.setitem(os.environ, 'FLASK_APP', str(tmpfile))
+    expect_rv = prepare_exec_for_file(str(tmpfile))
+    assert find_default_import_path() == expect_rv
+
+
+def test_get_version(test_apps, capsys):
+    """Test of get_version."""
+    from flask import __version__ as flask_ver
+    from sys import version as py_ver
+    class MockCtx(object):
+        resilient_parsing = False
+        color = None
+        def exit(self): return
+    ctx = MockCtx()
+    get_version(ctx, None, "test")
+    out, err = capsys.readouterr()
+    assert flask_ver in out
+    assert py_ver in out
+
+
+def test_scriptinfo(test_apps):
+    """Test of ScriptInfo."""
+    obj = ScriptInfo(app_import_path="cliapp.app:testapp")
+    assert obj.load_app().name == "testapp"
+    assert obj.load_app().name == "testapp"
+
+    def create_app(info):
+        return Flask("createapp")
+
+    obj = ScriptInfo(create_app=create_app)
+    app = obj.load_app()
+    assert app.name == "createapp"
+    assert obj.load_app() == app
+
+
+def test_with_appcontext(runner):
+    """Test of with_appcontext."""
+    @click.command()
+    @with_appcontext
+    def testcmd():
+        click.echo(current_app.name)
+
+    obj = ScriptInfo(create_app=lambda info: Flask("testapp"))
+
+    result = runner.invoke(testcmd, obj=obj)
+    assert result.exit_code == 0
+    assert result.output == 'testapp\n'
+
+
+def test_appgroup(runner):
+    """Test of with_appcontext."""
+    @click.group(cls=AppGroup)
+    def cli():
+        pass
+
+    @cli.command(with_appcontext=True)
+    def test():
+        click.echo(current_app.name)
+
+    @cli.group()
+    def subgroup():
+        pass
+
+    @subgroup.command(with_appcontext=True)
+    def test2():
+        click.echo(current_app.name)
+
+    obj = ScriptInfo(create_app=lambda info: Flask("testappgroup"))
+
+    result = runner.invoke(cli, ['test'], obj=obj)
+    assert result.exit_code == 0
+    assert result.output == 'testappgroup\n'
+
+    result = runner.invoke(cli, ['subgroup', 'test2'], obj=obj)
+    assert result.exit_code == 0
+    assert result.output == 'testappgroup\n'
+
+
+def test_flaskgroup(runner):
+    """Test FlaskGroup."""
+    def create_app(info):
+        return Flask("flaskgroup")
+
+    @click.group(cls=FlaskGroup, create_app=create_app)
+    def cli(**params):
+        pass
+
+    @cli.command()
+    def test():
+        click.echo(current_app.name)
+
+    result = runner.invoke(cli, ['test'])
+    assert result.exit_code == 0
+    assert result.output == 'flaskgroup\n'
+
+
+def test_print_exceptions(runner):
+    """Print the stacktrace if the CLI."""
+    def create_app(info):
+        raise Exception("oh no")
+        return Flask("flaskgroup")
+
+    @click.group(cls=FlaskGroup, create_app=create_app)
+    def cli(**params):
+        pass
+
+    result = runner.invoke(cli, ['--help'])
+    assert result.exit_code == 0
+    assert 'Exception: oh no' in result.output
+    assert 'Traceback' in result.output
+
+
+class TestRoutes:
+    @pytest.fixture
+    def invoke(self, runner):
+        def create_app(info):
+            app = Flask(__name__)
+            app.testing = True
+
+            @app.route('/get_post/<int:x>/<int:y>', methods=['GET', 'POST'])
+            def yyy_get_post(x, y):
+                pass
+
+            @app.route('/zzz_post', methods=['POST'])
+            def aaa_post():
+                pass
+
+            return app
+
+        cli = FlaskGroup(create_app=create_app)
+        return partial(runner.invoke, cli)
+
+    def expect_order(self, order, output):
+        # skip the header and match the start of each row
+        for expect, line in zip(order, output.splitlines()[2:]):
+            # do this instead of startswith for nicer pytest output
+            assert line[:len(expect)] == expect
+
+    def test_simple(self, invoke):
+        result = invoke(['routes'])
+        assert result.exit_code == 0
+        self.expect_order(
+            ['aaa_post', 'static', 'yyy_get_post'],
+            result.output
+        )
+
+    def test_sort(self, invoke):
+        default_output = invoke(['routes']).output
+        endpoint_output = invoke(['routes', '-s', 'endpoint']).output
+        assert default_output == endpoint_output
+        self.expect_order(
+            ['static', 'yyy_get_post', 'aaa_post'],
+            invoke(['routes', '-s', 'methods']).output
+        )
+        self.expect_order(
+            ['yyy_get_post', 'static', 'aaa_post'],
+            invoke(['routes', '-s', 'rule']).output
+        )
+        self.expect_order(
+            ['aaa_post', 'yyy_get_post', 'static'],
+            invoke(['routes', '-s', 'match']).output
+        )
+
+    def test_all_methods(self, invoke):
+        output = invoke(['routes']).output
+        assert 'GET, HEAD, OPTIONS, POST' not in output
+        output = invoke(['routes', '--all-methods']).output
+        assert 'GET, HEAD, OPTIONS, POST' in output

tests/test_config.py

@@ -0,0 +1,207 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.test_config
+    ~~~~~~~~~~~~~~~~~
+
+    :copyright: (c) 2015 by the Flask Team, see AUTHORS for more details.
+    :license: BSD, see LICENSE for more details.
+"""
+
+
+from datetime import timedelta
+import os
+import textwrap
+
+import flask
+from flask._compat import PY2
+import pytest
+
+
+# config keys used for the TestConfig
+TEST_KEY = 'foo'
+SECRET_KEY = 'devkey'
+
+
+def common_object_test(app):
+    assert app.secret_key == 'devkey'
+    assert app.config['TEST_KEY'] == 'foo'
+    assert 'TestConfig' not in app.config
+
+
+def test_config_from_file():
+    app = flask.Flask(__name__)
+    app.config.from_pyfile(__file__.rsplit('.', 1)[0] + '.py')
+    common_object_test(app)
+
+
+def test_config_from_object():
+    app = flask.Flask(__name__)
+    app.config.from_object(__name__)
+    common_object_test(app)
+
+
+def test_config_from_json():
+    app = flask.Flask(__name__)
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    app.config.from_json(os.path.join(current_dir, 'static', 'config.json'))
+    common_object_test(app)
+
+
+def test_config_from_mapping():
+    app = flask.Flask(__name__)
+    app.config.from_mapping({
+        'SECRET_KEY': 'devkey',
+        'TEST_KEY': 'foo'
+    })
+    common_object_test(app)
+
+    app = flask.Flask(__name__)
+    app.config.from_mapping([
+        ('SECRET_KEY', 'devkey'),
+        ('TEST_KEY', 'foo')
+    ])
+    common_object_test(app)
+
+    app = flask.Flask(__name__)
+    app.config.from_mapping(
+        SECRET_KEY='devkey',
+        TEST_KEY='foo'
+    )
+    common_object_test(app)
+
+    app = flask.Flask(__name__)
+    with pytest.raises(TypeError):
+        app.config.from_mapping(
+            {}, {}
+        )
+
+
+def test_config_from_class():
+    class Base(object):
+        TEST_KEY = 'foo'
+
+    class Test(Base):
+        SECRET_KEY = 'devkey'
+    app = flask.Flask(__name__)
+    app.config.from_object(Test)
+    common_object_test(app)
+
+
+def test_config_from_envvar():
+    env = os.environ
+    try:
+        os.environ = {}
+        app = flask.Flask(__name__)
+        with pytest.raises(RuntimeError) as e:
+            app.config.from_envvar('FOO_SETTINGS')
+        assert "'FOO_SETTINGS' is not set" in str(e.value)
+        assert not app.config.from_envvar('FOO_SETTINGS', silent=True)
+
+        os.environ = {'FOO_SETTINGS': __file__.rsplit('.', 1)[0] + '.py'}
+        assert app.config.from_envvar('FOO_SETTINGS')
+        common_object_test(app)
+    finally:
+        os.environ = env
+
+
+def test_config_from_envvar_missing():
+    env = os.environ
+    try:
+        os.environ = {'FOO_SETTINGS': 'missing.cfg'}
+        with pytest.raises(IOError) as e:
+            app = flask.Flask(__name__)
+            app.config.from_envvar('FOO_SETTINGS')
+        msg = str(e.value)
+        assert msg.startswith('[Errno 2] Unable to load configuration '
+                              'file (No such file or directory):')
+        assert msg.endswith("missing.cfg'")
+        assert not app.config.from_envvar('FOO_SETTINGS', silent=True)
+    finally:
+        os.environ = env
+
+
+def test_config_missing():
+    app = flask.Flask(__name__)
+    with pytest.raises(IOError) as e:
+        app.config.from_pyfile('missing.cfg')
+    msg = str(e.value)
+    assert msg.startswith('[Errno 2] Unable to load configuration '
+                          'file (No such file or directory):')
+    assert msg.endswith("missing.cfg'")
+    assert not app.config.from_pyfile('missing.cfg', silent=True)
+
+
+def test_config_missing_json():
+    app = flask.Flask(__name__)
+    with pytest.raises(IOError) as e:
+        app.config.from_json('missing.json')
+    msg = str(e.value)
+    assert msg.startswith('[Errno 2] Unable to load configuration '
+                          'file (No such file or directory):')
+    assert msg.endswith("missing.json'")
+    assert not app.config.from_json('missing.json', silent=True)
+
+
+def test_custom_config_class():
+    class Config(flask.Config):
+        pass
+
+    class Flask(flask.Flask):
+        config_class = Config
+    app = Flask(__name__)
+    assert isinstance(app.config, Config)
+    app.config.from_object(__name__)
+    common_object_test(app)
+
+
+def test_session_lifetime():
+    app = flask.Flask(__name__)
+    app.config['PERMANENT_SESSION_LIFETIME'] = 42
+    assert app.permanent_session_lifetime.seconds == 42
+
+
+def test_send_file_max_age():
+    app = flask.Flask(__name__)
+    app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 3600
+    assert app.send_file_max_age_default.seconds == 3600
+    app.config['SEND_FILE_MAX_AGE_DEFAULT'] = timedelta(hours=2)
+    assert app.send_file_max_age_default.seconds == 7200
+
+
+def test_get_namespace():
+    app = flask.Flask(__name__)
+    app.config['FOO_OPTION_1'] = 'foo option 1'
+    app.config['FOO_OPTION_2'] = 'foo option 2'
+    app.config['BAR_STUFF_1'] = 'bar stuff 1'
+    app.config['BAR_STUFF_2'] = 'bar stuff 2'
+    foo_options = app.config.get_namespace('FOO_')
+    assert 2 == len(foo_options)
+    assert 'foo option 1' == foo_options['option_1']
+    assert 'foo option 2' == foo_options['option_2']
+    bar_options = app.config.get_namespace('BAR_', lowercase=False)
+    assert 2 == len(bar_options)
+    assert 'bar stuff 1' == bar_options['STUFF_1']
+    assert 'bar stuff 2' == bar_options['STUFF_2']
+    foo_options = app.config.get_namespace('FOO_', trim_namespace=False)
+    assert 2 == len(foo_options)
+    assert 'foo option 1' == foo_options['foo_option_1']
+    assert 'foo option 2' == foo_options['foo_option_2']
+    bar_options = app.config.get_namespace('BAR_', lowercase=False, trim_namespace=False)
+    assert 2 == len(bar_options)
+    assert 'bar stuff 1' == bar_options['BAR_STUFF_1']
+    assert 'bar stuff 2' == bar_options['BAR_STUFF_2']
+
+
+@pytest.mark.parametrize('encoding', ['utf-8', 'iso-8859-15', 'latin-1'])
+def test_from_pyfile_weird_encoding(tmpdir, encoding):
+    f = tmpdir.join('my_config.py')
+    f.write_binary(textwrap.dedent(u'''
+    # -*- coding: {0} -*-
+    TEST_VALUE = "föö"
+    '''.format(encoding)).encode(encoding))
+    app = flask.Flask(__name__)
+    app.config.from_pyfile(str(f))
+    value = app.config['TEST_VALUE']
+    if PY2:
+        value = value.decode(encoding)
+    assert value == u'föö'

tests/test_deprecations.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.deprecations
+    ~~~~~~~~~~~~~~~~~~
+
+    Tests deprecation support. Not used currently.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import flask
+
+
+class TestRequestDeprecation(object):
+
+    def test_request_json(self, recwarn):
+        """Request.json is deprecated"""
+        app = flask.Flask(__name__)
+        app.testing = True
+
+        @app.route('/', methods=['POST'])
+        def index():
+            assert flask.request.json == {'spam': 42}
+            print(flask.request.json)
+            return 'OK'
+
+        c = app.test_client()
+        c.post('/', data='{"spam": 42}', content_type='application/json')
+        recwarn.pop(DeprecationWarning)
+
+    def test_request_module(self, recwarn):
+        """Request.module is deprecated"""
+        app = flask.Flask(__name__)
+        app.testing = True
+
+        @app.route('/')
+        def index():
+            assert flask.request.module is None
+            return 'OK'
+
+        c = app.test_client()
+        c.get('/')
+        recwarn.pop(DeprecationWarning)

tests/test_ext.py

@@ -0,0 +1,199 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.ext
+    ~~~~~~~~~~~~~~~~~~~
+
+    Tests the extension import thing.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import sys
+import pytest
+
+try:
+    from imp import reload as reload_module
+except ImportError:
+    reload_module = reload
+
+from flask._compat import PY2
+
+
+@pytest.fixture(autouse=True)
+def disable_extwarnings(request, recwarn):
+    from flask.exthook import ExtDeprecationWarning
+
+    def inner():
+        assert set(w.category for w in recwarn.list) \
+            <= set([ExtDeprecationWarning])
+        recwarn.clear()
+
+    request.addfinalizer(inner)
+
+
+@pytest.fixture(autouse=True)
+def importhook_setup(monkeypatch, request):
+    # we clear this out for various reasons.  The most important one is
+    # that a real flaskext could be in there which would disable our
+    # fake package.  Secondly we want to make sure that the flaskext
+    # import hook does not break on reloading.
+    for entry, value in list(sys.modules.items()):
+        if (
+            entry.startswith('flask.ext.') or
+            entry.startswith('flask_') or
+            entry.startswith('flaskext.') or
+            entry == 'flaskext'
+        ) and value is not None:
+            monkeypatch.delitem(sys.modules, entry)
+    from flask import ext
+    reload_module(ext)
+
+    # reloading must not add more hooks
+    import_hooks = 0
+    for item in sys.meta_path:
+        cls = type(item)
+        if cls.__module__ == 'flask.exthook' and \
+           cls.__name__ == 'ExtensionImporter':
+            import_hooks += 1
+    assert import_hooks == 1
+
+    def teardown():
+        from flask import ext
+        for key in ext.__dict__:
+            assert '.' not in key
+
+    request.addfinalizer(teardown)
+
+
+@pytest.fixture
+def newext_simple(modules_tmpdir):
+    x = modules_tmpdir.join('flask_newext_simple.py')
+    x.write('ext_id = "newext_simple"')
+
+
+@pytest.fixture
+def oldext_simple(modules_tmpdir):
+    flaskext = modules_tmpdir.mkdir('flaskext')
+    flaskext.join('__init__.py').write('\n')
+    flaskext.join('oldext_simple.py').write('ext_id = "oldext_simple"')
+
+
+@pytest.fixture
+def newext_package(modules_tmpdir):
+    pkg = modules_tmpdir.mkdir('flask_newext_package')
+    pkg.join('__init__.py').write('ext_id = "newext_package"')
+    pkg.join('submodule.py').write('def test_function():\n    return 42\n')
+
+
+@pytest.fixture
+def oldext_package(modules_tmpdir):
+    flaskext = modules_tmpdir.mkdir('flaskext')
+    flaskext.join('__init__.py').write('\n')
+    oldext = flaskext.mkdir('oldext_package')
+    oldext.join('__init__.py').write('ext_id = "oldext_package"')
+    oldext.join('submodule.py').write('def test_function():\n'
+                                      '    return 42')
+
+
+@pytest.fixture
+def flaskext_broken(modules_tmpdir):
+    ext = modules_tmpdir.mkdir('flask_broken')
+    ext.join('b.py').write('\n')
+    ext.join('__init__.py').write('import flask.ext.broken.b\n'
+                                  'import missing_module')
+
+
+def test_flaskext_new_simple_import_normal(newext_simple):
+    from flask.ext.newext_simple import ext_id
+    assert ext_id == 'newext_simple'
+
+
+def test_flaskext_new_simple_import_module(newext_simple):
+    from flask.ext import newext_simple
+    assert newext_simple.ext_id == 'newext_simple'
+    assert newext_simple.__name__ == 'flask_newext_simple'
+
+
+def test_flaskext_new_package_import_normal(newext_package):
+    from flask.ext.newext_package import ext_id
+    assert ext_id == 'newext_package'
+
+
+def test_flaskext_new_package_import_module(newext_package):
+    from flask.ext import newext_package
+    assert newext_package.ext_id == 'newext_package'
+    assert newext_package.__name__ == 'flask_newext_package'
+
+
+def test_flaskext_new_package_import_submodule_function(newext_package):
+    from flask.ext.newext_package.submodule import test_function
+    assert test_function() == 42
+
+
+def test_flaskext_new_package_import_submodule(newext_package):
+    from flask.ext.newext_package import submodule
+    assert submodule.__name__ == 'flask_newext_package.submodule'
+    assert submodule.test_function() == 42
+
+
+def test_flaskext_old_simple_import_normal(oldext_simple):
+    from flask.ext.oldext_simple import ext_id
+    assert ext_id == 'oldext_simple'
+
+
+def test_flaskext_old_simple_import_module(oldext_simple):
+    from flask.ext import oldext_simple
+    assert oldext_simple.ext_id == 'oldext_simple'
+    assert oldext_simple.__name__ == 'flaskext.oldext_simple'
+
+
+def test_flaskext_old_package_import_normal(oldext_package):
+    from flask.ext.oldext_package import ext_id
+    assert ext_id == 'oldext_package'
+
+
+def test_flaskext_old_package_import_module(oldext_package):
+    from flask.ext import oldext_package
+    assert oldext_package.ext_id == 'oldext_package'
+    assert oldext_package.__name__ == 'flaskext.oldext_package'
+
+
+def test_flaskext_old_package_import_submodule(oldext_package):
+    from flask.ext.oldext_package import submodule
+    assert submodule.__name__ == 'flaskext.oldext_package.submodule'
+    assert submodule.test_function() == 42
+
+
+def test_flaskext_old_package_import_submodule_function(oldext_package):
+    from flask.ext.oldext_package.submodule import test_function
+    assert test_function() == 42
+
+
+def test_flaskext_broken_package_no_module_caching(flaskext_broken):
+    for x in range(2):
+        with pytest.raises(ImportError):
+            import flask.ext.broken
+
+
+def test_no_error_swallowing(flaskext_broken):
+    with pytest.raises(ImportError) as excinfo:
+        import flask.ext.broken
+    # python3.6 raises a subclass of ImportError: 'ModuleNotFoundError'
+    assert issubclass(excinfo.type, ImportError)
+    if PY2:
+        message = 'No module named missing_module'
+    else:
+        message = 'No module named \'missing_module\''
+    assert str(excinfo.value) == message
+    assert excinfo.tb.tb_frame.f_globals is globals()
+
+    # reraise() adds a second frame so we need to skip that one too.
+    # On PY3 we even have another one :(
+    next = excinfo.tb.tb_next.tb_next
+    if not PY2:
+        next = next.tb_next
+
+    import os.path
+    assert os.path.join('flask_broken', '__init__.py') in \
+        next.tb_frame.f_code.co_filename

tests/test_helpers.py

@@ -0,0 +1,943 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.helpers
+    ~~~~~~~~~~~~~~~~~~~~~~~
+
+    Various helpers.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import os
+import uuid
+import datetime
+
+import flask
+from logging import StreamHandler
+from werkzeug.datastructures import Range
+from werkzeug.exceptions import BadRequest, NotFound
+from werkzeug.http import parse_cache_control_header, parse_options_header
+from werkzeug.http import http_date
+from flask._compat import StringIO, text_type
+
+
+def has_encoding(name):
+    try:
+        import codecs
+        codecs.lookup(name)
+        return True
+    except LookupError:
+        return False
+
+
+class TestJSON(object):
+
+    def test_ignore_cached_json(self):
+        app = flask.Flask(__name__)
+        with app.test_request_context('/', method='POST', data='malformed',
+                                      content_type='application/json'):
+            assert flask.request.get_json(silent=True, cache=True) is None
+            with pytest.raises(BadRequest):
+                flask.request.get_json(silent=False, cache=False)
+
+    def test_post_empty_json_adds_exception_to_response_content_in_debug(self):
+        app = flask.Flask(__name__)
+        app.config['DEBUG'] = True
+        @app.route('/json', methods=['POST'])
+        def post_json():
+            flask.request.get_json()
+            return None
+        c = app.test_client()
+        rv = c.post('/json', data=None, content_type='application/json')
+        assert rv.status_code == 400
+        assert b'Failed to decode JSON object' in rv.data
+
+    def test_post_empty_json_wont_add_exception_to_response_if_no_debug(self):
+        app = flask.Flask(__name__)
+        app.config['DEBUG'] = False
+        @app.route('/json', methods=['POST'])
+        def post_json():
+            flask.request.get_json()
+            return None
+        c = app.test_client()
+        rv = c.post('/json', data=None, content_type='application/json')
+        assert rv.status_code == 400
+        assert b'Failed to decode JSON object' not in rv.data
+
+    def test_json_bad_requests(self):
+        app = flask.Flask(__name__)
+        @app.route('/json', methods=['POST'])
+        def return_json():
+            return flask.jsonify(foo=text_type(flask.request.get_json()))
+        c = app.test_client()
+        rv = c.post('/json', data='malformed', content_type='application/json')
+        assert rv.status_code == 400
+
+    def test_json_custom_mimetypes(self):
+        app = flask.Flask(__name__)
+        @app.route('/json', methods=['POST'])
+        def return_json():
+            return flask.request.get_json()
+        c = app.test_client()
+        rv = c.post('/json', data='"foo"', content_type='application/x+json')
+        assert rv.data == b'foo'
+
+    def test_json_body_encoding(self):
+        app = flask.Flask(__name__)
+        app.testing = True
+        @app.route('/')
+        def index():
+            return flask.request.get_json()
+
+        c = app.test_client()
+        resp = c.get('/', data=u'"Hällo Wörld"'.encode('iso-8859-15'),
+                     content_type='application/json; charset=iso-8859-15')
+        assert resp.data == u'Hällo Wörld'.encode('utf-8')
+
+    def test_json_as_unicode(self):
+        app = flask.Flask(__name__)
+
+        app.config['JSON_AS_ASCII'] = True
+        with app.app_context():
+            rv = flask.json.dumps(u'\N{SNOWMAN}')
+            assert rv == '"\\u2603"'
+
+        app.config['JSON_AS_ASCII'] = False
+        with app.app_context():
+            rv = flask.json.dumps(u'\N{SNOWMAN}')
+            assert rv == u'"\u2603"'
+
+    def test_json_dump_to_file(self):
+        app = flask.Flask(__name__)
+        test_data = {'name': 'Flask'}
+        out = StringIO()
+
+        with app.app_context():
+            flask.json.dump(test_data, out)
+            out.seek(0)
+            rv = flask.json.load(out)
+            assert rv == test_data
+
+    @pytest.mark.parametrize('test_value', [0, -1, 1, 23, 3.14, 's', "longer string", True, False, None])
+    def test_jsonify_basic_types(self, test_value):
+        """Test jsonify with basic types."""
+        app = flask.Flask(__name__)
+        c = app.test_client()
+
+        url = '/jsonify_basic_types'
+        app.add_url_rule(url, url, lambda x=test_value: flask.jsonify(x))
+        rv = c.get(url)
+        assert rv.mimetype == 'application/json'
+        assert flask.json.loads(rv.data) == test_value
+
+    def test_jsonify_dicts(self):
+        """Test jsonify with dicts and kwargs unpacking."""
+        d = dict(
+            a=0, b=23, c=3.14, d='t', e='Hi', f=True, g=False,
+            h=['test list', 10, False],
+            i={'test':'dict'}
+        )
+        app = flask.Flask(__name__)
+        @app.route('/kw')
+        def return_kwargs():
+            return flask.jsonify(**d)
+        @app.route('/dict')
+        def return_dict():
+            return flask.jsonify(d)
+        c = app.test_client()
+        for url in '/kw', '/dict':
+            rv = c.get(url)
+            assert rv.mimetype == 'application/json'
+            assert flask.json.loads(rv.data) == d
+
+    def test_jsonify_arrays(self):
+        """Test jsonify of lists and args unpacking."""
+        l = [
+            0, 42, 3.14, 't', 'hello', True, False,
+            ['test list', 2, False],
+            {'test':'dict'}
+        ]
+        app = flask.Flask(__name__)
+        @app.route('/args_unpack')
+        def return_args_unpack():
+            return flask.jsonify(*l)
+        @app.route('/array')
+        def return_array():
+            return flask.jsonify(l)
+        c = app.test_client()
+        for url in '/args_unpack', '/array':
+            rv = c.get(url)
+            assert rv.mimetype == 'application/json'
+            assert flask.json.loads(rv.data) == l
+
+    def test_jsonify_date_types(self):
+        """Test jsonify with datetime.date and datetime.datetime types."""
+        test_dates = (
+            datetime.datetime(1973, 3, 11, 6, 30, 45),
+            datetime.date(1975, 1, 5)
+        )
+        app = flask.Flask(__name__)
+        c = app.test_client()
+
+        for i, d in enumerate(test_dates):
+            url = '/datetest{0}'.format(i)
+            app.add_url_rule(url, str(i), lambda val=d: flask.jsonify(x=val))
+            rv = c.get(url)
+            assert rv.mimetype == 'application/json'
+            assert flask.json.loads(rv.data)['x'] == http_date(d.timetuple())
+
+    def test_jsonify_uuid_types(self):
+        """Test jsonify with uuid.UUID types"""
+
+        test_uuid = uuid.UUID(bytes=b'\xDE\xAD\xBE\xEF' * 4)
+        app = flask.Flask(__name__)
+        url = '/uuid_test'
+        app.add_url_rule(url, url, lambda: flask.jsonify(x=test_uuid))
+
+        c = app.test_client()
+        rv = c.get(url)
+
+        rv_x = flask.json.loads(rv.data)['x']
+        assert rv_x == str(test_uuid)
+        rv_uuid = uuid.UUID(rv_x)
+        assert rv_uuid == test_uuid
+
+    def test_json_attr(self):
+        app = flask.Flask(__name__)
+        @app.route('/add', methods=['POST'])
+        def add():
+            json = flask.request.get_json()
+            return text_type(json['a'] + json['b'])
+        c = app.test_client()
+        rv = c.post('/add', data=flask.json.dumps({'a': 1, 'b': 2}),
+                            content_type='application/json')
+        assert rv.data == b'3'
+
+    def test_template_escaping(self):
+        app = flask.Flask(__name__)
+        render = flask.render_template_string
+        with app.test_request_context():
+            rv = flask.json.htmlsafe_dumps('</script>')
+            assert rv == u'"\\u003c/script\\u003e"'
+            assert type(rv) == text_type
+            rv = render('{{ "</script>"|tojson }}')
+            assert rv == '"\\u003c/script\\u003e"'
+            rv = render('{{ "<\0/script>"|tojson }}')
+            assert rv == '"\\u003c\\u0000/script\\u003e"'
+            rv = render('{{ "<!--<script>"|tojson }}')
+            assert rv == '"\\u003c!--\\u003cscript\\u003e"'
+            rv = render('{{ "&"|tojson }}')
+            assert rv == '"\\u0026"'
+            rv = render('{{ "\'"|tojson }}')
+            assert rv == '"\\u0027"'
+            rv = render("<a ng-data='{{ data|tojson }}'></a>",
+                data={'x': ["foo", "bar", "baz'"]})
+            assert rv == '<a ng-data=\'{"x": ["foo", "bar", "baz\\u0027"]}\'></a>'
+
+    def test_json_customization(self):
+        class X(object):
+            def __init__(self, val):
+                self.val = val
+        class MyEncoder(flask.json.JSONEncoder):
+            def default(self, o):
+                if isinstance(o, X):
+                    return '<%d>' % o.val
+                return flask.json.JSONEncoder.default(self, o)
+        class MyDecoder(flask.json.JSONDecoder):
+            def __init__(self, *args, **kwargs):
+                kwargs.setdefault('object_hook', self.object_hook)
+                flask.json.JSONDecoder.__init__(self, *args, **kwargs)
+            def object_hook(self, obj):
+                if len(obj) == 1 and '_foo' in obj:
+                    return X(obj['_foo'])
+                return obj
+        app = flask.Flask(__name__)
+        app.testing = True
+        app.json_encoder = MyEncoder
+        app.json_decoder = MyDecoder
+        @app.route('/', methods=['POST'])
+        def index():
+            return flask.json.dumps(flask.request.get_json()['x'])
+        c = app.test_client()
+        rv = c.post('/', data=flask.json.dumps({
+            'x': {'_foo': 42}
+        }), content_type='application/json')
+        assert rv.data == b'"<42>"'
+
+    def test_blueprint_json_customization(self):
+        class X(object):
+            def __init__(self, val):
+                self.val = val
+
+        class MyEncoder(flask.json.JSONEncoder):
+            def default(self, o):
+                if isinstance(o, X):
+                    return '<%d>' % o.val
+
+                return flask.json.JSONEncoder.default(self, o)
+
+        class MyDecoder(flask.json.JSONDecoder):
+            def __init__(self, *args, **kwargs):
+                kwargs.setdefault('object_hook', self.object_hook)
+                flask.json.JSONDecoder.__init__(self, *args, **kwargs)
+
+            def object_hook(self, obj):
+                if len(obj) == 1 and '_foo' in obj:
+                    return X(obj['_foo'])
+
+                return obj
+
+        bp = flask.Blueprint('bp', __name__)
+        bp.json_encoder = MyEncoder
+        bp.json_decoder = MyDecoder
+
+        @bp.route('/bp', methods=['POST'])
+        def index():
+            return flask.json.dumps(flask.request.get_json()['x'])
+
+        app = flask.Flask(__name__)
+        app.testing = True
+        app.register_blueprint(bp)
+
+        c = app.test_client()
+        rv = c.post('/bp', data=flask.json.dumps({
+            'x': {'_foo': 42}
+        }), content_type='application/json')
+        assert rv.data == b'"<42>"'
+
+    def test_modified_url_encoding(self):
+        class ModifiedRequest(flask.Request):
+            url_charset = 'euc-kr'
+        app = flask.Flask(__name__)
+        app.testing = True
+        app.request_class = ModifiedRequest
+        app.url_map.charset = 'euc-kr'
+
+        @app.route('/')
+        def index():
+            return flask.request.args['foo']
+
+        rv = app.test_client().get(u'/?foo=정상처리'.encode('euc-kr'))
+        assert rv.status_code == 200
+        assert rv.data == u'정상처리'.encode('utf-8')
+
+    if not has_encoding('euc-kr'):
+        test_modified_url_encoding = None
+
+    def test_json_key_sorting(self):
+        app = flask.Flask(__name__)
+        app.testing = True
+        app.debug = True
+
+        assert app.config['JSON_SORT_KEYS'] == True
+        d = dict.fromkeys(range(20), 'foo')
+
+        @app.route('/')
+        def index():
+            return flask.jsonify(values=d)
+
+        c = app.test_client()
+        rv = c.get('/')
+        lines = [x.strip() for x in rv.data.strip().decode('utf-8').splitlines()]
+        sorted_by_str = [
+            '{',
+            '"values": {',
+            '"0": "foo",',
+            '"1": "foo",',
+            '"10": "foo",',
+            '"11": "foo",',
+            '"12": "foo",',
+            '"13": "foo",',
+            '"14": "foo",',
+            '"15": "foo",',
+            '"16": "foo",',
+            '"17": "foo",',
+            '"18": "foo",',
+            '"19": "foo",',
+            '"2": "foo",',
+            '"3": "foo",',
+            '"4": "foo",',
+            '"5": "foo",',
+            '"6": "foo",',
+            '"7": "foo",',
+            '"8": "foo",',
+            '"9": "foo"',
+            '}',
+            '}'
+        ]
+        sorted_by_int = [
+            '{',
+            '"values": {',
+            '"0": "foo",',
+            '"1": "foo",',
+            '"2": "foo",',
+            '"3": "foo",',
+            '"4": "foo",',
+            '"5": "foo",',
+            '"6": "foo",',
+            '"7": "foo",',
+            '"8": "foo",',
+            '"9": "foo",',
+            '"10": "foo",',
+            '"11": "foo",',
+            '"12": "foo",',
+            '"13": "foo",',
+            '"14": "foo",',
+            '"15": "foo",',
+            '"16": "foo",',
+            '"17": "foo",',
+            '"18": "foo",',
+            '"19": "foo"',
+            '}',
+            '}'
+        ]
+
+        try:
+            assert lines == sorted_by_int
+        except AssertionError:
+            assert lines == sorted_by_str
+
+class TestSendfile(object):
+
+    def test_send_file_regular(self):
+        app = flask.Flask(__name__)
+        with app.test_request_context():
+            rv = flask.send_file('static/index.html')
+            assert rv.direct_passthrough
+            assert rv.mimetype == 'text/html'
+            with app.open_resource('static/index.html') as f:
+                rv.direct_passthrough = False
+                assert rv.data == f.read()
+            rv.close()
+
+    def test_send_file_xsendfile(self, catch_deprecation_warnings):
+        app = flask.Flask(__name__)
+        app.use_x_sendfile = True
+        with app.test_request_context():
+            rv = flask.send_file('static/index.html')
+            assert rv.direct_passthrough
+            assert 'x-sendfile' in rv.headers
+            assert rv.headers['x-sendfile'] == \
+                os.path.join(app.root_path, 'static/index.html')
+            assert rv.mimetype == 'text/html'
+            rv.close()
+
+    def test_send_file_last_modified(self):
+        app = flask.Flask(__name__)
+        last_modified = datetime.datetime(1999, 1, 1)
+
+        @app.route('/')
+        def index():
+            return flask.send_file(StringIO("party like it's"),
+                                   last_modified=last_modified,
+                                   mimetype='text/plain')
+
+        c = app.test_client()
+        rv = c.get('/')
+        assert rv.last_modified == last_modified
+
+    def test_send_file_object_without_mimetype(self):
+        app = flask.Flask(__name__)
+
+        with app.test_request_context():
+            with pytest.raises(ValueError) as excinfo:
+                flask.send_file(StringIO("LOL"))
+            assert 'Unable to infer MIME-type' in str(excinfo)
+            assert 'no filename is available' in str(excinfo)
+
+        with app.test_request_context():
+            flask.send_file(StringIO("LOL"), attachment_filename='filename')
+
+    def test_send_file_object(self):
+        app = flask.Flask(__name__)
+
+        with app.test_request_context():
+            with open(os.path.join(app.root_path, 'static/index.html'), mode='rb') as f:
+                rv = flask.send_file(f, mimetype='text/html')
+                rv.direct_passthrough = False
+                with app.open_resource('static/index.html') as f:
+                    assert rv.data == f.read()
+                assert rv.mimetype == 'text/html'
+                rv.close()
+
+        app.use_x_sendfile = True
+
+        with app.test_request_context():
+            with open(os.path.join(app.root_path, 'static/index.html')) as f:
+                rv = flask.send_file(f, mimetype='text/html')
+                assert rv.mimetype == 'text/html'
+                assert 'x-sendfile' not in rv.headers
+                rv.close()
+
+        app.use_x_sendfile = False
+        with app.test_request_context():
+            f = StringIO('Test')
+            rv = flask.send_file(f, mimetype='application/octet-stream')
+            rv.direct_passthrough = False
+            assert rv.data == b'Test'
+            assert rv.mimetype == 'application/octet-stream'
+            rv.close()
+
+            class PyStringIO(object):
+                def __init__(self, *args, **kwargs):
+                    self._io = StringIO(*args, **kwargs)
+                def __getattr__(self, name):
+                    return getattr(self._io, name)
+            f = PyStringIO('Test')
+            f.name = 'test.txt'
+            rv = flask.send_file(f, attachment_filename=f.name)
+            rv.direct_passthrough = False
+            assert rv.data == b'Test'
+            assert rv.mimetype == 'text/plain'
+            rv.close()
+
+            f = StringIO('Test')
+            rv = flask.send_file(f, mimetype='text/plain')
+            rv.direct_passthrough = False
+            assert rv.data == b'Test'
+            assert rv.mimetype == 'text/plain'
+            rv.close()
+
+        app.use_x_sendfile = True
+
+        with app.test_request_context():
+            f = StringIO('Test')
+            rv = flask.send_file(f, mimetype='text/html')
+            assert 'x-sendfile' not in rv.headers
+            rv.close()
+
+    @pytest.mark.skipif(
+        not callable(getattr(Range, 'to_content_range_header', None)),
+        reason="not implement within werkzeug"
+    )
+    def test_send_file_range_request(self):
+        app = flask.Flask(__name__)
+
+        @app.route('/')
+        def index():
+            return flask.send_file('static/index.html', conditional=True)
+
+        c = app.test_client()
+
+        rv = c.get('/', headers={'Range': 'bytes=4-15'})
+        assert rv.status_code == 206
+        with app.open_resource('static/index.html') as f:
+            assert rv.data == f.read()[4:16]
+        rv.close()
+
+        rv = c.get('/', headers={'Range': 'bytes=4-'})
+        assert rv.status_code == 206
+        with app.open_resource('static/index.html') as f:
+            assert rv.data == f.read()[4:]
+        rv.close()
+
+        rv = c.get('/', headers={'Range': 'bytes=4-1000'})
+        assert rv.status_code == 206
+        with app.open_resource('static/index.html') as f:
+            assert rv.data == f.read()[4:]
+        rv.close()
+
+        rv = c.get('/', headers={'Range': 'bytes=-10'})
+        assert rv.status_code == 206
+        with app.open_resource('static/index.html') as f:
+            assert rv.data == f.read()[-10:]
+        rv.close()
+
+        rv = c.get('/', headers={'Range': 'bytes=1000-'})
+        assert rv.status_code == 416
+        rv.close()
+
+        rv = c.get('/', headers={'Range': 'bytes=-'})
+        assert rv.status_code == 416
+        rv.close()
+
+        rv = c.get('/', headers={'Range': 'somethingsomething'})
+        assert rv.status_code == 416
+        rv.close()
+
+        last_modified = datetime.datetime.utcfromtimestamp(os.path.getmtime(
+            os.path.join(app.root_path, 'static/index.html'))).replace(
+            microsecond=0)
+
+        rv = c.get('/', headers={'Range': 'bytes=4-15',
+                                 'If-Range': http_date(last_modified)})
+        assert rv.status_code == 206
+        rv.close()
+
+        rv = c.get('/', headers={'Range': 'bytes=4-15', 'If-Range': http_date(
+            datetime.datetime(1999, 1, 1))})
+        assert rv.status_code == 200
+        rv.close()
+
+    def test_attachment(self):
+        app = flask.Flask(__name__)
+        with app.test_request_context():
+            with open(os.path.join(app.root_path, 'static/index.html')) as f:
+                rv = flask.send_file(f, as_attachment=True,
+                                     attachment_filename='index.html')
+                value, options = \
+                    parse_options_header(rv.headers['Content-Disposition'])
+                assert value == 'attachment'
+                assert options['filename'] == 'index.html'
+                assert 'filename*' not in rv.headers['Content-Disposition']
+                rv.close()
+
+        with app.test_request_context():
+            rv = flask.send_file('static/index.html', as_attachment=True)
+            value, options = parse_options_header(rv.headers['Content-Disposition'])
+            assert value == 'attachment'
+            assert options['filename'] == 'index.html'
+            rv.close()
+
+        with app.test_request_context():
+            rv = flask.send_file(StringIO('Test'), as_attachment=True,
+                                 attachment_filename='index.txt',
+                                 add_etags=False)
+            assert rv.mimetype == 'text/plain'
+            value, options = parse_options_header(rv.headers['Content-Disposition'])
+            assert value == 'attachment'
+            assert options['filename'] == 'index.txt'
+            rv.close()
+
+    def test_attachment_with_utf8_filename(self):
+        app = flask.Flask(__name__)
+
+        with app.test_request_context():
+            rv = flask.send_file('static/index.html', as_attachment=True, attachment_filename=u'Ñandú／pingüino.txt')
+            content_disposition = set(rv.headers['Content-Disposition'].split('; '))
+            assert content_disposition == set((
+                'attachment',
+                'filename="Nandu/pinguino.txt"',
+                "filename*=UTF-8''%C3%91and%C3%BA%EF%BC%8Fping%C3%BCino.txt"
+            ))
+            rv.close()
+
+    def test_static_file(self):
+        app = flask.Flask(__name__)
+        # default cache timeout is 12 hours
+        with app.test_request_context():
+            # Test with static file handler.
+            rv = app.send_static_file('index.html')
+            cc = parse_cache_control_header(rv.headers['Cache-Control'])
+            assert cc.max_age == 12 * 60 * 60
+            rv.close()
+            # Test again with direct use of send_file utility.
+            rv = flask.send_file('static/index.html')
+            cc = parse_cache_control_header(rv.headers['Cache-Control'])
+            assert cc.max_age == 12 * 60 * 60
+            rv.close()
+        app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 3600
+        with app.test_request_context():
+            # Test with static file handler.
+            rv = app.send_static_file('index.html')
+            cc = parse_cache_control_header(rv.headers['Cache-Control'])
+            assert cc.max_age == 3600
+            rv.close()
+            # Test again with direct use of send_file utility.
+            rv = flask.send_file('static/index.html')
+            cc = parse_cache_control_header(rv.headers['Cache-Control'])
+            assert cc.max_age == 3600
+            rv.close()
+        class StaticFileApp(flask.Flask):
+            def get_send_file_max_age(self, filename):
+                return 10
+        app = StaticFileApp(__name__)
+        with app.test_request_context():
+            # Test with static file handler.
+            rv = app.send_static_file('index.html')
+            cc = parse_cache_control_header(rv.headers['Cache-Control'])
+            assert cc.max_age == 10
+            rv.close()
+            # Test again with direct use of send_file utility.
+            rv = flask.send_file('static/index.html')
+            cc = parse_cache_control_header(rv.headers['Cache-Control'])
+            assert cc.max_age == 10
+            rv.close()
+
+    def test_send_from_directory(self):
+        app = flask.Flask(__name__)
+        app.testing = True
+        app.root_path = os.path.join(os.path.dirname(__file__),
+                                     'test_apps', 'subdomaintestmodule')
+        with app.test_request_context():
+            rv = flask.send_from_directory('static', 'hello.txt')
+            rv.direct_passthrough = False
+            assert rv.data.strip() == b'Hello Subdomain'
+            rv.close()
+
+    def test_send_from_directory_bad_request(self):
+        app = flask.Flask(__name__)
+        app.testing = True
+        app.root_path = os.path.join(os.path.dirname(__file__),
+                                     'test_apps', 'subdomaintestmodule')
+        with app.test_request_context():
+            with pytest.raises(BadRequest):
+                flask.send_from_directory('static', 'bad\x00')
+
+class TestLogging(object):
+
+    def test_logger_cache(self):
+        app = flask.Flask(__name__)
+        logger1 = app.logger
+        assert app.logger is logger1
+        assert logger1.name == __name__
+        app.logger_name = __name__ + '/test_logger_cache'
+        assert app.logger is not logger1
+
+    def test_debug_log(self, capsys):
+        app = flask.Flask(__name__)
+        app.debug = True
+
+        @app.route('/')
+        def index():
+            app.logger.warning('the standard library is dead')
+            app.logger.debug('this is a debug statement')
+            return ''
+
+        @app.route('/exc')
+        def exc():
+            1 // 0
+
+        with app.test_client() as c:
+            c.get('/')
+            out, err = capsys.readouterr()
+            assert 'WARNING in test_helpers [' in err
+            assert os.path.basename(__file__.rsplit('.', 1)[0] + '.py') in err
+            assert 'the standard library is dead' in err
+            assert 'this is a debug statement' in err
+
+            with pytest.raises(ZeroDivisionError):
+                c.get('/exc')
+
+    def test_debug_log_override(self):
+        app = flask.Flask(__name__)
+        app.debug = True
+        app.logger_name = 'flask_tests/test_debug_log_override'
+        app.logger.level = 10
+        assert app.logger.level == 10
+
+    def test_exception_logging(self):
+        out = StringIO()
+        app = flask.Flask(__name__)
+        app.config['LOGGER_HANDLER_POLICY'] = 'never'
+        app.logger_name = 'flask_tests/test_exception_logging'
+        app.logger.addHandler(StreamHandler(out))
+
+        @app.route('/')
+        def index():
+            1 // 0
+
+        rv = app.test_client().get('/')
+        assert rv.status_code == 500
+        assert b'Internal Server Error' in rv.data
+
+        err = out.getvalue()
+        assert 'Exception on / [GET]' in err
+        assert 'Traceback (most recent call last):' in err
+        assert '1 // 0' in err
+        assert 'ZeroDivisionError:' in err
+
+    def test_processor_exceptions(self):
+        app = flask.Flask(__name__)
+        app.config['LOGGER_HANDLER_POLICY'] = 'never'
+        @app.before_request
+        def before_request():
+            if trigger == 'before':
+                1 // 0
+        @app.after_request
+        def after_request(response):
+            if trigger == 'after':
+                1 // 0
+            return response
+        @app.route('/')
+        def index():
+            return 'Foo'
+        @app.errorhandler(500)
+        def internal_server_error(e):
+            return 'Hello Server Error', 500
+        for trigger in 'before', 'after':
+            rv = app.test_client().get('/')
+            assert rv.status_code == 500
+            assert rv.data == b'Hello Server Error'
+
+    def test_url_for_with_anchor(self):
+        app = flask.Flask(__name__)
+        @app.route('/')
+        def index():
+            return '42'
+        with app.test_request_context():
+            assert flask.url_for('index', _anchor='x y') == '/#x%20y'
+
+    def test_url_for_with_scheme(self):
+        app = flask.Flask(__name__)
+        @app.route('/')
+        def index():
+            return '42'
+        with app.test_request_context():
+            assert flask.url_for('index', _external=True, _scheme='https') == 'https://localhost/'
+
+    def test_url_for_with_scheme_not_external(self):
+        app = flask.Flask(__name__)
+        @app.route('/')
+        def index():
+            return '42'
+        with app.test_request_context():
+            pytest.raises(ValueError,
+                               flask.url_for,
+                               'index',
+                               _scheme='https')
+
+    def test_url_for_with_alternating_schemes(self):
+        app = flask.Flask(__name__)
+        @app.route('/')
+        def index():
+            return '42'
+        with app.test_request_context():
+            assert flask.url_for('index', _external=True) == 'http://localhost/'
+            assert flask.url_for('index', _external=True, _scheme='https') == 'https://localhost/'
+            assert flask.url_for('index', _external=True) == 'http://localhost/'
+
+    def test_url_with_method(self):
+        from flask.views import MethodView
+        app = flask.Flask(__name__)
+        class MyView(MethodView):
+            def get(self, id=None):
+                if id is None:
+                    return 'List'
+                return 'Get %d' % id
+            def post(self):
+                return 'Create'
+        myview = MyView.as_view('myview')
+        app.add_url_rule('/myview/', methods=['GET'],
+                         view_func=myview)
+        app.add_url_rule('/myview/<int:id>', methods=['GET'],
+                         view_func=myview)
+        app.add_url_rule('/myview/create', methods=['POST'],
+                         view_func=myview)
+
+        with app.test_request_context():
+            assert flask.url_for('myview', _method='GET') == '/myview/'
+            assert flask.url_for('myview', id=42, _method='GET') == '/myview/42'
+            assert flask.url_for('myview', _method='POST') == '/myview/create'
+
+
+class TestNoImports(object):
+    """Test Flasks are created without import.
+
+    Avoiding ``__import__`` helps create Flask instances where there are errors
+    at import time.  Those runtime errors will be apparent to the user soon
+    enough, but tools which build Flask instances meta-programmatically benefit
+    from a Flask which does not ``__import__``.  Instead of importing to
+    retrieve file paths or metadata on a module or package, use the pkgutil and
+    imp modules in the Python standard library.
+    """
+
+    def test_name_with_import_error(self, modules_tmpdir):
+        modules_tmpdir.join('importerror.py').write('raise NotImplementedError()')
+        try:
+            flask.Flask('importerror')
+        except NotImplementedError:
+            assert False, 'Flask(import_name) is importing import_name.'
+
+
+class TestStreaming(object):
+
+    def test_streaming_with_context(self):
+        app = flask.Flask(__name__)
+        app.testing = True
+        @app.route('/')
+        def index():
+            def generate():
+                yield 'Hello '
+                yield flask.request.args['name']
+                yield '!'
+            return flask.Response(flask.stream_with_context(generate()))
+        c = app.test_client()
+        rv = c.get('/?name=World')
+        assert rv.data == b'Hello World!'
+
+    def test_streaming_with_context_as_decorator(self):
+        app = flask.Flask(__name__)
+        app.testing = True
+        @app.route('/')
+        def index():
+            @flask.stream_with_context
+            def generate(hello):
+                yield hello
+                yield flask.request.args['name']
+                yield '!'
+            return flask.Response(generate('Hello '))
+        c = app.test_client()
+        rv = c.get('/?name=World')
+        assert rv.data == b'Hello World!'
+
+    def test_streaming_with_context_and_custom_close(self):
+        app = flask.Flask(__name__)
+        app.testing = True
+        called = []
+        class Wrapper(object):
+            def __init__(self, gen):
+                self._gen = gen
+            def __iter__(self):
+                return self
+            def close(self):
+                called.append(42)
+            def __next__(self):
+                return next(self._gen)
+            next = __next__
+        @app.route('/')
+        def index():
+            def generate():
+                yield 'Hello '
+                yield flask.request.args['name']
+                yield '!'
+            return flask.Response(flask.stream_with_context(
+                Wrapper(generate())))
+        c = app.test_client()
+        rv = c.get('/?name=World')
+        assert rv.data == b'Hello World!'
+        assert called == [42]
+
+
+class TestSafeJoin(object):
+    def test_safe_join(self):
+        # Valid combinations of *args and expected joined paths.
+        passing = (
+            (('a/b/c',), 'a/b/c'),
+            (('/', 'a/', 'b/', 'c/'), '/a/b/c'),
+            (('a', 'b', 'c'), 'a/b/c'),
+            (('/a', 'b/c'), '/a/b/c'),
+            (('a/b', 'X/../c'), 'a/b/c'),
+            (('/a/b', 'c/X/..'), '/a/b/c'),
+            # If last path is '' add a slash
+            (('/a/b/c', ''), '/a/b/c/'),
+            # Preserve dot slash
+            (('/a/b/c', './'), '/a/b/c/.'),
+            (('a/b/c', 'X/..'), 'a/b/c/.'),
+            # Base directory is always considered safe
+            (('../', 'a/b/c'), '../a/b/c'),
+            (('/..', ), '/..'),
+        )
+
+        for args, expected in passing:
+            assert flask.safe_join(*args) == expected
+
+    def test_safe_join_exceptions(self):
+        # Should raise werkzeug.exceptions.NotFound on unsafe joins.
+        failing = (
+            # path.isabs and ``..'' checks
+            ('/a', 'b', '/c'),
+            ('/a', '../b/c'),
+            ('/a', '..', 'b/c'),
+            # Boundaries violations after path normalization
+            ('/a', 'b/../b/../../c'),
+            ('/a', 'b', 'c/../..'),
+            ('/a', 'b/../../c'),
+        )
+
+        for args in failing:
+            with pytest.raises(NotFound):
+                print(flask.safe_join(*args))

tests/test_instance_config.py

@@ -0,0 +1,134 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.test_instance
+    ~~~~~~~~~~~~~~~~~~~
+
+    :copyright: (c) 2015 by the Flask Team, see AUTHORS for more details.
+    :license: BSD, see LICENSE for more details.
+"""
+import os
+import sys
+
+import pytest
+import flask
+from flask._compat import PY2
+
+
+def test_explicit_instance_paths(modules_tmpdir):
+    with pytest.raises(ValueError) as excinfo:
+        flask.Flask(__name__, instance_path='instance')
+    assert 'must be absolute' in str(excinfo.value)
+
+    app = flask.Flask(__name__, instance_path=str(modules_tmpdir))
+    assert app.instance_path == str(modules_tmpdir)
+
+
+def test_main_module_paths(modules_tmpdir, purge_module):
+    app = modules_tmpdir.join('main_app.py')
+    app.write('import flask\n\napp = flask.Flask("__main__")')
+    purge_module('main_app')
+
+    from main_app import app
+    here = os.path.abspath(os.getcwd())
+    assert app.instance_path == os.path.join(here, 'instance')
+
+
+def test_uninstalled_module_paths(modules_tmpdir, purge_module):
+    app = modules_tmpdir.join('config_module_app.py').write(
+        'import os\n'
+        'import flask\n'
+        'here = os.path.abspath(os.path.dirname(__file__))\n'
+        'app = flask.Flask(__name__)\n'
+    )
+    purge_module('config_module_app')
+
+    from config_module_app import app
+    assert app.instance_path == str(modules_tmpdir.join('instance'))
+
+
+def test_uninstalled_package_paths(modules_tmpdir, purge_module):
+    app = modules_tmpdir.mkdir('config_package_app')
+    init = app.join('__init__.py')
+    init.write(
+        'import os\n'
+        'import flask\n'
+        'here = os.path.abspath(os.path.dirname(__file__))\n'
+        'app = flask.Flask(__name__)\n'
+    )
+    purge_module('config_package_app')
+
+    from config_package_app import app
+    assert app.instance_path == str(modules_tmpdir.join('instance'))
+
+
+def test_installed_module_paths(modules_tmpdir, modules_tmpdir_prefix,
+                                purge_module, site_packages, limit_loader):
+    site_packages.join('site_app.py').write(
+        'import flask\n'
+        'app = flask.Flask(__name__)\n'
+    )
+    purge_module('site_app')
+
+    from site_app import app
+    assert app.instance_path == \
+        modules_tmpdir.join('var').join('site_app-instance')
+
+
+def test_installed_package_paths(limit_loader, modules_tmpdir,
+                                 modules_tmpdir_prefix, purge_module,
+                                 monkeypatch):
+    installed_path = modules_tmpdir.mkdir('path')
+    monkeypatch.syspath_prepend(installed_path)
+
+    app = installed_path.mkdir('installed_package')
+    init = app.join('__init__.py')
+    init.write('import flask\napp = flask.Flask(__name__)')
+    purge_module('installed_package')
+
+    from installed_package import app
+    assert app.instance_path == \
+        modules_tmpdir.join('var').join('installed_package-instance')
+
+
+def test_prefix_package_paths(limit_loader, modules_tmpdir,
+                              modules_tmpdir_prefix, purge_module,
+                              site_packages):
+    app = site_packages.mkdir('site_package')
+    init = app.join('__init__.py')
+    init.write('import flask\napp = flask.Flask(__name__)')
+    purge_module('site_package')
+
+    import site_package
+    assert site_package.app.instance_path == \
+        modules_tmpdir.join('var').join('site_package-instance')
+
+
+def test_egg_installed_paths(install_egg, modules_tmpdir,
+                             modules_tmpdir_prefix):
+    modules_tmpdir.mkdir('site_egg').join('__init__.py').write(
+        'import flask\n\napp = flask.Flask(__name__)'
+    )
+    install_egg('site_egg')
+    try:
+        import site_egg
+        assert site_egg.app.instance_path == \
+            str(modules_tmpdir.join('var/').join('site_egg-instance'))
+    finally:
+        if 'site_egg' in sys.modules:
+            del sys.modules['site_egg']
+
+
+@pytest.mark.skipif(not PY2, reason='This only works under Python 2.')
+def test_meta_path_loader_without_is_package(request, modules_tmpdir):
+    app = modules_tmpdir.join('unimportable.py')
+    app.write('import flask\napp = flask.Flask(__name__)')
+
+    class Loader(object):
+        def find_module(self, name, path=None):
+            return self
+
+    sys.meta_path.append(Loader())
+    request.addfinalizer(sys.meta_path.pop)
+
+    with pytest.raises(AttributeError):
+        import unimportable

tests/test_regression.py

@@ -0,0 +1,102 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.regression
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Tests regressions.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import os
+import gc
+import sys
+import flask
+import threading
+from werkzeug.exceptions import NotFound
+
+
+_gc_lock = threading.Lock()
+
+
+class assert_no_leak(object):
+
+    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()
+
+
+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()
+
+    # This test only works on CPython 2.7.
+    if sys.version_info >= (2, 7) and \
+            not hasattr(sys, 'pypy_translation_info'):
+        with assert_no_leak():
+            for x in range(10):
+                fire()
+
+
+def test_safe_join_toplevel_pardir():
+    from flask.helpers import safe_join
+    with pytest.raises(NotFound):
+        safe_join('/foo', '..')
+
+
+def test_aborting():
+    class Foo(Exception):
+        whatever = 42
+    app = flask.Flask(__name__)
+    app.testing = True
+
+    @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'

tests/test_reqctx.py

@@ -0,0 +1,220 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.reqctx
+    ~~~~~~~~~~~~
+
+    Tests the request context.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import flask
+from flask.sessions import SessionInterface
+
+try:
+    from greenlet import greenlet
+except ImportError:
+    greenlet = None
+
+
+def test_teardown_on_pop():
+    buffer = []
+    app = flask.Flask(__name__)
+    @app.teardown_request
+    def end_of_request(exception):
+        buffer.append(exception)
+
+    ctx = app.test_request_context()
+    ctx.push()
+    assert buffer == []
+    ctx.pop()
+    assert buffer == [None]
+
+def test_teardown_with_previous_exception():
+    buffer = []
+    app = flask.Flask(__name__)
+    @app.teardown_request
+    def end_of_request(exception):
+        buffer.append(exception)
+
+    try:
+        raise Exception('dummy')
+    except Exception:
+        pass
+
+    with app.test_request_context():
+        assert buffer == []
+    assert buffer == [None]
+
+def test_teardown_with_handled_exception():
+    buffer = []
+    app = flask.Flask(__name__)
+    @app.teardown_request
+    def end_of_request(exception):
+        buffer.append(exception)
+
+    with app.test_request_context():
+        assert buffer == []
+        try:
+            raise Exception('dummy')
+        except Exception:
+            pass
+    assert buffer == [None]
+
+def test_proper_test_request_context():
+    app = flask.Flask(__name__)
+    app.config.update(
+        SERVER_NAME='localhost.localdomain:5000'
+    )
+
+    @app.route('/')
+    def index():
+        return None
+
+    @app.route('/', subdomain='foo')
+    def sub():
+        return None
+
+    with app.test_request_context('/'):
+        assert flask.url_for('index', _external=True) == \
+            'http://localhost.localdomain:5000/'
+
+    with app.test_request_context('/'):
+        assert flask.url_for('sub', _external=True) == \
+            'http://foo.localhost.localdomain:5000/'
+
+    try:
+        with app.test_request_context('/', environ_overrides={'HTTP_HOST': 'localhost'}):
+            pass
+    except ValueError as e:
+        assert str(e) == (
+            "the server name provided "
+            "('localhost.localdomain:5000') does not match the "
+            "server name from the WSGI environment ('localhost')"
+        )
+
+    app.config.update(SERVER_NAME='localhost')
+    with app.test_request_context('/', environ_overrides={'SERVER_NAME': 'localhost'}):
+        pass
+
+    app.config.update(SERVER_NAME='localhost:80')
+    with app.test_request_context('/', environ_overrides={'SERVER_NAME': 'localhost:80'}):
+        pass
+
+def test_context_binding():
+    app = flask.Flask(__name__)
+    @app.route('/')
+    def index():
+        return 'Hello %s!' % flask.request.args['name']
+    @app.route('/meh')
+    def meh():
+        return flask.request.url
+
+    with app.test_request_context('/?name=World'):
+        assert index() == 'Hello World!'
+    with app.test_request_context('/meh'):
+        assert meh() == 'http://localhost/meh'
+    assert flask._request_ctx_stack.top is None
+
+def test_context_test():
+    app = flask.Flask(__name__)
+    assert not flask.request
+    assert not flask.has_request_context()
+    ctx = app.test_request_context()
+    ctx.push()
+    try:
+        assert flask.request
+        assert flask.has_request_context()
+    finally:
+        ctx.pop()
+
+def test_manual_context_binding():
+    app = flask.Flask(__name__)
+    @app.route('/')
+    def index():
+        return 'Hello %s!' % flask.request.args['name']
+
+    ctx = app.test_request_context('/?name=World')
+    ctx.push()
+    assert index() == 'Hello World!'
+    ctx.pop()
+    with pytest.raises(RuntimeError):
+        index()
+
+@pytest.mark.skipif(greenlet is None, reason='greenlet not installed')
+def test_greenlet_context_copying():
+    app = flask.Flask(__name__)
+    greenlets = []
+
+    @app.route('/')
+    def index():
+        reqctx = flask._request_ctx_stack.top.copy()
+        def g():
+            assert not flask.request
+            assert not flask.current_app
+            with reqctx:
+                assert flask.request
+                assert flask.current_app == app
+                assert flask.request.path == '/'
+                assert flask.request.args['foo'] == 'bar'
+            assert not flask.request
+            return 42
+        greenlets.append(greenlet(g))
+        return 'Hello World!'
+
+    rv = app.test_client().get('/?foo=bar')
+    assert rv.data == b'Hello World!'
+
+    result = greenlets[0].run()
+    assert result == 42
+
+@pytest.mark.skipif(greenlet is None, reason='greenlet not installed')
+def test_greenlet_context_copying_api():
+    app = flask.Flask(__name__)
+    greenlets = []
+
+    @app.route('/')
+    def index():
+        reqctx = flask._request_ctx_stack.top.copy()
+        @flask.copy_current_request_context
+        def g():
+            assert flask.request
+            assert flask.current_app == app
+            assert flask.request.path == '/'
+            assert flask.request.args['foo'] == 'bar'
+            return 42
+        greenlets.append(greenlet(g))
+        return 'Hello World!'
+
+    rv = app.test_client().get('/?foo=bar')
+    assert rv.data == b'Hello World!'
+
+    result = greenlets[0].run()
+    assert result == 42
+
+
+def test_session_error_pops_context():
+    class SessionError(Exception):
+        pass
+
+    class FailingSessionInterface(SessionInterface):
+        def open_session(self, app, request):
+            raise SessionError()
+
+    class CustomFlask(flask.Flask):
+        session_interface = FailingSessionInterface()
+
+    app = CustomFlask(__name__)
+
+    @app.route('/')
+    def index():
+        # shouldn't get here
+        assert False
+
+    response = app.test_client().get('/')
+    assert response.status_code == 500
+    assert not flask.request
+    assert not flask.current_app

tests/test_signals.py

@@ -0,0 +1,202 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.signals
+    ~~~~~~~~~~~~~~~~~~~~~~~
+
+    Signalling.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+try:
+    import blinker
+except ImportError:
+    blinker = None
+
+import flask
+
+
+pytestmark = pytest.mark.skipif(
+    blinker is None,
+    reason='Signals require the blinker library.'
+)
+
+def test_template_rendered():
+    app = flask.Flask(__name__)
+
+    @app.route('/')
+    def index():
+        return flask.render_template('simple_template.html', whiskey=42)
+
+    recorded = []
+
+    def record(sender, template, context):
+        recorded.append((template, context))
+
+    flask.template_rendered.connect(record, app)
+    try:
+        app.test_client().get('/')
+        assert len(recorded) == 1
+        template, context = recorded[0]
+        assert template.name == 'simple_template.html'
+        assert context['whiskey'] == 42
+    finally:
+        flask.template_rendered.disconnect(record, app)
+
+def test_before_render_template():
+    app = flask.Flask(__name__)
+
+    @app.route('/')
+    def index():
+        return flask.render_template('simple_template.html', whiskey=42)
+
+    recorded = []
+
+    def record(sender, template, context):
+        context['whiskey'] = 43
+        recorded.append((template, context))
+
+    flask.before_render_template.connect(record, app)
+    try:
+        rv = app.test_client().get('/')
+        assert len(recorded) == 1
+        template, context = recorded[0]
+        assert template.name == 'simple_template.html'
+        assert context['whiskey'] == 43
+        assert rv.data == b'<h1>43</h1>'
+    finally:
+        flask.before_render_template.disconnect(record, app)
+
+def test_request_signals():
+    app = flask.Flask(__name__)
+    calls = []
+
+    def before_request_signal(sender):
+        calls.append('before-signal')
+
+    def after_request_signal(sender, response):
+        assert response.data == b'stuff'
+        calls.append('after-signal')
+
+    @app.before_request
+    def before_request_handler():
+        calls.append('before-handler')
+
+    @app.after_request
+    def after_request_handler(response):
+        calls.append('after-handler')
+        response.data = 'stuff'
+        return response
+
+    @app.route('/')
+    def index():
+        calls.append('handler')
+        return 'ignored anyway'
+
+    flask.request_started.connect(before_request_signal, app)
+    flask.request_finished.connect(after_request_signal, app)
+
+    try:
+        rv = app.test_client().get('/')
+        assert rv.data == b'stuff'
+
+        assert calls == ['before-signal', 'before-handler', 'handler',
+                         'after-handler', 'after-signal']
+    finally:
+        flask.request_started.disconnect(before_request_signal, app)
+        flask.request_finished.disconnect(after_request_signal, app)
+
+def test_request_exception_signal():
+    app = flask.Flask(__name__)
+    recorded = []
+
+    @app.route('/')
+    def index():
+        1 // 0
+
+    def record(sender, exception):
+        recorded.append(exception)
+
+    flask.got_request_exception.connect(record, app)
+    try:
+        assert app.test_client().get('/').status_code == 500
+        assert len(recorded) == 1
+        assert isinstance(recorded[0], ZeroDivisionError)
+    finally:
+        flask.got_request_exception.disconnect(record, app)
+
+def test_appcontext_signals():
+    app = flask.Flask(__name__)
+    recorded = []
+
+    def record_push(sender, **kwargs):
+        recorded.append('push')
+
+    def record_pop(sender, **kwargs):
+        recorded.append('pop')
+
+    @app.route('/')
+    def index():
+        return 'Hello'
+
+    flask.appcontext_pushed.connect(record_push, app)
+    flask.appcontext_popped.connect(record_pop, app)
+    try:
+        with app.test_client() as c:
+            rv = c.get('/')
+            assert rv.data == b'Hello'
+            assert recorded == ['push']
+        assert recorded == ['push', 'pop']
+    finally:
+        flask.appcontext_pushed.disconnect(record_push, app)
+        flask.appcontext_popped.disconnect(record_pop, app)
+
+def test_flash_signal():
+    app = flask.Flask(__name__)
+    app.config['SECRET_KEY'] = 'secret'
+
+    @app.route('/')
+    def index():
+        flask.flash('This is a flash message', category='notice')
+        return flask.redirect('/other')
+
+    recorded = []
+
+    def record(sender, message, category):
+        recorded.append((message, category))
+
+    flask.message_flashed.connect(record, app)
+    try:
+        client = app.test_client()
+        with client.session_transaction():
+            client.get('/')
+            assert len(recorded) == 1
+            message, category = recorded[0]
+            assert message == 'This is a flash message'
+            assert category == 'notice'
+    finally:
+        flask.message_flashed.disconnect(record, app)
+
+def test_appcontext_tearing_down_signal():
+    app = flask.Flask(__name__)
+    recorded = []
+
+    def record_teardown(sender, **kwargs):
+        recorded.append(('tear_down', kwargs))
+
+    @app.route('/')
+    def index():
+        1 // 0
+
+    flask.appcontext_tearing_down.connect(record_teardown, app)
+    try:
+        with app.test_client() as c:
+            rv = c.get('/')
+            assert rv.status_code == 500
+            assert recorded == []
+        assert recorded == [('tear_down', {'exc': None})]
+    finally:
+        flask.appcontext_tearing_down.disconnect(record_teardown, app)

tests/test_subclassing.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.subclassing
+    ~~~~~~~~~~~~~~~~~
+
+    Test that certain behavior of flask can be customized by
+    subclasses.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+import flask
+from logging import StreamHandler
+
+from flask._compat import StringIO
+
+
+def test_suppressed_exception_logging():
+    class SuppressedFlask(flask.Flask):
+        def log_exception(self, exc_info):
+            pass
+
+    out = StringIO()
+    app = SuppressedFlask(__name__)
+    app.logger_name = 'flask_tests/test_suppressed_exception_logging'
+    app.logger.addHandler(StreamHandler(out))
+
+    @app.route('/')
+    def index():
+        1 // 0
+
+    rv = app.test_client().get('/')
+    assert rv.status_code == 500
+    assert b'Internal Server Error' in rv.data
+
+    err = out.getvalue()
+    assert err == ''

tests/test_templating.py

@@ -0,0 +1,392 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.templating
+    ~~~~~~~~~~~~~~~~
+
+    Template functionality
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import flask
+import logging
+from jinja2 import TemplateNotFound
+
+
+def test_context_processing():
+    app = flask.Flask(__name__)
+    @app.context_processor
+    def context_processor():
+        return {'injected_value': 42}
+    @app.route('/')
+    def index():
+        return flask.render_template('context_template.html', value=23)
+    rv = app.test_client().get('/')
+    assert rv.data == b'<p>23|42'
+
+def test_original_win():
+    app = flask.Flask(__name__)
+    @app.route('/')
+    def index():
+        return flask.render_template_string('{{ config }}', config=42)
+    rv = app.test_client().get('/')
+    assert rv.data == b'42'
+
+def test_request_less_rendering():
+    app = flask.Flask(__name__)
+    app.config['WORLD_NAME'] = 'Special World'
+    @app.context_processor
+    def context_processor():
+        return dict(foo=42)
+
+    with app.app_context():
+        rv = flask.render_template_string('Hello {{ config.WORLD_NAME }} '
+                                          '{{ foo }}')
+        assert rv == 'Hello Special World 42'
+
+def test_standard_context():
+    app = flask.Flask(__name__)
+    app.secret_key = 'development key'
+    @app.route('/')
+    def index():
+        flask.g.foo = 23
+        flask.session['test'] = 'aha'
+        return flask.render_template_string('''
+            {{ request.args.foo }}
+            {{ g.foo }}
+            {{ config.DEBUG }}
+            {{ session.test }}
+        ''')
+    rv = app.test_client().get('/?foo=42')
+    assert rv.data.split() == [b'42', b'23', b'False', b'aha']
+
+def test_escaping():
+    text = '<p>Hello World!'
+    app = flask.Flask(__name__)
+    @app.route('/')
+    def index():
+        return flask.render_template('escaping_template.html', text=text,
+                                     html=flask.Markup(text))
+    lines = app.test_client().get('/').data.splitlines()
+    assert lines == [
+        b'&lt;p&gt;Hello World!',
+        b'<p>Hello World!',
+        b'<p>Hello World!',
+        b'<p>Hello World!',
+        b'&lt;p&gt;Hello World!',
+        b'<p>Hello World!'
+    ]
+
+def test_no_escaping():
+    text = '<p>Hello World!'
+    app = flask.Flask(__name__)
+    @app.route('/')
+    def index():
+        return flask.render_template('non_escaping_template.txt', text=text,
+                                     html=flask.Markup(text))
+    lines = app.test_client().get('/').data.splitlines()
+    assert lines == [
+        b'<p>Hello World!',
+        b'<p>Hello World!',
+        b'<p>Hello World!',
+        b'<p>Hello World!',
+        b'&lt;p&gt;Hello World!',
+        b'<p>Hello World!',
+        b'<p>Hello World!',
+        b'<p>Hello World!'
+    ]
+
+def test_escaping_without_template_filename():
+    app = flask.Flask(__name__)
+    with app.test_request_context():
+        assert flask.render_template_string(
+            '{{ foo }}', foo='<test>') == '&lt;test&gt;'
+        assert flask.render_template('mail.txt', foo='<test>') == \
+            '<test> Mail'
+
+def test_macros():
+    app = flask.Flask(__name__)
+    with app.test_request_context():
+        macro = flask.get_template_attribute('_macro.html', 'hello')
+        assert macro('World') == 'Hello World!'
+
+def test_template_filter():
+    app = flask.Flask(__name__)
+    @app.template_filter()
+    def my_reverse(s):
+        return s[::-1]
+    assert 'my_reverse' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['my_reverse'] == my_reverse
+    assert app.jinja_env.filters['my_reverse']('abcd') == 'dcba'
+
+def test_add_template_filter():
+    app = flask.Flask(__name__)
+    def my_reverse(s):
+        return s[::-1]
+    app.add_template_filter(my_reverse)
+    assert 'my_reverse' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['my_reverse'] == my_reverse
+    assert app.jinja_env.filters['my_reverse']('abcd') == 'dcba'
+
+def test_template_filter_with_name():
+    app = flask.Flask(__name__)
+    @app.template_filter('strrev')
+    def my_reverse(s):
+        return s[::-1]
+    assert 'strrev' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['strrev'] == my_reverse
+    assert app.jinja_env.filters['strrev']('abcd') == 'dcba'
+
+def test_add_template_filter_with_name():
+    app = flask.Flask(__name__)
+    def my_reverse(s):
+        return s[::-1]
+    app.add_template_filter(my_reverse, 'strrev')
+    assert 'strrev' in app.jinja_env.filters.keys()
+    assert app.jinja_env.filters['strrev'] == my_reverse
+    assert app.jinja_env.filters['strrev']('abcd') == 'dcba'
+
+def test_template_filter_with_template():
+    app = flask.Flask(__name__)
+    @app.template_filter()
+    def super_reverse(s):
+        return s[::-1]
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_add_template_filter_with_template():
+    app = flask.Flask(__name__)
+    def super_reverse(s):
+        return s[::-1]
+    app.add_template_filter(super_reverse)
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_template_filter_with_name_and_template():
+    app = flask.Flask(__name__)
+    @app.template_filter('super_reverse')
+    def my_reverse(s):
+        return s[::-1]
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_add_template_filter_with_name_and_template():
+    app = flask.Flask(__name__)
+    def my_reverse(s):
+        return s[::-1]
+    app.add_template_filter(my_reverse, 'super_reverse')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_filter.html', value='abcd')
+    rv = app.test_client().get('/')
+    assert rv.data == b'dcba'
+
+def test_template_test():
+    app = flask.Flask(__name__)
+    @app.template_test()
+    def boolean(value):
+        return isinstance(value, bool)
+    assert 'boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['boolean'] == boolean
+    assert app.jinja_env.tests['boolean'](False)
+
+def test_add_template_test():
+    app = flask.Flask(__name__)
+    def boolean(value):
+        return isinstance(value, bool)
+    app.add_template_test(boolean)
+    assert 'boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['boolean'] == boolean
+    assert app.jinja_env.tests['boolean'](False)
+
+def test_template_test_with_name():
+    app = flask.Flask(__name__)
+    @app.template_test('boolean')
+    def is_boolean(value):
+        return isinstance(value, bool)
+    assert 'boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['boolean'] == is_boolean
+    assert app.jinja_env.tests['boolean'](False)
+
+def test_add_template_test_with_name():
+    app = flask.Flask(__name__)
+    def is_boolean(value):
+        return isinstance(value, bool)
+    app.add_template_test(is_boolean, 'boolean')
+    assert 'boolean' in app.jinja_env.tests.keys()
+    assert app.jinja_env.tests['boolean'] == is_boolean
+    assert app.jinja_env.tests['boolean'](False)
+
+def test_template_test_with_template():
+    app = flask.Flask(__name__)
+    @app.template_test()
+    def boolean(value):
+        return isinstance(value, bool)
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_add_template_test_with_template():
+    app = flask.Flask(__name__)
+    def boolean(value):
+        return isinstance(value, bool)
+    app.add_template_test(boolean)
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_template_test_with_name_and_template():
+    app = flask.Flask(__name__)
+    @app.template_test('boolean')
+    def is_boolean(value):
+        return isinstance(value, bool)
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_add_template_test_with_name_and_template():
+    app = flask.Flask(__name__)
+    def is_boolean(value):
+        return isinstance(value, bool)
+    app.add_template_test(is_boolean, 'boolean')
+    @app.route('/')
+    def index():
+        return flask.render_template('template_test.html', value=False)
+    rv = app.test_client().get('/')
+    assert b'Success!' in rv.data
+
+def test_add_template_global():
+    app = flask.Flask(__name__)
+    @app.template_global()
+    def get_stuff():
+        return 42
+    assert 'get_stuff' in app.jinja_env.globals.keys()
+    assert app.jinja_env.globals['get_stuff'] == get_stuff
+    assert app.jinja_env.globals['get_stuff'](), 42
+    with app.app_context():
+        rv = flask.render_template_string('{{ get_stuff() }}')
+        assert rv == '42'
+
+def test_custom_template_loader():
+    class MyFlask(flask.Flask):
+        def create_global_jinja_loader(self):
+            from jinja2 import DictLoader
+            return DictLoader({'index.html': 'Hello Custom World!'})
+    app = MyFlask(__name__)
+    @app.route('/')
+    def index():
+        return flask.render_template('index.html')
+    c = app.test_client()
+    rv = c.get('/')
+    assert rv.data == b'Hello Custom World!'
+
+def test_iterable_loader():
+    app = flask.Flask(__name__)
+    @app.context_processor
+    def context_processor():
+        return {'whiskey': 'Jameson'}
+    @app.route('/')
+    def index():
+        return flask.render_template(
+            ['no_template.xml',  # should skip this one
+            'simple_template.html',  # should render this
+            'context_template.html'],
+            value=23)
+
+    rv = app.test_client().get('/')
+    assert rv.data == b'<h1>Jameson</h1>'
+
+def test_templates_auto_reload():
+    # debug is False, config option is None
+    app = flask.Flask(__name__)
+    assert app.debug is False
+    assert app.config['TEMPLATES_AUTO_RELOAD'] is None
+    assert app.jinja_env.auto_reload is False
+    # debug is False, config option is False
+    app = flask.Flask(__name__)
+    app.config['TEMPLATES_AUTO_RELOAD'] = False
+    assert app.debug is False
+    assert app.jinja_env.auto_reload is False
+    # debug is False, config option is True
+    app = flask.Flask(__name__)
+    app.config['TEMPLATES_AUTO_RELOAD'] = True
+    assert app.debug is False
+    assert app.jinja_env.auto_reload is True
+    # debug is True, config option is None
+    app = flask.Flask(__name__)
+    app.config['DEBUG'] = True
+    assert app.config['TEMPLATES_AUTO_RELOAD'] is None
+    assert app.jinja_env.auto_reload is True
+    # debug is True, config option is False
+    app = flask.Flask(__name__)
+    app.config['DEBUG'] = True
+    app.config['TEMPLATES_AUTO_RELOAD'] = False
+    assert app.jinja_env.auto_reload is False
+    # debug is True, config option is True
+    app = flask.Flask(__name__)
+    app.config['DEBUG'] = True
+    app.config['TEMPLATES_AUTO_RELOAD'] = True
+    assert app.jinja_env.auto_reload is True
+
+def test_template_loader_debugging(test_apps):
+    from blueprintapp import app
+
+    called = []
+    class _TestHandler(logging.Handler):
+        def handle(x, record):
+            called.append(True)
+            text = str(record.msg)
+            assert '1: trying loader of application "blueprintapp"' in text
+            assert ('2: trying loader of blueprint "admin" '
+                    '(blueprintapp.apps.admin)') in text
+            assert ('trying loader of blueprint "frontend" '
+                    '(blueprintapp.apps.frontend)') in text
+            assert 'Error: the template could not be found' in text
+            assert ('looked up from an endpoint that belongs to '
+                    'the blueprint "frontend"') in text
+            assert 'See http://flask.pocoo.org/docs/blueprints/#templates' in text
+
+    with app.test_client() as c:
+        try:
+            old_load_setting = app.config['EXPLAIN_TEMPLATE_LOADING']
+            old_handlers = app.logger.handlers[:]
+            app.logger.handlers = [_TestHandler()]
+            app.config['EXPLAIN_TEMPLATE_LOADING'] = True
+
+            with pytest.raises(TemplateNotFound) as excinfo:
+                c.get('/missing')
+
+            assert 'missing_template.html' in str(excinfo.value)
+        finally:
+            app.logger.handlers[:] = old_handlers
+            app.config['EXPLAIN_TEMPLATE_LOADING'] = old_load_setting
+
+    assert len(called) == 1
+
+def test_custom_jinja_env():
+    class CustomEnvironment(flask.templating.Environment):
+        pass
+
+    class CustomFlask(flask.Flask):
+        jinja_environment = CustomEnvironment
+
+    app = CustomFlask(__name__)
+    assert isinstance(app.jinja_env, CustomEnvironment)

tests/test_testing.py

@@ -0,0 +1,271 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.testing
+    ~~~~~~~~~~~~~
+
+    Test client and more.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+import pytest
+
+import flask
+import werkzeug
+
+from flask._compat import text_type
+
+
+def test_environ_defaults_from_config():
+    app = flask.Flask(__name__)
+    app.testing = True
+    app.config['SERVER_NAME'] = 'example.com:1234'
+    app.config['APPLICATION_ROOT'] = '/foo'
+    @app.route('/')
+    def index():
+        return flask.request.url
+
+    ctx = app.test_request_context()
+    assert ctx.request.url == 'http://example.com:1234/foo/'
+    with app.test_client() as c:
+        rv = c.get('/')
+        assert rv.data == b'http://example.com:1234/foo/'
+
+def test_environ_defaults():
+    app = flask.Flask(__name__)
+    app.testing = True
+    @app.route('/')
+    def index():
+        return flask.request.url
+
+    ctx = app.test_request_context()
+    assert ctx.request.url == 'http://localhost/'
+    with app.test_client() as c:
+        rv = c.get('/')
+        assert rv.data == b'http://localhost/'
+
+def test_environ_base_default():
+    app = flask.Flask(__name__)
+    app.testing = True
+    @app.route('/')
+    def index():
+        flask.g.user_agent = flask.request.headers["User-Agent"]
+        return flask.request.remote_addr
+
+    with app.test_client() as c:
+        rv = c.get('/')
+        assert rv.data == b'127.0.0.1'
+        assert flask.g.user_agent == 'werkzeug/' + werkzeug.__version__
+
+def test_environ_base_modified():
+    app = flask.Flask(__name__)
+    app.testing = True
+    @app.route('/')
+    def index():
+        flask.g.user_agent = flask.request.headers["User-Agent"]
+        return flask.request.remote_addr
+
+    with app.test_client() as c:
+        c.environ_base['REMOTE_ADDR'] = '0.0.0.0'
+        c.environ_base['HTTP_USER_AGENT'] = 'Foo'
+        rv = c.get('/')
+        assert rv.data == b'0.0.0.0'
+        assert flask.g.user_agent == 'Foo'
+
+        c.environ_base['REMOTE_ADDR'] = '0.0.0.1'
+        c.environ_base['HTTP_USER_AGENT'] = 'Bar'
+        rv = c.get('/')
+        assert rv.data == b'0.0.0.1'
+        assert flask.g.user_agent == 'Bar'
+
+def test_redirect_keep_session():
+    app = flask.Flask(__name__)
+    app.secret_key = 'testing'
+
+    @app.route('/', methods=['GET', 'POST'])
+    def index():
+        if flask.request.method == 'POST':
+            return flask.redirect('/getsession')
+        flask.session['data'] = 'foo'
+        return 'index'
+
+    @app.route('/getsession')
+    def get_session():
+        return flask.session.get('data', '<missing>')
+
+    with app.test_client() as c:
+        rv = c.get('/getsession')
+        assert rv.data == b'<missing>'
+
+        rv = c.get('/')
+        assert rv.data == b'index'
+        assert flask.session.get('data') == 'foo'
+        rv = c.post('/', data={}, follow_redirects=True)
+        assert rv.data == b'foo'
+
+        # This support requires a new Werkzeug version
+        if not hasattr(c, 'redirect_client'):
+            assert flask.session.get('data') == 'foo'
+
+        rv = c.get('/getsession')
+        assert rv.data == b'foo'
+
+def test_session_transactions():
+    app = flask.Flask(__name__)
+    app.testing = True
+    app.secret_key = 'testing'
+
+    @app.route('/')
+    def index():
+        return text_type(flask.session['foo'])
+
+    with app.test_client() as c:
+        with c.session_transaction() as sess:
+            assert len(sess) == 0
+            sess['foo'] = [42]
+            assert len(sess) == 1
+        rv = c.get('/')
+        assert rv.data == b'[42]'
+        with c.session_transaction() as sess:
+            assert len(sess) == 1
+            assert sess['foo'] == [42]
+
+def test_session_transactions_no_null_sessions():
+    app = flask.Flask(__name__)
+    app.testing = True
+
+    with app.test_client() as c:
+        with pytest.raises(RuntimeError) as e:
+            with c.session_transaction() as sess:
+                pass
+        assert 'Session backend did not open a session' in str(e.value)
+
+def test_session_transactions_keep_context():
+    app = flask.Flask(__name__)
+    app.testing = True
+    app.secret_key = 'testing'
+
+    with app.test_client() as c:
+        rv = c.get('/')
+        req = flask.request._get_current_object()
+        assert req is not None
+        with c.session_transaction():
+            assert req is flask.request._get_current_object()
+
+def test_session_transaction_needs_cookies():
+    app = flask.Flask(__name__)
+    app.testing = True
+    c = app.test_client(use_cookies=False)
+    with pytest.raises(RuntimeError) as e:
+        with c.session_transaction() as s:
+            pass
+    assert 'cookies' in str(e.value)
+
+def test_test_client_context_binding():
+    app = flask.Flask(__name__)
+    app.config['LOGGER_HANDLER_POLICY'] = 'never'
+    @app.route('/')
+    def index():
+        flask.g.value = 42
+        return 'Hello World!'
+
+    @app.route('/other')
+    def other():
+        1 // 0
+
+    with app.test_client() as c:
+        resp = c.get('/')
+        assert flask.g.value == 42
+        assert resp.data == b'Hello World!'
+        assert resp.status_code == 200
+
+        resp = c.get('/other')
+        assert not hasattr(flask.g, 'value')
+        assert b'Internal Server Error' in resp.data
+        assert resp.status_code == 500
+        flask.g.value = 23
+
+    try:
+        flask.g.value
+    except (AttributeError, RuntimeError):
+        pass
+    else:
+        raise AssertionError('some kind of exception expected')
+
+def test_reuse_client():
+    app = flask.Flask(__name__)
+    c = app.test_client()
+
+    with c:
+        assert c.get('/').status_code == 404
+
+    with c:
+        assert c.get('/').status_code == 404
+
+def test_test_client_calls_teardown_handlers():
+    app = flask.Flask(__name__)
+    called = []
+    @app.teardown_request
+    def remember(error):
+        called.append(error)
+
+    with app.test_client() as c:
+        assert called == []
+        c.get('/')
+        assert called == []
+    assert called == [None]
+
+    del called[:]
+    with app.test_client() as c:
+        assert called == []
+        c.get('/')
+        assert called == []
+        c.get('/')
+        assert called == [None]
+    assert called == [None, None]
+
+def test_full_url_request():
+    app = flask.Flask(__name__)
+    app.testing = True
+
+    @app.route('/action', methods=['POST'])
+    def action():
+        return 'x'
+
+    with app.test_client() as c:
+        rv = c.post('http://domain.com/action?vodka=42', data={'gin': 43})
+        assert rv.status_code == 200
+        assert 'gin' in flask.request.form
+        assert 'vodka' in flask.request.args
+
+def test_subdomain():
+    app = flask.Flask(__name__)
+    app.config['SERVER_NAME'] = 'example.com'
+    @app.route('/', subdomain='<company_id>')
+    def view(company_id):
+        return company_id
+
+    with app.test_request_context():
+        url = flask.url_for('view', company_id='xxx')
+
+    with app.test_client() as c:
+        response = c.get(url)
+
+    assert 200 == response.status_code
+    assert b'xxx' == response.data
+
+def test_nosubdomain():
+    app = flask.Flask(__name__)
+    app.config['SERVER_NAME'] = 'example.com'
+    @app.route('/<company_id>')
+    def view(company_id):
+        return company_id
+
+    with app.test_request_context():
+        url = flask.url_for('view', company_id='xxx')
+
+    with app.test_client() as c:
+        response = c.get(url)
+
+    assert 200 == response.status_code
+    assert b'xxx' == response.data

tests/test_user_error_handler.py

@@ -0,0 +1,140 @@
+# -*- coding: utf-8 -*-
+from werkzeug.exceptions import Forbidden, InternalServerError
+import flask
+
+
+def test_error_handler_no_match():
+    app = flask.Flask(__name__)
+
+    class CustomException(Exception):
+        pass
+
+    @app.errorhandler(CustomException)
+    def custom_exception_handler(e):
+        assert isinstance(e, CustomException)
+        return 'custom'
+
+    @app.errorhandler(500)
+    def handle_500(e):
+        return type(e).__name__
+
+    @app.route('/custom')
+    def custom_test():
+        raise CustomException()
+
+    @app.route('/keyerror')
+    def key_error():
+        raise KeyError()
+
+    c = app.test_client()
+
+    assert c.get('/custom').data == b'custom'
+    assert c.get('/keyerror').data == b'KeyError'
+
+
+def test_error_handler_subclass():
+    app = flask.Flask(__name__)
+
+    class ParentException(Exception):
+        pass
+
+    class ChildExceptionUnregistered(ParentException):
+        pass
+
+    class ChildExceptionRegistered(ParentException):
+        pass
+
+    @app.errorhandler(ParentException)
+    def parent_exception_handler(e):
+        assert isinstance(e, ParentException)
+        return 'parent'
+
+    @app.errorhandler(ChildExceptionRegistered)
+    def child_exception_handler(e):
+        assert isinstance(e, ChildExceptionRegistered)
+        return 'child-registered'
+
+    @app.route('/parent')
+    def parent_test():
+        raise ParentException()
+
+    @app.route('/child-unregistered')
+    def unregistered_test():
+        raise ChildExceptionUnregistered()
+
+    @app.route('/child-registered')
+    def registered_test():
+        raise ChildExceptionRegistered()
+
+    c = app.test_client()
+
+    assert c.get('/parent').data == b'parent'
+    assert c.get('/child-unregistered').data == b'parent'
+    assert c.get('/child-registered').data == b'child-registered'
+
+
+def test_error_handler_http_subclass():
+    app = flask.Flask(__name__)
+
+    class ForbiddenSubclassRegistered(Forbidden):
+        pass
+
+    class ForbiddenSubclassUnregistered(Forbidden):
+        pass
+
+    @app.errorhandler(403)
+    def code_exception_handler(e):
+        assert isinstance(e, Forbidden)
+        return 'forbidden'
+
+    @app.errorhandler(ForbiddenSubclassRegistered)
+    def subclass_exception_handler(e):
+        assert isinstance(e, ForbiddenSubclassRegistered)
+        return 'forbidden-registered'
+
+    @app.route('/forbidden')
+    def forbidden_test():
+        raise Forbidden()
+
+    @app.route('/forbidden-registered')
+    def registered_test():
+        raise ForbiddenSubclassRegistered()
+
+    @app.route('/forbidden-unregistered')
+    def unregistered_test():
+        raise ForbiddenSubclassUnregistered()
+
+    c = app.test_client()
+
+    assert c.get('/forbidden').data == b'forbidden'
+    assert c.get('/forbidden-unregistered').data == b'forbidden'
+    assert c.get('/forbidden-registered').data == b'forbidden-registered'
+
+
+def test_error_handler_blueprint():
+    bp = flask.Blueprint('bp', __name__)
+
+    @bp.errorhandler(500)
+    def bp_exception_handler(e):
+        return 'bp-error'
+
+    @bp.route('/error')
+    def bp_test():
+        raise InternalServerError()
+
+    app = flask.Flask(__name__)
+
+    @app.errorhandler(500)
+    def app_exception_handler(e):
+        return 'app-error'
+
+    @app.route('/error')
+    def app_test():
+        raise InternalServerError()
+
+    app.register_blueprint(bp, url_prefix='/bp')
+
+    c = app.test_client()
+
+    assert c.get('/error').data == b'app-error'
+    assert c.get('/bp/error').data == b'bp-error'

tests/test_views.py

@@ -0,0 +1,204 @@
+# -*- coding: utf-8 -*-
+"""
+    tests.views
+    ~~~~~~~~~~~
+
+    Pluggable views.
+
+    :copyright: (c) 2015 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import pytest
+
+import flask
+import flask.views
+
+from werkzeug.http import parse_set_header
+
+def common_test(app):
+    c = app.test_client()
+
+    assert c.get('/').data == b'GET'
+    assert c.post('/').data == b'POST'
+    assert c.put('/').status_code == 405
+    meths = parse_set_header(c.open('/', method='OPTIONS').headers['Allow'])
+    assert sorted(meths) == ['GET', 'HEAD', 'OPTIONS', 'POST']
+
+def test_basic_view():
+    app = flask.Flask(__name__)
+
+    class Index(flask.views.View):
+        methods = ['GET', 'POST']
+        def dispatch_request(self):
+            return flask.request.method
+
+    app.add_url_rule('/', view_func=Index.as_view('index'))
+    common_test(app)
+
+def test_method_based_view():
+    app = flask.Flask(__name__)
+
+    class Index(flask.views.MethodView):
+        def get(self):
+            return 'GET'
+        def post(self):
+            return 'POST'
+
+    app.add_url_rule('/', view_func=Index.as_view('index'))
+
+    common_test(app)
+
+def test_view_patching():
+    app = flask.Flask(__name__)
+
+    class Index(flask.views.MethodView):
+        def get(self):
+            1 // 0
+        def post(self):
+            1 // 0
+
+    class Other(Index):
+        def get(self):
+            return 'GET'
+        def post(self):
+            return 'POST'
+
+    view = Index.as_view('index')
+    view.view_class = Other
+    app.add_url_rule('/', view_func=view)
+    common_test(app)
+
+def test_view_inheritance():
+    app = flask.Flask(__name__)
+
+    class Index(flask.views.MethodView):
+        def get(self):
+            return 'GET'
+        def post(self):
+            return 'POST'
+
+    class BetterIndex(Index):
+        def delete(self):
+            return 'DELETE'
+
+    app.add_url_rule('/', view_func=BetterIndex.as_view('index'))
+    c = app.test_client()
+
+    meths = parse_set_header(c.open('/', method='OPTIONS').headers['Allow'])
+    assert sorted(meths) == ['DELETE', 'GET', 'HEAD', 'OPTIONS', 'POST']
+
+def test_view_decorators():
+    app = flask.Flask(__name__)
+
+    def add_x_parachute(f):
+        def new_function(*args, **kwargs):
+            resp = flask.make_response(f(*args, **kwargs))
+            resp.headers['X-Parachute'] = 'awesome'
+            return resp
+        return new_function
+
+    class Index(flask.views.View):
+        decorators = [add_x_parachute]
+        def dispatch_request(self):
+            return 'Awesome'
+
+    app.add_url_rule('/', view_func=Index.as_view('index'))
+    c = app.test_client()
+    rv = c.get('/')
+    assert rv.headers['X-Parachute'] == 'awesome'
+    assert rv.data == b'Awesome'
+
+def test_implicit_head():
+    app = flask.Flask(__name__)
+
+    class Index(flask.views.MethodView):
+        def get(self):
+            return flask.Response('Blub', headers={
+                'X-Method': flask.request.method
+            })
+
+    app.add_url_rule('/', view_func=Index.as_view('index'))
+    c = app.test_client()
+    rv = c.get('/')
+    assert rv.data == b'Blub'
+    assert rv.headers['X-Method'] == 'GET'
+    rv = c.head('/')
+    assert rv.data == b''
+    assert rv.headers['X-Method'] == 'HEAD'
+
+def test_explicit_head():
+    app = flask.Flask(__name__)
+
+    class Index(flask.views.MethodView):
+        def get(self):
+            return 'GET'
+        def head(self):
+            return flask.Response('', headers={'X-Method': 'HEAD'})
+
+    app.add_url_rule('/', view_func=Index.as_view('index'))
+    c = app.test_client()
+    rv = c.get('/')
+    assert rv.data == b'GET'
+    rv = c.head('/')
+    assert rv.data == b''
+    assert rv.headers['X-Method'] == 'HEAD'
+
+def test_endpoint_override():
+    app = flask.Flask(__name__)
+    app.debug = True
+
+    class Index(flask.views.View):
+        methods = ['GET', 'POST']
+        def dispatch_request(self):
+            return flask.request.method
+
+    app.add_url_rule('/', view_func=Index.as_view('index'))
+
+    with pytest.raises(AssertionError):
+        app.add_url_rule('/', view_func=Index.as_view('index'))
+
+    # But these tests should still pass. We just log a warning.
+    common_test(app)
+
+def test_multiple_inheritance():
+    app = flask.Flask(__name__)
+
+    class GetView(flask.views.MethodView):
+        def get(self):
+            return 'GET'
+
+    class DeleteView(flask.views.MethodView):
+        def delete(self):
+            return 'DELETE'
+
+    class GetDeleteView(GetView, DeleteView):
+        pass
+
+    app.add_url_rule('/', view_func=GetDeleteView.as_view('index'))
+
+    c = app.test_client()
+    assert c.get('/').data == b'GET'
+    assert c.delete('/').data == b'DELETE'
+    assert sorted(GetDeleteView.methods) == ['DELETE', 'GET']
+
+def test_remove_method_from_parent():
+    app = flask.Flask(__name__)
+
+    class GetView(flask.views.MethodView):
+        def get(self):
+            return 'GET'
+
+    class OtherView(flask.views.MethodView):
+        def post(self):
+            return 'POST'
+
+    class View(GetView, OtherView):
+        methods = ['GET']
+
+    app.add_url_rule('/', view_func=View.as_view('index'))
+
+    c = app.test_client()
+    assert c.get('/').data == b'GET'
+    assert c.post('/').status_code == 405
+    assert sorted(View.methods) == ['GET']