diff options
Diffstat (limited to 'packages/meshbay-hub/tests/test_password_change.py')
| -rw-r--r-- | packages/meshbay-hub/tests/test_password_change.py | 140 |
1 files changed, 140 insertions, 0 deletions
diff --git a/packages/meshbay-hub/tests/test_password_change.py b/packages/meshbay-hub/tests/test_password_change.py new file mode 100644 index 0000000..aba2d9a --- /dev/null +++ b/packages/meshbay-hub/tests/test_password_change.py @@ -0,0 +1,140 @@ +""" +Passphrase change — Flow A of docs/auth-confirm.md. + +The hub's part is small: re-prove the current passphrase, swap the auth_key +verifier, invalidate every other session, keep the caller's. The re-wrapping of +per-node identity bundles is the client's job and does not touch the hub, so it +is not exercised here. +""" + +import base64 +import hashlib + +import pytest +from sqlalchemy import select + +from meshbay_hub.db.models import IPLog, RefreshToken, User + + +def _auth_key(password: str, username: str) -> str: + salt = hashlib.sha256(f"meshbay:auth:v1:{username}".encode()).digest() + return base64.b64encode( + hashlib.pbkdf2_hmac("sha512", password.encode(), salt, 600_000, 32)).decode() + + +async def _register(client, username, password="the-first-passphrase"): + r = await client.post("/v1/users/register", json={ + "username": username, "email": f"{username}@example.com", + "auth_key": _auth_key(password, username), + }) + assert r.status_code in (200, 201), r.text + login = await client.post("/v1/users/login", json={ + "username": username, "auth_key": _auth_key(password, username)}) + assert login.status_code == 200, login.text + return login.json() + + +@pytest.mark.asyncio +async def test_change_then_sign_in_with_the_new_passphrase(client): + old, new = "the-first-passphrase", "a-second-passphrase-entirely" + session = await _register(client, "alice", old) + + r = await client.post("/v1/users/password", json={ + "old_auth_key": _auth_key(old, "alice"), + "new_auth_key": _auth_key(new, "alice"), + }, headers={"Authorization": f"Bearer {session['access_token']}"}) + assert r.status_code == 200, r.text + assert r.json()["status"] == "changed" + + assert (await client.post("/v1/users/login", json={ + "username": "alice", "auth_key": _auth_key(old, "alice")})).status_code == 401 + assert (await client.post("/v1/users/login", json={ + "username": "alice", "auth_key": _auth_key(new, "alice")})).status_code == 200 + + +@pytest.mark.asyncio +async def test_wrong_current_passphrase_is_refused_and_changes_nothing(client): + old = "the-first-passphrase" + session = await _register(client, "bob", old) + + r = await client.post("/v1/users/password", json={ + "old_auth_key": _auth_key("not the passphrase", "bob"), + "new_auth_key": _auth_key("some-new-passphrase", "bob"), + }, headers={"Authorization": f"Bearer {session['access_token']}"}) + assert r.status_code == 403 + + assert (await client.post("/v1/users/login", json={ + "username": "bob", "auth_key": _auth_key(old, "bob")})).status_code == 200 + + +@pytest.mark.asyncio +async def test_new_must_differ_from_old(client): + old = "the-first-passphrase" + session = await _register(client, "carol", old) + + r = await client.post("/v1/users/password", json={ + "old_auth_key": _auth_key(old, "carol"), + "new_auth_key": _auth_key(old, "carol"), + }, headers={"Authorization": f"Bearer {session['access_token']}"}) + assert r.status_code == 400 + + +@pytest.mark.asyncio +async def test_unauthenticated_call_is_rejected(client): + """A session is required — the current passphrase alone is not a credential.""" + await _register(client, "dave", "the-first-passphrase") + r = await client.post("/v1/users/password", json={ + "old_auth_key": _auth_key("the-first-passphrase", "dave"), + "new_auth_key": _auth_key("a-new-one", "dave"), + }) + assert r.status_code in (401, 403, 422) + + +@pytest.mark.asyncio +async def test_other_sessions_are_invalidated_and_the_caller_keeps_one( + client, db_session): + old, new = "the-first-passphrase", "a-second-passphrase-entirely" + first = await _register(client, "erin", old) + # A second browser signs in before the change. + second = (await client.post("/v1/users/login", json={ + "username": "erin", "auth_key": _auth_key(old, "erin")})).json() + + r = await client.post("/v1/users/password", json={ + "old_auth_key": _auth_key(old, "erin"), + "new_auth_key": _auth_key(new, "erin"), + }, headers={"Authorization": f"Bearer {first['access_token']}"}) + assert r.status_code == 200, r.text + + # The other browser's refresh token is dead. + stale = await client.post("/v1/users/token/refresh", json={ + "refresh_token": second["refresh_token"]}) + assert stale.status_code == 401 + + # The caller was handed a fresh pair that still works. + fresh = await client.post("/v1/users/token/refresh", json={ + "refresh_token": r.json()["refresh_token"]}) + assert fresh.status_code == 200, fresh.text + + uid = (await db_session.execute( + select(User.id).where(User.username == "erin"))).scalar_one() + live = (await db_session.execute( + select(RefreshToken).where(RefreshToken.user_id == uid, + RefreshToken.revoked.is_(False)))).scalars().all() + # Only the family issued to the caller (the refresh above rotated it once). + assert len(live) == 1 + + +@pytest.mark.asyncio +async def test_the_change_is_logged(client, db_session): + old, new = "the-first-passphrase", "a-second-passphrase-entirely" + session = await _register(client, "frank", old) + await client.post("/v1/users/password", json={ + "old_auth_key": _auth_key(old, "frank"), + "new_auth_key": _auth_key(new, "frank"), + }, headers={"Authorization": f"Bearer {session['access_token']}"}) + + uid = (await db_session.execute( + select(User.id).where(User.username == "frank"))).scalar_one() + events = {e.event for e in (await db_session.execute( + select(IPLog).where(IPLog.user_id == uid))).scalars().all()} + assert "password_change" in events |