47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
import auth as app
|
|
from auth import _parse_bearer_token
|
|
|
|
from .conftest import LambdaContext
|
|
|
|
|
|
def test_bearer_jwt(lambda_context: LambdaContext, dynamodb_seeds):
|
|
# You should mock the Cognito user to pass the test
|
|
app.get_user = lambda *args, **kwargs: {
|
|
'sub': '58efed8d-d276-41a8-8502-4ab8b5a6415e',
|
|
'name': 'pytest',
|
|
'custom:user_id': '5OxmMjL-ujoR5IMGegQz',
|
|
}
|
|
event = {
|
|
'headers': {
|
|
'authorization': 'Bearer 3c51cdfd-d23e-47f9-8d7c-e3e31a432921',
|
|
},
|
|
}
|
|
|
|
assert app.lambda_handler(event, lambda_context) == {
|
|
'isAuthorized': True,
|
|
'context': {
|
|
'user': {
|
|
'sub': '58efed8d-d276-41a8-8502-4ab8b5a6415e',
|
|
'name': 'pytest',
|
|
'custom:user_id': '5OxmMjL-ujoR5IMGegQz',
|
|
}
|
|
},
|
|
}
|
|
|
|
|
|
def test_parse_bearer_token_api_key():
|
|
bearer = _parse_bearer_token(
|
|
'Bearer pptx-35433970-6857-4062-bb43-f71683b2f68e',
|
|
apikey_prefix='pptx',
|
|
)
|
|
|
|
assert bearer.token == '35433970-6857-4062-bb43-f71683b2f68e' # type: ignore
|
|
assert bearer.auth_type == 'API_KEY' # type: ignore
|
|
|
|
|
|
def test_parse_bearer_token_user_token():
|
|
bearer = _parse_bearer_token('Bearer d977f5a2-0302-4dd2-87c7-57414264d27a')
|
|
|
|
assert bearer.token == 'd977f5a2-0302-4dd2-87c7-57414264d27a' # type: ignore
|
|
assert bearer.auth_type == 'USER_TOKEN' # type: ignore
|