python - 用 DispatcherMiddleware 包裹的 Flask 应用不再有 test_client

Question

我们可以通过以下方式获取示例应用程序的test_client:

class MyTestCase(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        my_app.app.config['TESTING'] = True
        cls.client = my_app.app.test_client()

但是，如果我们用 DispatcherMiddleware 包装应用程序——我们会得到类似这样的错误 AttributeError: 'DispatcherMiddleware' 对象没有属性 'test_client'。

有没有办法测试 Flask 应用程序的组成？

我们希望能够做这样的事情:

cls.client = my_app.all_apps.test_client()

当 all_apps 是像这样的中间件时:

all_apps = DispatcherMiddleware(my_app, {
    '/backend': backend_app,
})

Answer 1

要将 WSGI 中间件添加到 Flask 应用程序，包装并替换应用程序的 wsgi_app 属性。您正在用对其他一些 WSGI 应用程序的引用替换对 Flask 应用程序的引用，这显然不会具有相同的属性。通过替换 wsgi_app，您保留了对 Flask 应用程序的引用，但更改了支持它的 WSGI 可调用对象。

app.wsgi_app = DispatcherMiddleware(app.wsgi_app, {
    '/backend': backend_app.wsgi_app,
})