Files
cma-management/backend/tests/test_auth.py
T
Hermes CI Fix 40d522cd34 fix(security): retire default-credential surface (yanxue-20260914-cma-defaultcred-code-fix)
- Login.vue: no prefilled default creds; no anonymous entity enumeration on mount

- deploy.sh: readiness polling (0.5s/30s cap) replaces fixed sleep 3; login smoke reads same-source 600 secret file, missing file = explicit abort

- 19 verify/check/gen_report scripts: shared _admin_creds() loader (CMA_ADMIN_ENV -> ops .secrets -> profile copy); no hardcoded creds

- tests: single TEST_USER_PASSWORD fixture constant; docs: pointer to 600 secret file; removed 4 legacy .bak + pycache residue
2026-09-14 17:39:53 +08:00

97 lines
2.9 KiB
Python

"""
认证模块测试
"""
import pytest
from fastapi.testclient import TestClient
from sqlalchemy.orm import Session
from tests.conftest import create_test_user, get_token_for_user, auth_header, TEST_USER_PASSWORD
class TestAuth:
"""用户认证测试"""
def test_login_success(self, client: TestClient, db: Session):
"""登录成功"""
create_test_user(db)
resp = client.post("/api/cma/auth/login", json={
"username": "testadmin",
"password": TEST_USER_PASSWORD,
"entity_id": 1,
})
assert resp.status_code == 200
data = resp.json()
assert "token" in data
assert data["user"]["username"] == "testadmin"
assert data["user"]["role"] == "ceo"
def test_login_wrong_password(self, client: TestClient, db: Session):
"""密码错误"""
create_test_user(db)
resp = client.post("/api/cma/auth/login", json={
"username": "testadmin",
"password": "wrongpass",
})
assert resp.status_code == 401
def test_login_nonexistent_user(self, client: TestClient):
"""用户不存在"""
resp = client.post("/api/cma/auth/login", json={
"username": "nobody",
"password": TEST_USER_PASSWORD,
})
assert resp.status_code == 401
def test_me_with_valid_token(self, client: TestClient, db: Session):
"""有效token获取用户信息"""
create_test_user(db)
token = get_token_for_user(client)
resp = client.get("/api/cma/auth/me", headers={
"Authorization": f"Bearer {token}"
})
assert resp.status_code == 200
assert resp.json()["username"] == "testadmin"
def test_me_without_token(self, client: TestClient):
"""无token访问需要认证的接口"""
resp = client.get("/api/cma/auth/me")
assert resp.status_code == 403 # HTTPBearer auto_error
def test_register(self, client: TestClient, db: Session):
"""注册新用户"""
resp = client.post("/api/cma/auth/register", json={
"username": "newuser",
"password": "newpass123",
"name": "新用户",
"role": "business",
})
assert resp.status_code == 200
assert resp.json()["message"] == "注册成功"
# 验证可以登录
login_resp = client.post("/api/cma/auth/login", json={
"username": "newuser",
"password": "newpass123",
"entity_id": 1,
})
assert login_resp.status_code == 200
def test_register_duplicate(self, client: TestClient, db: Session):
"""重复用户名注册"""
create_test_user(db)
resp = client.post("/api/cma/auth/register", json={
"username": "testadmin",
"password": TEST_USER_PASSWORD,
"name": "重复用户",
})
assert resp.status_code == 400