补充测试覆盖:认证、云图、管理后台、资料、收藏、云类型模块共 78 个测试用例

This commit is contained in:
2026-07-26 01:16:17 +08:00
parent 77bf076ad0
commit 6b1f0d5f5f
7 changed files with 1288 additions and 0 deletions
+171
View File
@@ -0,0 +1,171 @@
import uuid
import pytest
from fastapi.testclient import TestClient
from app.routers import auth as auth_router
from tests.conftest import bearer, png_bytes
ADMIN_EMAIL = "admin@example.com"
ADMIN_PASSWORD = "admin-password-123"
def _capture_email(monkeypatch) -> dict[str, str]:
captured: dict[str, str] = {}
async def capture(recipient: str, token: str) -> None:
captured["token"] = token
monkeypatch.setattr(auth_router, "send_confirmation_email", capture)
return captured
def _register_and_confirm(client, monkeypatch, email, password, username) -> dict[str, str]:
captured = _capture_email(monkeypatch)
r = client.post(
"/api/v1/auth/register",
json={"email": email, "password": password, "username": username},
)
assert r.status_code == 201
client.post("/api/v1/auth/confirm-email", json={"token": captured["token"]})
lr = client.post("/api/v1/auth/login", json={"email": email, "password": password})
body = lr.json()
return {"token": body["access_token"], "user_id": body["user"]["id"]}
def _admin_token(client: TestClient) -> str:
r = client.post(
"/api/v1/auth/login",
json={"email": ADMIN_EMAIL, "password": ADMIN_PASSWORD},
)
return r.json()["access_token"]
def _upload_and_approve(client: TestClient, token: str) -> str:
ur = client.post(
"/api/v1/clouds",
headers=bearer(token),
files={"image": ("cloud.png", png_bytes(), "image/png")},
data={"cloud_type_id": "1", "is_hidden": "false"},
)
cloud_id = ur.json()["cloud"]["id"]
client.patch(
"/api/v1/admin/clouds/status",
headers=bearer(_admin_token(client)),
json={"ids": [cloud_id], "status": "approved"},
)
return cloud_id
# ---------------------------------------------------------------------------
# get profile
# ---------------------------------------------------------------------------
def test_get_profile(client: TestClient, monkeypatch) -> None:
info = _register_and_confirm(
client, monkeypatch,
f"prof-{uuid.uuid4().hex[:8]}@example.com",
"profpass1", "资料展示",
)
r = client.get(f"/api/v1/profiles/{info['user_id']}")
assert r.status_code == 200
assert r.json()["username"] == "资料展示"
def test_get_nonexistent_profile(client: TestClient) -> None:
r = client.get(f"/api/v1/profiles/{uuid.uuid4()}")
assert r.status_code == 404
# ---------------------------------------------------------------------------
# update my profile
# ---------------------------------------------------------------------------
def test_update_me(client: TestClient, monkeypatch) -> None:
info = _register_and_confirm(
client, monkeypatch,
f"updme-{uuid.uuid4().hex[:8]}@example.com",
"updmepass", "原昵称",
)
r = client.patch(
"/api/v1/profiles/me",
headers=bearer(info["token"]),
json={"username": "新昵称"},
)
assert r.status_code == 200
assert r.json()["username"] == "新昵称"
def test_update_me_short_username(client: TestClient, monkeypatch) -> None:
info = _register_and_confirm(
client, monkeypatch,
f"shortname-{uuid.uuid4().hex[:8]}@example.com",
"shortname", "短名字测试",
)
r = client.patch(
"/api/v1/profiles/me",
headers=bearer(info["token"]),
json={"username": "A"},
)
assert r.status_code == 422
def test_update_me_duplicate_username(client: TestClient, monkeypatch) -> None:
_register_and_confirm(
client, monkeypatch,
f"dupname-u1-{uuid.uuid4().hex[:8]}@example.com",
"pass11112", "已占用的名字",
)
info2 = _register_and_confirm(
client, monkeypatch,
f"dupname-u2-{uuid.uuid4().hex[:8]}@example.com",
"pass22223", "第二人",
)
r = client.patch(
"/api/v1/profiles/me",
headers=bearer(info2["token"]),
json={"username": "已占用的名字"},
)
assert r.status_code == 409
assert "昵称" in r.json()["detail"]
# ---------------------------------------------------------------------------
# get user clouds
# ---------------------------------------------------------------------------
def test_get_user_clouds_public(client: TestClient, monkeypatch) -> None:
info = _register_and_confirm(
client, monkeypatch,
f"pubcloud-{uuid.uuid4().hex[:8]}@example.com",
"pubcloud1", "公开云主",
)
_upload_and_approve(client, info["token"])
r = client.get(f"/api/v1/profiles/{info['user_id']}/clouds")
assert r.status_code == 200
assert r.json()["total"] >= 1
for item in r.json()["items"]:
assert item["status"] == "approved"
def test_get_user_clouds_as_owner(client: TestClient, monkeypatch) -> None:
info = _register_and_confirm(
client, monkeypatch,
f"owncloud-{uuid.uuid4().hex[:8]}@example.com",
"owncloud1", "私有云主",
)
cloud_id = _upload_and_approve(client, info["token"])
r = client.get(
f"/api/v1/profiles/{info['user_id']}/clouds",
headers=bearer(info["token"]),
)
assert r.status_code == 200
assert r.json()["total"] >= 1
def test_get_nonexistent_user_clouds(client: TestClient) -> None:
r = client.get(f"/api/v1/profiles/{uuid.uuid4()}/clouds")
assert r.status_code == 404