python - 为 Flask 测试中的所有请求设置 HTTP header

Question

我正在使用 Flask 并且有需要授权的端点(偶尔还有其他特定于应用程序的 header )。在我的测试中，使用 test_client 函数创建一个客户端，然后执行各种获取、放置、删除调用。所有这些调用都需要授权，并添加其他 header 。我如何设置测试客户端以将此类 header 放在所有请求上？

Answer 1

Client类采用与 EnvironBuilder 相同的参数类，其中有 headers关键字参数。

因此您可以简单地使用 client.get( '/', headers={ ... } ) 发送您的身份验证。

现在，如果您想从客户端提供一组默认 header ，则需要提供您自己的 open 实现它提供修改后的环境构建器(类似于 make_test_environ_builder )并设置 app.test_client_class指向您的新类(class)。