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