如何使用测试客户端运行烧瓶测试时访问g变量？

Question

当我尝试测试一个无法访问 g 变量的烧录应用程序时出现问题。

将要测试的api看起来像这样：

@app.route('/user'):
def get_user_info():
    user = query_object.get(g.user_id)   # here the exception raises
    # other code

，测试用例如下：

def test_get_user_info(self):
    resp = self.test_client.get('/user')
    self.assertTrue(resp.status_code == 200)

当我进行测试时，它会引发：

NameError: global name 'g' is not defined

在get_user_info() api。

内

我的问题是：

如何处理此问题，并且可以访问get_user_info()中的 g 变量？

Answer 1

你的进口是什么？

您应该尝试添加

from flask import g