auth(2fa): block setup after enable to avoid secret reissue
All checks were successful
CI / test (push) Successful in 43s

This commit is contained in:
2026-03-08 19:07:20 +03:00
parent af1ce20640
commit d069ff1121
4 changed files with 38 additions and 1 deletions

View File

@@ -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