72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
from datetime import UTC, datetime, timedelta
|
|
|
|
from httpx import AsyncClient
|
|
|
|
from porchlight.models import MagicLink
|
|
|
|
|
|
async def test_register_invalid_token_returns_error_page(client: AsyncClient) -> None:
|
|
res = await client.get("/register/nope", follow_redirects=False)
|
|
assert res.status_code == 400
|
|
assert "Invalid or expired" in res.text
|
|
|
|
|
|
async def test_register_expired_token_returns_error_page(client: AsyncClient) -> None:
|
|
app = client._transport.app # type: ignore[union-attr]
|
|
repo = app.state.magic_link_repo
|
|
await repo.create(
|
|
MagicLink(
|
|
token="expired",
|
|
username="newuser",
|
|
expires_at=datetime.now(UTC) - timedelta(hours=1),
|
|
)
|
|
)
|
|
|
|
res = await client.get("/register/expired", follow_redirects=False)
|
|
assert res.status_code == 400
|
|
assert "Invalid or expired" in res.text
|
|
|
|
|
|
async def test_register_valid_token_creates_user_and_redirects(client: AsyncClient) -> None:
|
|
app = client._transport.app # type: ignore[union-attr]
|
|
magic_link_repo = app.state.magic_link_repo
|
|
user_repo = app.state.user_repo
|
|
|
|
await magic_link_repo.create(
|
|
MagicLink(
|
|
token="t1",
|
|
username="newuser",
|
|
expires_at=datetime.now(UTC) + timedelta(hours=1),
|
|
)
|
|
)
|
|
|
|
res = await client.get("/register/t1", follow_redirects=False)
|
|
assert res.status_code in (302, 303)
|
|
assert "/manage/credentials" in res.headers["location"]
|
|
assert "setup=1" in res.headers["location"]
|
|
|
|
# Token should be marked used
|
|
link = await magic_link_repo.get_by_token("t1")
|
|
assert link is not None
|
|
assert link.used is True
|
|
|
|
# User should exist
|
|
user = await user_repo.get_by_username("newuser")
|
|
assert user is not None
|
|
assert "users" in user.groups
|
|
|
|
|
|
async def test_register_used_token_returns_error(client: AsyncClient) -> None:
|
|
app = client._transport.app # type: ignore[union-attr]
|
|
repo = app.state.magic_link_repo
|
|
await repo.create(
|
|
MagicLink(
|
|
token="used",
|
|
username="newuser",
|
|
expires_at=datetime.now(UTC) + timedelta(hours=1),
|
|
used=True,
|
|
)
|
|
)
|
|
|
|
res = await client.get("/register/used", follow_redirects=False)
|
|
assert res.status_code == 400
|