Unittest Flask flask.g

时间:2016-06-06 09:53:45

标签: python python-2.7 unit-testing flask python-unittest

当我进行单元测试时,我希望能够设置和访问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

但是,当我在生产中运行时,一切正常。如果我使用setattrgetattr,则单元测试有效,但生产中断, 我收到以下错误

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()中运行unittest。目标是进行实际代表生产的单元测试。我怎么能得到这个?

1 个答案:

答案 0 :(得分:4)

这两个用法现在是等价的

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 documentationtesting documentation