为烧瓶应用编写单元测试用例

时间:2016-01-19 19:42:51

标签: python flask

我正在使用flask框架制作Web应用程序。要使用flask显示网页,我使用 render_template()函数。 例如:

@app.route('/restaurants/<int:restaurant_id>/')
def restaurantMenu(restaurant_id):
    try:
        # pdb.set_trace()
        session = get_session()
        restaurant = session.query(Restaurant).filter_by(id=restaurant_id).one()
        if restaurant is not None:
            menu_items = session.query(MenuItem).filter(
                  MenuItem.restaurant_id == restaurant.id
            )
        session.close()
        if menu_items is not None:
            return render_template('menu.html', restaurant=restaurant, items=menu_items)
    except NoResultFound:
        return render_template('error.html', error_message='No such restaurant id.')

@app.route('/')
def welcomePage():
    return render_template('index.html')

如何为这些函数编写测试用例?我是测试的新手,所以我想为我的代码编写测试用例。

1 个答案:

答案 0 :(得分:0)

您可以在测试中创建一个测试应用并简单地调用它,如下所示:

app = create_app().test_client()
result = app.get('/')
assert('something' in result.data)

这给了你一般的想法 - 它非常容易使用。更多信息可在Flask测试文档中找到:http://flask.pocoo.org/docs/0.10/testing/