Flask JWT单元测试：如何在请求上下文外获取当前用户身份

我之前也碰到过一模一样的情况——直接测试服务层的create函数时，因为跳过了Flask的请求处理链，JWT的上下文根本没被初始化，所以get_current_user()自然拿不到用户。下面是我亲测有效的解决办法：

核心思路 #

接口测试时，请求会经过@jwt_required装饰器，它会解析请求头里的token，再通过你的user_loader回调加载用户并绑定到上下文。但单元测试直接调用create函数时，这个流程完全没走，所以我们需要手动在测试中设置JWT上下文的当前用户，而不是依赖请求头。

具体实现步骤 #

1. 确认JWT用户加载回调已配置 #

首先检查你的Flask应用是否正确配置了user_loader，这是JWT根据身份信息找到用户的关键（你接口测试正常的话应该已经配置了，但还是确认一下）：

from flask_jwt_extended import JWTManager
from app.models import User

jwt = JWTManager(app)

@jwt.user_loader_callback_loader
def load_user(identity):
    # 这里的identity就是你创建token时传入的参数（你代码里是用户email）
    return User.query.filter_by(email=identity).first()

2. 在单元测试中手动绑定当前用户 #

在测试函数里，我们需要先推入Flask应用上下文，然后用flask_jwt_extended.set_current_user()直接把测试用户绑定到JWT上下文里。

给你一个完整的测试示例：

import pytest
from flask_jwt_extended import set_current_user
from app import app, db, listing_svc
from app.models import User

def test_create_listing_with_authenticated_user():
    # 1. 创建测试用户并保存到数据库
    test_user = User(email="test_unit@example.com", password="hashed_test_pass")
    db.session.add(test_user)
    db.session.commit()

    # 2. 推入Flask应用上下文（必须，JWT上下文依赖Flask的app上下文）
    with app.app_context():
        # 3. 手动设置当前用户到JWT上下文
        set_current_user(test_user)

        # 4. 准备测试用的payload（不需要传creator_user_id，set_creator_id会自动填充）
        test_payload = {
            "title": "Test Listing",
            # 其他必填字段...
        }

        # 5. 调用要测试的create函数
        created_listing = listing_svc.create(test_payload)

        # 6. 断言结果符合预期
        assert created_listing.creator_user_id == test_user.id
        assert created_listing.title == "Test Listing"

    # 清理测试数据（根据你的测试数据库策略调整）
    db.session.delete(test_user)
    db.session.commit()

3. 用Pytest Fixture优化测试代码 #

如果有多个测试需要用到用户和上下文，可以把这些逻辑封装成fixture，让代码更简洁复用：

@pytest.fixture
def test_user():
    # 创建测试用户
    user = User(email="fixture_test@example.com", password="hashed_pass")
    db.session.add(user)
    db.session.commit()
    yield user
    # 测试完成后自动清理
    db.session.delete(user)
    db.session.commit()

@pytest.fixture
def app_context(app):
    # 提供Flask应用上下文
    with app.app_context():
        yield

def test_create_listing_fixture_version(test_user, app_context):
    set_current_user(test_user)
    test_payload = {"title": "Fixture Test Listing", ...}
    created_listing = listing_svc.create(test_payload)
    assert created_listing.creator_user_id == test_user.id

为什么之前的方法没用？ #

你尝试用test_request_context设置请求头是针对接口测试的——当你用客户端发送请求时，请求头会被JWT中间件解析。但直接调用create函数时，根本没有触发这个中间件，所以请求头里的token完全不会被处理，get_current_user()自然返回None。

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