27 lines
840 B
Python
27 lines
840 B
Python
from fastapi import HTTPException, status
|
|
|
|
from app.models.auth import CognitoUser
|
|
|
|
MOCK_USERS = {"testuser": {"username": "testuser", "roles": ["admin"]}}
|
|
|
|
|
|
def mock_get_user_from_token(token: str) -> CognitoUser:
|
|
"""
|
|
Mock version of get_user_from_token for local testing
|
|
Accepts 'testuser' as a valid token and returns admin user
|
|
"""
|
|
if token == "testuser":
|
|
return CognitoUser(**MOCK_USERS["testuser"])
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="Invalid mock token - use 'testuser'",
|
|
)
|
|
|
|
|
|
def mock_initiate_auth(username: str, password: str) -> dict:
|
|
"""
|
|
Mock version of initiate_auth for local testing
|
|
Accepts any username/password and returns a mock token
|
|
"""
|
|
return {"AccessToken": "testuser", "ExpiresIn": 3600, "TokenType": "Bearer"}
|