如何在FastAPI测试中覆盖Auth0顶层认证依赖？

方案1：重构依赖为独立函数（推荐） #

把原有的实例方法包装成独立的依赖函数，让依赖覆盖更直观：

原代码修改 #

# main.py
from fastapi import FastAPI, Security
from auth_module import VerifyToken

# 初始化Auth0验证实例
auth = VerifyToken()

# 包装成独立的依赖函数
def get_current_user():
    return auth.verify()

# 创建FastAPI应用时使用该函数作为顶层依赖
app = FastAPI(
    # 其他配置项
    dependencies=[Security(get_current_user)]
)

测试代码 #

# test_main.py
from fastapi.testclient import TestClient
from main import app, get_current_user

def get_test_user():
    return {"sub": "testuser"}

# 覆盖依赖
app.dependency_overrides[get_current_user] = get_test_user

client = TestClient(app)

# 测试接口时将自动使用测试用户
response = client.get("/your-endpoint")
assert response.status_code == 200

方案2：复用全局实例的绑定方法 #

如果不想重构代码，确保测试时覆盖的是应用中使用的同一个auth实例的verify方法：

原代码（需保证auth是全局实例） #

# main.py
auth = VerifyToken()

app = FastAPI(
    dependencies=[Security(auth.verify)]
)

测试代码 #

# test_main.py
from fastapi.testclient import TestClient
from main import app, auth  # 导入原代码中的同一个auth实例

def get_test_user(*args, **kwargs):
    return {"sub": "testuser"}

# 覆盖该实例的verify方法对应的依赖
app.dependency_overrides[auth.verify] = get_test_user

client = TestClient(app)

方案3：直接替换顶层依赖列表（快速临时方案） #

如果只是简单测试，可直接替换应用的顶层依赖列表：

# test_main.py
from fastapi.testclient import TestClient, Security
from main import app

def get_test_user():
    return {"sub": "testuser"}

# 替换整个顶层依赖列表（若有其他依赖需手动保留）
app.dependencies = [Security(get_test_user)]

client = TestClient(app)

原方法无效的原因 #

• Try1：VerifyToken.verify是类的未绑定方法，但应用中实际使用的是实例绑定的verify方法，二者不是同一个对象，覆盖无效。
• Try2：创建了新的auth实例，覆盖的是新实例的方法，和应用中使用的全局实例不匹配。
• Try3：Security(auth.verify)返回的是Security包装对象，FastAPI的依赖覆盖基于底层可调用函数，而非包装器，因此覆盖无效。

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