如何在Python Flask应用的pytest测试中跳过/拦截before_request调用

有三种常用方法可以实现跳过before_request钩子的测试需求：

方法1：测试时临时移除before_request钩子 #

Flask的before_request钩子存储在app.before_request_funcs字典中，你可以在测试fixture或者单个测试用例中临时移除对应的钩子函数，测试完成后再恢复，避免影响其他测试。

import pytest
from your_app_module import app, before_request

@pytest.fixture
def client_without_before_request():
    # 移除指定的before_request钩子
    app.before_request_funcs[None].remove(before_request)
    tester = app.test_client()
    yield tester
    # 测试完成后加回去，避免影响其他测试
    app.before_request_funcs[None].append(before_request)

def test_index(client_without_before_request):
    response = client_without_before_request.get('/')
    assert response.status_code == 200
    assert response.data == b"Hello World!"

方法2：直接调用视图函数，不走请求链路 #

如果只需要测试视图函数本身的逻辑，完全可以跳过Flask的请求处理流程，直接调用视图函数获取返回值，这种方法执行速度更快，也不会触发任何请求钩子。

from your_app_module import index

def test_index_logic():
    # 直接调用视图函数，不会触发任何钩子
    result = index()
    assert result == "Hello World!"

方法3：给before_request添加测试环境跳过逻辑 #

可以在钩子函数内部加判断，当应用处于测试环境时直接跳过钩子执行：
首先修改Flask应用的钩子：

@app.before_request
def before_request():
    # 测试环境下跳过钩子执行
    if app.config.get("TESTING"):
        return
    # 原有逻辑
    if True:
        return True
    else:
        return False

然后测试fixture里开启TESTING配置即可：

@pytest.fixture
def client():
    app.config["TESTING"] = True
    tester = app.test_client()
    yield tester

内容的提问来源于stack exchange，提问作者Lavanya