32 lines
893 B
Python
32 lines
893 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"
|
|
} |