auth(2fa): block setup after enable to avoid secret reissue
All checks were successful
CI / test (push) Successful in 43s
All checks were successful
CI / test (push) Successful in 43s
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from sqlalchemy import select
|
||||
|
||||
from app.auth.models import EmailVerificationToken
|
||||
from app.utils.totp import _totp_code
|
||||
|
||||
|
||||
async def test_register_verify_login_and_me(client, db_session):
|
||||
@@ -115,3 +116,35 @@ async def test_revoke_all_sessions_invalidates_access_and_refresh(client, db_ses
|
||||
|
||||
refresh_response = await client.post("/api/v1/auth/refresh", json={"refresh_token": refresh_token})
|
||||
assert refresh_response.status_code == 401
|
||||
|
||||
|
||||
async def test_twofa_setup_is_blocked_when_already_enabled(client, db_session):
|
||||
payload = {
|
||||
"email": "dana@example.com",
|
||||
"name": "Dana",
|
||||
"username": "dana",
|
||||
"password": "strongpass123",
|
||||
}
|
||||
await client.post("/api/v1/auth/register", json=payload)
|
||||
token_row = await db_session.execute(select(EmailVerificationToken).order_by(EmailVerificationToken.id.desc()))
|
||||
verify_token = token_row.scalar_one().token
|
||||
await client.post("/api/v1/auth/verify-email", json={"token": verify_token})
|
||||
|
||||
login_response = await client.post(
|
||||
"/api/v1/auth/login",
|
||||
json={"email": payload["email"], "password": payload["password"]},
|
||||
)
|
||||
assert login_response.status_code == 200
|
||||
access_token = login_response.json()["access_token"]
|
||||
headers = {"Authorization": f"Bearer {access_token}"}
|
||||
|
||||
setup_response = await client.post("/api/v1/auth/2fa/setup", headers=headers)
|
||||
assert setup_response.status_code == 200
|
||||
secret = setup_response.json()["secret"]
|
||||
code = _totp_code(secret)
|
||||
|
||||
enable_response = await client.post("/api/v1/auth/2fa/enable", headers=headers, json={"code": code})
|
||||
assert enable_response.status_code == 200
|
||||
|
||||
setup_again_response = await client.post("/api/v1/auth/2fa/setup", headers=headers)
|
||||
assert setup_again_response.status_code == 400
|
||||
|
||||
Reference in New Issue
Block a user