Xiaoxia AI
179 lines
4.9 KiB
Python
179 lines
4.9 KiB
Python
"""
|
|
API 集成测试
|
|
"""
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from apps.api.main import app
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
class TestAuthAPI:
|
|
"""认证 API 集成测试"""
|
|
|
|
def test_register_success(self):
|
|
"""测试注册成功"""
|
|
response = client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "test@example.com",
|
|
"password": "SecurePass123",
|
|
"username": "testuser",
|
|
"display_name": "Test User",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 201
|
|
data = response.json()
|
|
assert data["email"] == "test@example.com"
|
|
assert data["username"] == "testuser"
|
|
assert "user_id" in data
|
|
|
|
def test_register_duplicate_email(self):
|
|
"""测试重复邮箱注册"""
|
|
# 先注册一个用户
|
|
client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "duplicate@example.com",
|
|
"password": "SecurePass123",
|
|
"username": "user1",
|
|
"display_name": "User 1",
|
|
},
|
|
)
|
|
|
|
# 尝试用相同邮箱再次注册
|
|
response = client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "duplicate@example.com",
|
|
"password": "SecurePass123",
|
|
"username": "user2",
|
|
"display_name": "User 2",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 400
|
|
assert "already registered" in response.json()["detail"].lower()
|
|
|
|
def test_login_success(self):
|
|
"""测试登录成功"""
|
|
# 先注册
|
|
client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "login@example.com",
|
|
"password": "SecurePass123",
|
|
"username": "loginuser",
|
|
"display_name": "Login User",
|
|
},
|
|
)
|
|
|
|
# 登录
|
|
response = client.post(
|
|
"/api/v1/auth/login",
|
|
json={
|
|
"email": "login@example.com",
|
|
"password": "SecurePass123",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert "access_token" in data
|
|
assert "refresh_token" in data
|
|
assert data["token_type"] == "bearer"
|
|
|
|
def test_login_wrong_password(self):
|
|
"""测试密码错误"""
|
|
response = client.post(
|
|
"/api/v1/auth/login",
|
|
json={
|
|
"email": "login@example.com",
|
|
"password": "WrongPassword123",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 401
|
|
|
|
|
|
class TestWorkspaceAPI:
|
|
"""工作空间 API 集成测试"""
|
|
|
|
def setup_method(self):
|
|
"""每个测试前的准备"""
|
|
# 注册并登录，获取 token
|
|
client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "workspace@example.com",
|
|
"password": "SecurePass123",
|
|
"username": "workspaceuser",
|
|
"display_name": "Workspace User",
|
|
},
|
|
)
|
|
|
|
response = client.post(
|
|
"/api/v1/auth/login",
|
|
json={
|
|
"email": "workspace@example.com",
|
|
"password": "SecurePass123",
|
|
},
|
|
)
|
|
|
|
self.token = response.json()["access_token"]
|
|
self.headers = {"Authorization": f"Bearer {self.token}"}
|
|
|
|
def test_create_workspace(self):
|
|
"""测试创建工作空间"""
|
|
response = client.post(
|
|
"/api/v1/workspaces",
|
|
json={
|
|
"name": "My Workspace",
|
|
"subscription_plan": "free",
|
|
},
|
|
headers=self.headers,
|
|
)
|
|
|
|
assert response.status_code == 201
|
|
data = response.json()
|
|
assert data["name"] == "My Workspace"
|
|
assert data["subscription_plan"] == "free"
|
|
assert data["max_projects"] == 3
|
|
|
|
def test_list_workspaces(self):
|
|
"""测试获取工作空间列表"""
|
|
# 创建工作空间
|
|
client.post(
|
|
"/api/v1/workspaces",
|
|
json={
|
|
"name": "Workspace 1",
|
|
},
|
|
headers=self.headers,
|
|
)
|
|
|
|
# 获取列表
|
|
response = client.get("/api/v1/workspaces", headers=self.headers)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert len(data["workspaces"]) > 0
|
|
assert data["workspaces"][0]["name"] == "Workspace 1"
|
|
|
|
def test_create_workspace_unauthorized(self):
|
|
"""测试未登录创建工作空间"""
|
|
response = client.post(
|
|
"/api/v1/workspaces",
|
|
json={
|
|
"name": "Unauthorized Workspace",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 403 # FastAPI HTTPBearer 返回 403
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__, "-v"])
|