Python app.test_client函数代码示例

    def test_templates_and_static(self):
        from moduleapp import app
        c = app.test_client()

        rv = c.get('/')
        assert rv.data == 'Hello from the Frontend'
        rv = c.get('/admin/')
        assert rv.data == 'Hello from the Admin'
        rv = c.get('/admin/index2')
        assert rv.data == 'Hello from the Admin'
        rv = c.get('/admin/static/test.txt')
        assert rv.data.strip() == 'Admin File'
        rv = c.get('/admin/static/css/test.css')
        assert rv.data.strip() == '/* nested file */'

        with app.test_request_context():
            assert flask.url_for('admin.static', filename='test.txt') \
                == '/admin/static/test.txt'

        with app.test_request_context():
            try:
                flask.render_template('missing.html')
            except TemplateNotFound, e:
                assert e.name == 'missing.html'
            else:

    def test_processor_exceptions(self):
        app = flask.Flask(__name__)

        @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 == "Hello Server Error"

    def test_debug_log(self):
        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

        c = app.test_client()

        with catch_stderr() as err:
            c.get("/")
            out = err.getvalue()
            assert "WARNING in flask_tests [" in out
            assert "flask_tests.py" in out
            assert "the standard library is dead" in out
            assert "this is a debug statement" in out

        with catch_stderr() as err:
            try:
                c.get("/exc")
            except ZeroDivisionError:
                pass
            else:
                assert False, "debug log ate the exception"

    def test_templates_and_static(self):
        from moduleapp import app

        c = app.test_client()

        rv = c.get("/")
        assert rv.data == "Hello from the Frontend"
        rv = c.get("/admin/")
        assert rv.data == "Hello from the Admin"
        rv = c.get("/admin/index2")
        assert rv.data == "Hello from the Admin"
        rv = c.get("/admin/static/test.txt")
        assert rv.data.strip() == "Admin File"
        rv = c.get("/admin/static/css/test.css")
        assert rv.data.strip() == "/* nested file */"

        with app.test_request_context():
            assert flask.url_for("admin.static", filename="test.txt") == "/admin/static/test.txt"

        with app.test_request_context():
            try:
                flask.render_template("missing.html")
            except TemplateNotFound, e:
                assert e.name == "missing.html"
            else:

    def test_debug_log(self):
        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
        c = app.test_client()

        with catch_stderr() as err:
            c.get('/')
            out = err.getvalue()
            assert 'WARNING in flask_tests [' in out
            assert 'flask_tests.py' in out
            assert 'the standard library is dead' in out
            assert 'this is a debug statement' in out

        with catch_stderr() as err:
            try:
                c.get('/exc')
            except ZeroDivisionError:
                pass
            else:
                assert False, 'debug log ate the exception'

 def test_module_static_path_subdomain(self):
     app = flask.Flask(__name__)
     app.config['SERVER_NAME'] = 'example.com'
     from subdomaintestmodule import mod
     app.register_module(mod)
     c = app.test_client()
     rv = c.get('/static/hello.txt', 'http://foo.example.com/')
     assert rv.data.strip() == 'Hello Subdomain'

    def test_module_static_path_subdomain(self):
        app = flask.Flask(__name__)
        app.config["SERVER_NAME"] = "example.com"
        from subdomaintestmodule import mod

        app.register_module(mod)
        c = app.test_client()
        rv = c.get("/static/hello.txt", "http://foo.example.com/")
        assert rv.data.strip() == "Hello Subdomain"

    def test_subdomain_matching(self):
        app = flask.Flask(__name__)
        app.config['SERVER_NAME'] = 'localhost'
        @app.route('/', subdomain='<user>')
        def index(user):
            return 'index for %s' % user

        c = app.test_client()
        rv = c.get('/', 'http://mitsuhiko.localhost/')
        assert rv.data == 'index for mitsuhiko'

    def test_template_rendered(self):
        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_static_path_without_url_prefix(self):
     app = flask.Flask(__name__)
     app.config['SERVER_NAME'] = 'example.com'
     from testmodule import mod
     app.register_module(mod)
     c = app.test_client()
     f = 'hello.txt'
     rv = c.get('/static/' + f, 'http://example.com/')
     assert rv.data.strip() == 'Hello Maindomain'
     with app.test_request_context(base_url='http://example.com'):
         assert flask.url_for('static', filename=f) == '/static/' + f
         assert flask.url_for('static', filename=f, _external=True) \
             == 'http://example.com/static/' + f

    def test_basic_support(self):
        app = flask.Flask(__name__)
        app.config['SERVER_NAME'] = 'localhost'
        @app.route('/')
        def normal_index():
            return 'normal index'
        @app.route('/', subdomain='test')
        def test_index():
            return 'test index'

        c = app.test_client()
        rv = c.get('/', 'http://localhost/')
        assert rv.data == 'normal index'

        rv = c.get('/', 'http://test.localhost/')
        assert rv.data == 'test index'

    def test_request_exception_signal(self):
        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_exception_logging(self):
        out = StringIO()
        app = flask.Flask(__name__)
        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 "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_exception_logging(self):
        out = StringIO()
        app = flask.Flask(__name__)
        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 '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_module_subdomain_support(self):
        app = flask.Flask(__name__)
        mod = flask.Module(__name__, 'test', subdomain='testing')
        app.config['SERVER_NAME'] = 'localhost'

        @mod.route('/test')
        def test():
            return 'Test'

        @mod.route('/outside', subdomain='xtesting')
        def bar():
            return 'Outside'

        app.register_module(mod)

        c = app.test_client()
        rv = c.get('/test', 'http://testing.localhost/')
        assert rv.data == 'Test'
        rv = c.get('/outside', 'http://xtesting.localhost/')
        assert rv.data == 'Outside'