python - 单元测试 flask flask.g

Question

当我进行单元测试时，我希望能够设置和访问 flask.g。

flask.g['test'] = {'test': '123'}
test_dict = flask.g['test']

产生这个错误:

Error
Traceback (most recent call last):
  File "/usr/local/lib/python2.7/dist-packages/werkzeug/local.py", line 346, in __setitem__
    self._get_current_object()[key] = value
TypeError: '_AppCtxGlobals' object does not support item assignment

但是，当我在生产环境中运行它时，一切正常。如果我使用 setattr 和 getattr，单元测试有效，但在生产中中断，我收到以下错误

Error
Traceback (most recent call last):
File "/usr/local/lib/python2.7/dist-packages/flask_restful/__init__.py", line 268, in error_router
    return self.handle_error(e)
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1475, in full_dispatch_request
    rv = self.dispatch_request()
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1461, in dispatch_request
    return self.view_functions[rule.endpoint](**req.view_args)
  File "/usr/local/lib/python2.7/dist-packages/flask_restful/__init__.py", line 477, in wrapper
    resp = resource(*args, **kwargs)
  File "/usr/local/lib/python2.7/dist-packages/flask/views.py", line 84, in view
    return self.dispatch_request(*args, **kwargs)
  File "/usr/local/lib/python2.7/dist-packages/flask_restful/__init__.py", line 587, in dispatch_request
resp = meth(*args, **kwargs)
    args = getattr(flask.g, 'test')
AttributeError: 'dict' object has no attribute 'test'

我在 app_context() 中运行单元测试。目标是拥有实际代表生产的单元测试。我怎样才能做到这一点？

Answer 1

现在这两种用法是等价的

user = getattr(flask.g, 'user', None)
user = flask.g.get('user', None)

使用示例

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

更多信息 official documentation或 testing documentation