mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
5f63b3a7bb
* chore: drop deprecated tables and remove unused code * isort imports * migration: check for table existence before drop
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
from test.fixtures import *
|
|
|
|
import pytest
|
|
|
|
from auth.auth_context import get_authenticated_context
|
|
from auth.validateresult import AuthKind, ValidateResult
|
|
from data import model
|
|
from data.database import AppSpecificAuthToken
|
|
|
|
|
|
def get_user():
|
|
return model.user.get_user("devtable")
|
|
|
|
|
|
def get_app_specific_token():
|
|
return AppSpecificAuthToken.get()
|
|
|
|
|
|
def get_robot():
|
|
robot, _ = model.user.create_robot("somebot", get_user())
|
|
return robot
|
|
|
|
|
|
def get_token():
|
|
return model.token.create_delegate_token("devtable", "simple", "sometoken")
|
|
|
|
|
|
def get_oauthtoken():
|
|
user = model.user.get_user("devtable")
|
|
return list(model.oauth.list_access_tokens_for_user(user))[0]
|
|
|
|
|
|
def get_signeddata():
|
|
return {"grants": {"a": "b"}, "user_context": {"c": "d"}}
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"get_entity,entity_kind",
|
|
[
|
|
(get_user, "user"),
|
|
(get_robot, "robot"),
|
|
(get_token, "token"),
|
|
(get_oauthtoken, "oauthtoken"),
|
|
(get_signeddata, "signed_data"),
|
|
(get_app_specific_token, "appspecifictoken"),
|
|
],
|
|
)
|
|
def test_apply_context(get_entity, entity_kind, app):
|
|
assert get_authenticated_context() is None
|
|
|
|
entity = get_entity()
|
|
args = {}
|
|
args[entity_kind] = entity
|
|
|
|
result = ValidateResult(AuthKind.basic, **args)
|
|
result.apply_to_context()
|
|
|
|
expected_user = entity if entity_kind == "user" or entity_kind == "robot" else None
|
|
if entity_kind == "oauthtoken":
|
|
expected_user = entity.authorized_user
|
|
|
|
if entity_kind == "appspecifictoken":
|
|
expected_user = entity.user
|
|
|
|
expected_token = entity if entity_kind == "token" else None
|
|
expected_oauth = entity if entity_kind == "oauthtoken" else None
|
|
expected_appspecifictoken = entity if entity_kind == "appspecifictoken" else None
|
|
expected_grant = entity if entity_kind == "signed_data" else None
|
|
|
|
assert get_authenticated_context().authed_user == expected_user
|
|
assert get_authenticated_context().token == expected_token
|
|
assert get_authenticated_context().oauthtoken == expected_oauth
|
|
assert get_authenticated_context().appspecifictoken == expected_appspecifictoken
|
|
assert get_authenticated_context().signed_data == expected_grant
|