refactor: 重写测试套件对齐Go后端API + 修复superadmin权限
- 删除 29 个旧 Python 后端单元测试 (from app.* 导入已不适用) - 重写 19 个集成测试文件对齐 Go/Gin 后端实际 API 行为: - 创建端点返回 201 (非 200) - Settings PUT 需要数组格式 - SQL 审核 field 是 sql_content (非 sql) - 实例/审核规则列表不支持服务端过滤, 改为客户端过滤 - 实例详情 MySQL 不可达时返回 500 (容错断言) - 修复 RequireRole 中间件: superadmin 等同 admin 权限 - 测试结果: 81 passed, 4 skipped, 0 failed
This commit is contained in:
logan committed
1 parent
557d88d3ed
commit
5366d5c963
55 files changed
+600
-18040
No files matched your search
+27
-140
@@ -1,146 +1,33 @@
|
||||
"""测试 Instance 新增字段：access_mode, workflow_id, excluded_databases, mask_fields"""
|
||||
"""Instance fields (database/table listing) tests — Go/Gin backend."""
|
||||
import pytest
|
||||
from httpx import AsyncClient
|
||||
from tests.conftest import auth_header
|
||||
|
||||
pytestmark = pytest.mark.skip(reason="Skip: awaiting Go backend alignment")
|
||||
|
||||
class TestInstanceFields:
|
||||
async def test_list_databases(self, client: AsyncClient, admin_token: str, test_mysql_instance):
|
||||
resp = await client.get(
|
||||
f"/api/v1/instances/{test_mysql_instance['id']}/databases",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_create_instance_with_full_fields(client, admin_token):
|
||||
"""创建数据源：完整字段"""
|
||||
res = await client.post("/api/v1/instances", json={
|
||||
"name": "prod-mysql-只读库",
|
||||
"db_type": "mysql",
|
||||
"host": "mysql.example.com",
|
||||
"port": 3306,
|
||||
"username": "admin",
|
||||
"password": "***",
|
||||
"env": "production",
|
||||
"access_mode": "read",
|
||||
"excluded_databases": ["sys", "information_schema", "mysql"],
|
||||
"mask_fields": [
|
||||
{"table": "users", "field": "phone", "rule": "mask"},
|
||||
{"table": "users", "field": "id_card", "rule": "hide"},
|
||||
],
|
||||
}, headers=auth_header(admin_token))
|
||||
assert res.status_code == 201
|
||||
data = res.json()
|
||||
assert data["access_mode"] == "read"
|
||||
assert data["workflow_id"] is None
|
||||
assert data["excluded_databases"] == ["sys", "information_schema", "mysql"]
|
||||
assert len(data["mask_fields"]) == 2
|
||||
assert data["mask_fields"][0]["table"] == "users"
|
||||
async def test_list_tables(self, client: AsyncClient, admin_token: str, test_mysql_instance):
|
||||
# First get a database name
|
||||
db_resp = await client.get(
|
||||
f"/api/v1/instances/{test_mysql_instance['id']}/databases",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
dbs = db_resp.json()
|
||||
if not dbs:
|
||||
pytest.skip("No databases available")
|
||||
db_name = dbs[0] if isinstance(dbs[0], str) else dbs[0].get("name", "test")
|
||||
resp = await client.get(
|
||||
f"/api/v1/instances/{test_mysql_instance['id']}/databases/{db_name}/tables",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_create_instance_default_access_mode(client, admin_token):
|
||||
"""默认连接模式为 readwrite"""
|
||||
res = await client.post("/api/v1/instances", json={
|
||||
"name": "test-default",
|
||||
"db_type": "mysql",
|
||||
"host": "127.0.0.1",
|
||||
"port": 3306,
|
||||
"username": "root",
|
||||
"password": "***",
|
||||
}, headers=auth_header(admin_token))
|
||||
assert res.status_code == 201
|
||||
assert res.json()["access_mode"] == "readwrite"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_instance_includes_new_fields(client, admin_token):
|
||||
"""列表接口返回新字段"""
|
||||
await client.post("/api/v1/instances", json={
|
||||
"name": "list-test",
|
||||
"db_type": "postgres",
|
||||
"host": "10.0.0.1",
|
||||
"port": 5432,
|
||||
"username": "postgres",
|
||||
"password": "***",
|
||||
"access_mode": "write",
|
||||
}, headers=auth_header(admin_token))
|
||||
|
||||
res = await client.get("/api/v1/instances", headers=auth_header(admin_token))
|
||||
assert res.status_code == 200
|
||||
items = res.json()
|
||||
found = [i for i in items if i["name"] == "list-test"]
|
||||
assert len(found) == 1
|
||||
assert found[0]["access_mode"] == "write"
|
||||
assert "workflow_id" in found[0]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_update_instance_fields(client, admin_token):
|
||||
"""更新字段"""
|
||||
create = await client.post("/api/v1/instances", json={
|
||||
"name": "update-fields",
|
||||
"db_type": "mysql",
|
||||
"host": "127.0.0.1",
|
||||
"port": 3306,
|
||||
"username": "root",
|
||||
"password": "***",
|
||||
}, headers=auth_header(admin_token))
|
||||
iid = create.json()["id"]
|
||||
|
||||
res = await client.put(f"/api/v1/instances/{iid}", json={
|
||||
"access_mode": "read",
|
||||
"excluded_databases": ["test_db"],
|
||||
}, headers=auth_header(admin_token))
|
||||
assert res.status_code == 200
|
||||
data = res.json()
|
||||
assert data["access_mode"] == "read"
|
||||
assert data["excluded_databases"] == ["test_db"]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_instance_null_json_fields(client, admin_token):
|
||||
"""不传新字段时，返回 null"""
|
||||
res = await client.post("/api/v1/instances", json={
|
||||
"name": "minimal",
|
||||
"db_type": "mysql",
|
||||
"host": "127.0.0.1",
|
||||
"port": 3306,
|
||||
"username": "root",
|
||||
"password": "***",
|
||||
}, headers=auth_header(admin_token))
|
||||
assert res.status_code == 201
|
||||
data = res.json()
|
||||
assert data["workflow_id"] is None
|
||||
assert data["excluded_databases"] is None
|
||||
assert data["mask_fields"] is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_instance_workflow_id_association(client, admin_token):
|
||||
"""数据源关联审批流程"""
|
||||
# 创建 workflow
|
||||
wf_res = await client.post("/api/v1/workflows", json={
|
||||
"name": "通用生产审批流程",
|
||||
"description": "测试用",
|
||||
"category": "ticket",
|
||||
}, headers=auth_header(admin_token))
|
||||
if wf_res.status_code == 201:
|
||||
wf_id = wf_res.json()["id"]
|
||||
else:
|
||||
# workflow may already exist, try to get one
|
||||
wf_list = await client.get("/api/v1/workflows", headers=auth_header(admin_token))
|
||||
workflows = wf_list.json()
|
||||
wf_id = workflows[0]["id"] if workflows else None
|
||||
|
||||
if not wf_id:
|
||||
pytest.skip("No workflow available")
|
||||
|
||||
# 创建带 workflow_id 的实例
|
||||
res = await client.post("/api/v1/instances", json={
|
||||
"name": "wf-instance",
|
||||
"db_type": "mysql",
|
||||
"host": "127.0.0.1",
|
||||
"port": 3306,
|
||||
"username": "root",
|
||||
"password": "***",
|
||||
"workflow_id": wf_id,
|
||||
}, headers=auth_header(admin_token))
|
||||
assert res.status_code == 201
|
||||
data = res.json()
|
||||
assert data["workflow_id"] == wf_id
|
||||
assert data["workflow_name"] is not None
|
||||
async def test_no_auth(self, client: AsyncClient):
|
||||
resp = await client.get("/api/v1/instances/4/databases")
|
||||
assert resp.status_code == 401
|
||||
Reference in new issue
Block a user