forked from Github-Mirrors/canaille
101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
from urllib.parse import parse_qs
|
|
from urllib.parse import urlsplit
|
|
|
|
from canaille.oidc.models import AuthorizationCode
|
|
from canaille.oidc.models import Token
|
|
|
|
from . import client_credentials
|
|
|
|
|
|
def test_token_introspection(testclient, user, client, token):
|
|
res = testclient.post(
|
|
"/oauth/introspect",
|
|
params=dict(
|
|
token=token.oauthAccessToken,
|
|
),
|
|
headers={"Authorization": f"Basic {client_credentials(client)}"},
|
|
status=200,
|
|
)
|
|
assert {
|
|
"active": True,
|
|
"client_id": client.oauthClientID,
|
|
"token_type": token.oauthTokenType,
|
|
"username": user.name,
|
|
"scope": token.get_scope(),
|
|
"sub": user.uid[0],
|
|
"aud": [client.oauthClientID],
|
|
"iss": "https://mydomain.tld",
|
|
"exp": token.get_expires_at(),
|
|
"iat": token.get_issued_at(),
|
|
} == res.json
|
|
|
|
|
|
def test_token_invalid(testclient, client):
|
|
res = testclient.post(
|
|
"/oauth/introspect",
|
|
params=dict(token="invalid"),
|
|
headers={"Authorization": f"Basic {client_credentials(client)}"},
|
|
status=200,
|
|
)
|
|
assert {"active": False} == res.json
|
|
|
|
|
|
def test_full_flow(
|
|
testclient, slapd_connection, logged_user, client, user, other_client
|
|
):
|
|
res = testclient.get(
|
|
"/oauth/authorize",
|
|
params=dict(
|
|
response_type="code",
|
|
client_id=client.oauthClientID,
|
|
scope="profile",
|
|
nonce="somenonce",
|
|
),
|
|
status=200,
|
|
)
|
|
|
|
res = res.form.submit(name="answer", value="accept", status=302)
|
|
|
|
assert res.location.startswith(client.oauthRedirectURIs[0])
|
|
params = parse_qs(urlsplit(res.location).query)
|
|
code = params["code"][0]
|
|
authcode = AuthorizationCode.get(code, conn=slapd_connection)
|
|
assert authcode is not None
|
|
|
|
res = testclient.post(
|
|
"/oauth/token",
|
|
params=dict(
|
|
grant_type="authorization_code",
|
|
code=code,
|
|
scope="profile",
|
|
redirect_uri=client.oauthRedirectURIs[0],
|
|
),
|
|
headers={"Authorization": f"Basic {client_credentials(client)}"},
|
|
status=200,
|
|
)
|
|
access_token = res.json["access_token"]
|
|
|
|
token = Token.get(access_token, conn=slapd_connection)
|
|
assert token.oauthClient == client.dn
|
|
assert token.oauthSubject == logged_user.dn
|
|
|
|
res = testclient.post(
|
|
"/oauth/introspect",
|
|
params=dict(
|
|
token=token.oauthAccessToken,
|
|
),
|
|
headers={"Authorization": f"Basic {client_credentials(client)}"},
|
|
status=200,
|
|
)
|
|
assert {
|
|
"aud": [client.oauthClientID, other_client.oauthClientID],
|
|
"active": True,
|
|
"client_id": client.oauthClientID,
|
|
"token_type": token.oauthTokenType,
|
|
"username": user.name,
|
|
"scope": token.get_scope(),
|
|
"sub": user.uid[0],
|
|
"iss": "https://mydomain.tld",
|
|
"exp": token.get_expires_at(),
|
|
"iat": token.get_issued_at(),
|
|
} == res.json
|