我有一个测试装置可以在我的 Flask 应用程序中创建一个测试客户端:
@pytest.fixture(scope='session')
def test_client():
""" Create the application and the test client.
"""
print('----------Setup test client----------')
app = create_app(config_class=config.TestConfig)
testing_client = app.test_client()
ctx = app.test_request_context()
ctx.push()
yield testing_client # this is where the testing happens
print('-------Teardown test client--------')
ctx.pop()
我想将“用户代理”的 header 添加到此测试请求上下文中,因为在我的应用程序中我检查了用户的浏览器。我正在通过以下方式检索我的应用程序中的浏览器名称
user_agent = flask.request.user_agent.browser
如果我使用上面的测试客户端,这一行将返回 None 。虽然我可以在单个测试中的单个请求中成功设置“User-Agent” header ,例如:
user_agent_str = 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36'
response = test_client.get('/some_flask_route',
environ_base={'HTTP_USER_AGENT': user_agent_str},
follow_redirects=True)
这样做之后:
flask.request.user_agent.browser
按预期返回 'chrome'。然而,这是非常多余的,因为它需要我将 environ_base 代码行插入到我的许多测试中的每一个中。
预期行为
我希望在我的测试装置中创建测试请求上下文时可以设置 header ,例如:
user_agent_str = 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36'
@pytest.fixture(scope='session')
def test_client():
""" Create the application and the test client.
"""
print('----------Setup test client----------')
app = create_app(config_class=config.TestConfig)
testing_client = app.test_client()
ctx = app.test_request_context(environ_base={'HTTP_USER_AGENT': user_agent_str})
ctx.push()
yield testing_client # this is where the testing happens
print('-------Teardown test client--------')
ctx.pop()
这将为所有请求设置环境,从而无需在我的每个请求中都设置 environment_base。
实际行为
虽然将environ_base 添加到test_request_context() 中不会破坏fixture,但它不会设置“User-Agent” header 和
flask.request.user_agent.browser
返回无。
环境
我已经尝试了这里问题中建议的解决方案:Setting (mocking) request headers for Flask app unit test
但它似乎没有改变任何东西。我只是在没有部署的情况下在本地主机上运行 Flask 服务器,即基本上只是:
app = Flask(__name__)
app.run(host='127.0.0.1',port=5000,debug=True)
最佳答案
测试客户端有一个 environ_base
在构建每个请求时设置默认环境的属性。
c = app.test_client()
c.environ_base["HTTP_USER_AGENT"] = "Firefox/71.0"
要更好地控制传递给每个请求的内容,
FlaskClient
可以被子类化以覆盖 open
:class CustomClient(FlaskClient):
def open(self, *args, **kwargs):
headers = kwargs.setdefault("headers", {})
headers.setdefault("User-Agent", "Firefox/71.0")
return super().open(*args, **kwargs)
app.test_client_class = CustomClient
c = app.test_client()
assert c.get("/browser") == "firefox"
同样,
Flask
可以被子类化以覆盖 test_request_context
:class CustomFlask(Flask):
def test_request_context(self, *args, **kwargs):
headers = kwargs.setdefault("headers", {})
headers.setdefault("User-Agent", "Firefox/71.0")
return super().test_request_context(*args, **kwargs)
app = CustomFlask()
with app.test_request_client():
assert browser() == "firefox"
不要为所有测试全局推送测试请求上下文。应该为每个请求创建一个新的上下文,并在使用
client
发出测试请求时自动处理。 .在 session 装置中推送一个不会产生您想要的效果。在大多数情况下,您应该使用 test_client
在test_request_context
.重新考虑您首先依赖用户代理的原因。以用户代理为基础的应用程序行为通常是糟糕的设计,因为浏览器会误报它们并且功能会随着时间的推移而改变。
关于python - 如何将标题添加到flask test_request_context?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59396114/