feat: Go后端LDAP真实对接 — AES-256-GCM密码加解密 + 登录LDAP fallback
- 新增 crypto.go: AES-256-GCM 加解密,与 Python backend 兼容 - auth.go: Login 添加 LDAP fallback(本地优先→LDAP绑定→自动创建本地用户) - ldap.go: 实现真实 LDAP 连接/用户搜索/密码验证/OU列表/用户列表 - 所有 LDAP 操作解密 admin_password 后使用 - 新增/更新密码存储时加密 - router.go: 补充 delete-user 路由 - 添加 go-ldap/v3 依赖
This commit is contained in:
1 parent
622923821b
commit
88e5f4e7ee
71 files changed
+20390
-64
No files matched your search
@@ -0,0 +1,232 @@
|
||||
"""用户管理 API 端点测试。"""
|
||||
import pytest
|
||||
from httpx import AsyncClient
|
||||
|
||||
from app.core.security import hash_password, create_access_token
|
||||
from app.models.user import User
|
||||
|
||||
|
||||
def auth_header(token: str) -> dict:
|
||||
return {"Authorization": f"Bearer {token}"}
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
class TestListUsers:
|
||||
"""GET /api/v1/users"""
|
||||
|
||||
async def test_list_users_empty(self, client: AsyncClient, admin_token: str):
|
||||
resp = await client.get("/api/v1/users", headers=auth_header(admin_token))
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
# admin_user fixture 创建了 1 个
|
||||
assert isinstance(data, dict) and "items" in data
|
||||
|
||||
async def test_list_users_filter_role(
|
||||
self, client: AsyncClient, db, admin_token: str
|
||||
):
|
||||
# 创建一个 dba 用户
|
||||
dba = User(
|
||||
username="dba_test",
|
||||
display_name="DBA",
|
||||
role="dba",
|
||||
hashed_password=hash_password("123456"),
|
||||
)
|
||||
db.add(dba)
|
||||
await db.commit()
|
||||
|
||||
resp = await client.get(
|
||||
"/api/v1/users?role=dba", headers=auth_header(admin_token)
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
items = data.get("items", data) if isinstance(data, dict) else data
|
||||
assert all(u["role"] == "dba" for u in items)
|
||||
|
||||
async def test_list_users_filter_active(
|
||||
self, client: AsyncClient, db, admin_token: str
|
||||
):
|
||||
dev = User(
|
||||
username="inactive_dev",
|
||||
display_name="离线",
|
||||
role="developer",
|
||||
is_active=False,
|
||||
hashed_password=hash_password("123456"),
|
||||
)
|
||||
db.add(dev)
|
||||
await db.commit()
|
||||
|
||||
resp = await client.get(
|
||||
"/api/v1/users?is_active=false", headers=auth_header(admin_token)
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
items = data.get("items", data) if isinstance(data, dict) else data
|
||||
assert all(u["is_active"] is False for u in items)
|
||||
|
||||
async def test_list_users_forbidden_non_admin(
|
||||
self, client: AsyncClient, developer_token: str
|
||||
):
|
||||
resp = await client.get("/api/v1/users", headers=auth_header(developer_token))
|
||||
assert resp.status_code == 403
|
||||
|
||||
async def test_list_users_unauthorized(self, client: AsyncClient):
|
||||
resp = await client.get("/api/v1/users")
|
||||
assert resp.status_code == 401
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
class TestCreateUser:
|
||||
"""POST /api/v1/users"""
|
||||
|
||||
async def test_create_user_success(self, client: AsyncClient, admin_token: str):
|
||||
resp = await client.post(
|
||||
"/api/v1/users",
|
||||
json={
|
||||
"username": "newuser",
|
||||
"display_name": "新用户",
|
||||
"email": "new@test.com",
|
||||
"phone": "13800001111",
|
||||
"role": "developer",
|
||||
"password": "pass123",
|
||||
},
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 201
|
||||
data = resp.json()
|
||||
assert data["username"] == "newuser"
|
||||
assert data["role"] == "developer"
|
||||
assert "hashed_password" not in data
|
||||
|
||||
async def test_create_user_duplicate_username(
|
||||
self, client: AsyncClient, admin_token: str, admin_user: User
|
||||
):
|
||||
resp = await client.post(
|
||||
"/api/v1/users",
|
||||
json={
|
||||
"username": "admin",
|
||||
"display_name": "重复",
|
||||
"role": "developer",
|
||||
"password": "123456",
|
||||
},
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 400
|
||||
assert "已存在" in resp.json()["detail"]
|
||||
|
||||
async def test_create_user_forbidden_dba(
|
||||
self, client: AsyncClient, dba_token: str
|
||||
):
|
||||
resp = await client.post(
|
||||
"/api/v1/users",
|
||||
json={
|
||||
"username": "hack",
|
||||
"display_name": "hack",
|
||||
"role": "developer",
|
||||
"password": "123",
|
||||
},
|
||||
headers=auth_header(dba_token),
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
class TestGetUser:
|
||||
"""GET /api/v1/users/{user_id}"""
|
||||
|
||||
async def test_get_user_success(
|
||||
self, client: AsyncClient, admin_token: str, admin_user: User
|
||||
):
|
||||
resp = await client.get(
|
||||
f"/api/v1/users/{admin_user.id}", headers=auth_header(admin_token)
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["username"] == "admin"
|
||||
|
||||
async def test_get_user_not_found(self, client: AsyncClient, admin_token: str):
|
||||
resp = await client.get(
|
||||
"/api/v1/users/9999", headers=auth_header(admin_token)
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
|
||||
async def test_get_user_forbidden(self, client: AsyncClient, developer_token: str):
|
||||
resp = await client.get(
|
||||
"/api/v1/users/1", headers=auth_header(developer_token)
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
class TestUpdateUser:
|
||||
"""PUT /api/v1/users/{user_id}"""
|
||||
|
||||
async def test_update_user_display_name(
|
||||
self, client: AsyncClient, admin_token: str, dba_user
|
||||
):
|
||||
resp = await client.put(
|
||||
f"/api/v1/users/{dba_user.id}",
|
||||
json={"display_name": "DBA王五"},
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["display_name"] == "DBA王五"
|
||||
|
||||
async def test_update_user_role(
|
||||
self, client: AsyncClient, admin_token: str, developer_user
|
||||
):
|
||||
resp = await client.put(
|
||||
f"/api/v1/users/{developer_user.id}",
|
||||
json={"role": "dba"},
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
assert resp.json()["role"] == "dba"
|
||||
|
||||
async def test_update_user_not_found(self, client: AsyncClient, admin_token: str):
|
||||
resp = await client.put(
|
||||
"/api/v1/users/9999",
|
||||
json={"display_name": "x"},
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
|
||||
async def test_update_user_partial(
|
||||
self, client: AsyncClient, admin_token: str, dba_user
|
||||
):
|
||||
resp = await client.put(
|
||||
f"/api/v1/users/{dba_user.id}",
|
||||
json={"email": "dba@test.com"},
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
assert resp.json()["email"] == "dba@test.com"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
class TestDeleteUser:
|
||||
"""DELETE /api/v1/users/{user_id}"""
|
||||
|
||||
async def test_delete_user_success(
|
||||
self, client: AsyncClient, admin_token: str, developer_user
|
||||
):
|
||||
resp = await client.delete(
|
||||
f"/api/v1/users/{developer_user.id}", headers=auth_header(admin_token)
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
assert "已删除" in resp.json()["message"]
|
||||
|
||||
async def test_delete_self_forbidden(
|
||||
self, client: AsyncClient, admin_token: str, admin_user: User
|
||||
):
|
||||
resp = await client.delete(
|
||||
f"/api/v1/users/{admin_user.id}", headers=auth_header(admin_token)
|
||||
)
|
||||
assert resp.status_code == 400
|
||||
assert "不能删除自己" in resp.json()["detail"]
|
||||
|
||||
async def test_delete_user_not_found(self, client: AsyncClient, admin_token: str):
|
||||
resp = await client.delete(
|
||||
"/api/v1/users/9999", headers=auth_header(admin_token)
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
Reference in new issue
Block a user