138 lines
4.7 KiB
Python
138 lines
4.7 KiB
Python
import secrets
|
|
from base64 import b64encode
|
|
from datetime import UTC, datetime
|
|
from urllib.parse import parse_qs, urlparse
|
|
|
|
from argon2 import PasswordHasher
|
|
from httpx import AsyncClient
|
|
|
|
from fastapi_oidc_op.authn.password import PasswordService
|
|
from fastapi_oidc_op.models import PasswordCredential, User
|
|
|
|
|
|
def _register_test_client(client: AsyncClient) -> str:
|
|
app = client._transport.app # type: ignore[union-attr]
|
|
oidc_server = app.state.oidc_server
|
|
client_secret = "test-secret-0123456789abcdef"
|
|
oidc_server.context.cdb["test-rp"] = {
|
|
"client_id": "test-rp",
|
|
"client_secret": client_secret,
|
|
"redirect_uris": [("http://localhost:9000/callback", {})],
|
|
"response_types_supported": ["code"],
|
|
"token_endpoint_auth_method": "client_secret_basic",
|
|
"scope": ["openid", "profile", "email"],
|
|
"allowed_scopes": ["openid", "profile", "email"],
|
|
"client_salt": secrets.token_hex(8),
|
|
}
|
|
oidc_server.keyjar.add_symmetric("test-rp", client_secret)
|
|
return client_secret
|
|
|
|
|
|
async def _create_user_and_login(client: AsyncClient) -> str:
|
|
"""Create user, log in, return userid."""
|
|
app = client._transport.app # type: ignore[union-attr]
|
|
user_repo = app.state.user_repo
|
|
cred_repo = app.state.credential_repo
|
|
|
|
user = User(
|
|
userid="lusab-bansen",
|
|
username="alice",
|
|
email="alice@example.com",
|
|
email_verified=True,
|
|
created_at=datetime.now(UTC),
|
|
updated_at=datetime.now(UTC),
|
|
)
|
|
await user_repo.create(user)
|
|
|
|
svc = PasswordService(hasher=PasswordHasher(time_cost=1, memory_cost=8192))
|
|
await cred_repo.create_password(PasswordCredential(user_id=user.userid, password_hash=svc.hash("testpass")))
|
|
|
|
await client.post(
|
|
"/login/password",
|
|
data={"username": "alice", "password": "testpass"},
|
|
headers={"HX-Request": "true"},
|
|
)
|
|
return user.userid
|
|
|
|
|
|
async def _get_authorization_code(client: AsyncClient) -> str:
|
|
"""Run full auth flow and extract the authorization code."""
|
|
_register_test_client(client)
|
|
|
|
# Start authorization (unauthenticated — stores in session)
|
|
await client.get(
|
|
"/authorization",
|
|
params={
|
|
"response_type": "code",
|
|
"client_id": "test-rp",
|
|
"redirect_uri": "http://localhost:9000/callback",
|
|
"scope": "openid profile email",
|
|
"state": "test-state",
|
|
"nonce": "test-nonce",
|
|
},
|
|
follow_redirects=False,
|
|
)
|
|
|
|
# Create user and log in
|
|
await _create_user_and_login(client)
|
|
|
|
# Complete authorization (now authenticated, session has oidc_auth_request)
|
|
complete_res = await client.get("/authorization/complete", follow_redirects=False)
|
|
assert complete_res.status_code in (302, 303), (
|
|
f"Expected redirect, got {complete_res.status_code}: {complete_res.text}"
|
|
)
|
|
|
|
location = complete_res.headers["location"]
|
|
parsed = urlparse(location)
|
|
params = parse_qs(parsed.query)
|
|
assert "code" in params, f"No code in redirect: {location}"
|
|
return params["code"][0]
|
|
|
|
|
|
async def test_authorization_complete_redirects_with_code(client: AsyncClient) -> None:
|
|
code = await _get_authorization_code(client)
|
|
assert len(code) > 0
|
|
|
|
|
|
async def test_token_endpoint_exchanges_code(client: AsyncClient) -> None:
|
|
code = await _get_authorization_code(client)
|
|
client_secret = "test-secret-0123456789abcdef"
|
|
|
|
auth_header = b64encode(f"test-rp:{client_secret}".encode()).decode()
|
|
token_res = await client.post(
|
|
"/token",
|
|
data={
|
|
"grant_type": "authorization_code",
|
|
"code": code,
|
|
"redirect_uri": "http://localhost:9000/callback",
|
|
},
|
|
headers={
|
|
"Authorization": f"Basic {auth_header}",
|
|
"Content-Type": "application/x-www-form-urlencoded",
|
|
},
|
|
)
|
|
assert token_res.status_code == 200, f"Token endpoint failed: {token_res.text}"
|
|
data = token_res.json()
|
|
assert "access_token" in data
|
|
assert "id_token" in data
|
|
assert data["token_type"].lower() == "bearer"
|
|
|
|
|
|
async def test_token_endpoint_invalid_code_returns_error(client: AsyncClient) -> None:
|
|
_register_test_client(client)
|
|
client_secret = "test-secret-0123456789abcdef"
|
|
|
|
auth_header = b64encode(f"test-rp:{client_secret}".encode()).decode()
|
|
token_res = await client.post(
|
|
"/token",
|
|
data={
|
|
"grant_type": "authorization_code",
|
|
"code": "invalid-code",
|
|
"redirect_uri": "http://localhost:9000/callback",
|
|
},
|
|
headers={
|
|
"Authorization": f"Basic {auth_header}",
|
|
"Content-Type": "application/x-www-form-urlencoded",
|
|
},
|
|
)
|
|
assert token_res.status_code == 400 or "error" in token_res.json()
|