小编Eyo*_*nyo的帖子

我有一个在我的所有 API 中作为依赖项调用的函数。

from fastapi import HTTPException, Request, status


async def get_user_id_or_401(request: Request) -> str:
    user_id: str = request.headers.get("x-cognito-user-id")

    if not user_id:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED, detail="Unauthorized"
        )
    return user_id

如何对该功能进行单元测试？我最初的举动是修补request.headers.get，但我不知道该怎么做。

所以，我的问题是，如何修补request.headers.get. 但是，如果我有更好的方法可以对此进行测试，请告诉我。