笨鸟编程-零基础入门Pyhton教程 › 首页 ›Flask中文手册 › 查看内容

测试 Flask 应用

其他测试技巧

除了如上文演示的使用测试客户端完成测试的方法，也有一个 test_request_context() 方法可以配合 with 语句用于激活一个临时的请求上下文。通过它，您可以访问 request 、g 和 session 类的对象，就像在视图中一样。这里有一个完整的例子示范了这种用法:

app = flask.Flask(__name__)

with app.test_request_context('/?name=Peter'):
    assert flask.request.path == '/'
    assert flask.request.args['name'] == 'Peter'

所有其他的和上下文绑定的对象都可以使用同样的方法访问。

如果您希望测试应用在不同配置的情况下的表现，这里似乎没有一个很好的方法，考虑使用应用的工厂函数(参考应用程序的工厂函数)

注意，尽管你在使用一个测试用的请求环境，函数 before_request() 以及 after_request() 都不会自动运行。然而，teardown_request() 函数在测试请求的上下文离开 with 块的时候会执行。如果您希望 before_request() 函数仍然执行。您需要手动调用 preprocess_request() 方法:

app = flask.Flask(__name__)

with app.test_request_context('/?name=Peter'):
    app.preprocess_request()
    ...

这对于打开数据库连接或者其他类似的操作来说，很可能是必须的，这视您应用的设计方式而定。

如果您希望调用 after_request() 函数，您需要使用 process_response() 方法。这个方法需要您传入一个 response 对象:

app = flask.Flask(__name__)

with app.test_request_context('/?name=Peter'):
    resp = Response('...')
    resp = app.process_response(resp)
    ...

这通常不是很有效，因为这时您可以直接转向使用测试客户端。

伪造资源和上下文

0.10 新版功能.

在应用上下文或 flask.g 对象上存储用户认证信息和数据库连接非常常见。一般的模式是在第一次使用对象时，把对象放在应用上下文或 flask.g 上面，而在请求销毁时移除对象。试想一下例如下面的获取当前用户的代码:

def get_user():
    user = getattr(g, 'user', None)
    if user is None:
        user = fetch_current_user_from_database()
        g.user = user
    return user

对于测试，这样易于从外部覆盖这个用户，而不用修改代码。连接 flask.appcontext_pushed 信号可以很容易地完成这个任务:

from contextlib import contextmanager
from flask import appcontext_pushed

@contextmanager
def user_set(app, user):
    def handler(sender, **kwargs):
        g.user = user
    with appcontext_pushed.connected_to(handler, app):
        yield

并且之后使用它:

from flask import json, jsonify

@app.route('/users/me')
def users_me():
    return jsonify(username=g.user.username)

with user_set(app, my_user):
    with app.test_client() as c:
        resp = c.get('/users/me')
        data = json.loads(resp.data)
        self.assert_equal(data['username'], my_user.username) 

1 234 / 4 页下一页

		自动登录	找回密码
密码			立即注册