feat: login por sessao (troca HTTP Basic), UsuarioLogado, exigir_admin

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>

tests/test_auth.py

@@ -1,10 +1,9 @@
-import base64
-
 import pytest
-from fastapi import Depends, FastAPI
+from fastapi import Depends, FastAPI, HTTPException
 from fastapi.testclient import TestClient
 
-from app.auth import exigir_login
+from app import session
+from app.auth import NaoAutenticado, UsuarioLogado, autenticar_credenciais, exigir_admin, exigir_login
 from app.db import funcionarios as db_funcionarios
 from app.db.connection import get_connection
 
@@ -12,42 +11,79 @@ app_teste = FastAPI()
 
 
 @app_teste.get("/protegido")
-def rota_protegida(usuario: str = Depends(exigir_login)):
-    return {"usuario": usuario}
+def rota_protegida(usuario: UsuarioLogado = Depends(exigir_login)):
+    return {"usuario": str(usuario), "perfil": usuario.perfil}
 
 
-client = TestClient(app_teste)
+@app_teste.get("/so-admin")
+def rota_admin(usuario: UsuarioLogado = Depends(exigir_admin)):
+    return {"usuario": str(usuario)}
 
 
-def _basic(usuario: str, senha: str) -> dict:
-    token = base64.b64encode(f"{usuario}:{senha}".encode()).decode()
-    return {"Authorization": f"Basic {token}"}
+@app_teste.exception_handler(NaoAutenticado)
+def _handler_nao_autenticado(request, exc):
+    raise HTTPException(status_code=401, detail="não autenticado")
 
 
-def test_sem_credenciais_retorna_401():
+client = TestClient(app_teste, follow_redirects=False)
+
+
+def _cookie_sessao(login: str, nome: str, perfil: str) -> dict:
+    return {session.SESSION_COOKIE_NAME: session.criar_token(login, nome, perfil)}
+
+
+def test_sem_cookie_retorna_401():
     resposta = client.get("/protegido")
     assert resposta.status_code == 401
 
 
-def test_credenciais_erradas_retorna_401():
-    resposta = client.get("/protegido", headers=_basic("admin", "senha_errada"))
+def test_cookie_invalido_retorna_401():
+    resposta = client.get("/protegido", cookies={session.SESSION_COOKIE_NAME: "lixo"})
     assert resposta.status_code == 401
 
 
-def test_credenciais_corretas_retorna_200():
-    # ADMIN_USER/ADMIN_PASSWORD vêm de tests/conftest.py ("admin"/"teste123")
-    resposta = client.get("/protegido", headers=_basic("admin", "teste123"))
+def test_cookie_valido_retorna_dados_do_usuario():
+    resposta = client.get("/protegido", cookies=_cookie_sessao("admin", "Admin", "ADMIN"))
     assert resposta.status_code == 200
-    assert resposta.json() == {"usuario": "admin"}
+    assert resposta.json() == {"usuario": "Admin", "perfil": "ADMIN"}
 
 
-def test_admin_user_e_senha_vazios_nao_autentica_com_credenciais_vazias(monkeypatch):
-    # Sem ADMIN_USER/ADMIN_PASSWORD configurados, o painel não pode ficar
-    # aberto para uma requisição com usuário/senha vazios (Basic Og==).
+def test_usuario_logado_str_retorna_nome():
+    usuario = UsuarioLogado(login="fulano", nome="Fulano da Silva", perfil="PADRAO")
+    assert str(usuario) == "Fulano da Silva"
+
+
+def test_usuario_logado_eh_admin():
+    assert UsuarioLogado(login="a", nome="A", perfil="ADMIN").eh_admin is True
+    assert UsuarioLogado(login="a", nome="A", perfil="PADRAO").eh_admin is False
+
+
+def test_exigir_admin_aceita_admin():
+    resposta = client.get("/so-admin", cookies=_cookie_sessao("admin", "Admin", "ADMIN"))
+    assert resposta.status_code == 200
+
+
+def test_exigir_admin_rejeita_padrao():
+    resposta = client.get("/so-admin", cookies=_cookie_sessao("fulano", "Fulano", "PADRAO"))
+    assert resposta.status_code == 403
+
+
+def test_autenticar_credenciais_admin_env():
+    # ADMIN_USER/ADMIN_PASSWORD vêm de tests/conftest.py ("admin"/"teste123")
+    usuario = autenticar_credenciais("admin", "teste123")
+    assert usuario is not None
+    assert usuario.perfil == "ADMIN"
+    assert usuario.login == "admin"
+
+
+def test_autenticar_credenciais_admin_env_senha_errada():
+    assert autenticar_credenciais("admin", "senha_errada") is None
+
+
+def test_autenticar_credenciais_admin_env_vazio_nao_autentica_vazio(monkeypatch):
     monkeypatch.setenv("ADMIN_USER", "")
     monkeypatch.setenv("ADMIN_PASSWORD", "")
-    resposta = client.get("/protegido", headers=_basic("", ""))
-    assert resposta.status_code == 401
+    assert autenticar_credenciais("", "") is None
 
 
 def _limpar_login(login: str) -> None:
@@ -63,29 +99,27 @@ def _limpar_login(login: str) -> None:
 def funcionario_teste():
     login = "teste.login.auth"
     _limpar_login(login)
-    funcionario_id = db_funcionarios.criar_funcionario("Fulano de Teste", login, "senhaForte123")
+    funcionario_id = db_funcionarios.criar_funcionario("Fulano de Teste", login, "senhaForte123", "PADRAO")
     yield login
     _limpar_login(login)
 
 
-def test_funcionario_ativo_com_senha_correta_autentica(funcionario_teste):
-    resposta = client.get("/protegido", headers=_basic(funcionario_teste, "senhaForte123"))
-    assert resposta.status_code == 200
-    assert resposta.json() == {"usuario": "Fulano de Teste"}
+def test_autenticar_credenciais_funcionario_ativo(funcionario_teste):
+    usuario = autenticar_credenciais(funcionario_teste, "senhaForte123")
+    assert usuario is not None
+    assert usuario.nome == "Fulano de Teste"
+    assert usuario.perfil == "PADRAO"
 
 
-def test_funcionario_com_senha_errada_nao_autentica(funcionario_teste):
-    resposta = client.get("/protegido", headers=_basic(funcionario_teste, "senhaErrada"))
-    assert resposta.status_code == 401
+def test_autenticar_credenciais_funcionario_senha_errada(funcionario_teste):
+    assert autenticar_credenciais(funcionario_teste, "senhaErrada") is None
 
 
-def test_funcionario_inativo_nao_autentica(funcionario_teste):
+def test_autenticar_credenciais_funcionario_inativo(funcionario_teste):
     funcionario = db_funcionarios.obter_por_login(funcionario_teste)
     db_funcionarios.alternar_ativo(funcionario["ID"])  # desativa
-    resposta = client.get("/protegido", headers=_basic(funcionario_teste, "senhaForte123"))
-    assert resposta.status_code == 401
+    assert autenticar_credenciais(funcionario_teste, "senhaForte123") is None
 
 
-def test_login_inexistente_nao_autentica():
-    resposta = client.get("/protegido", headers=_basic("nao-existe-esse-login", "qualquer"))
-    assert resposta.status_code == 401
+def test_autenticar_credenciais_login_inexistente():
+    assert autenticar_credenciais("nao-existe-esse-login", "qualquer") is None