sync: 同步 sql-platform-pro 最新代码 — 权限体系安全修复 v0.8.1
- P0: 实例访问控制全线接入权限组 (829测试) - P1: permissions字段激活 + 操作级权限检查 (843测试) - 前端: 删除数据权限页面, 权限组合并为唯一入口 - 新增: CodeMirror编辑器 + 查询模板 + 慢查询分析 - 新增: MongoDB管理页面 + Dashboard增强 - 修复: Pydantic V2迁移 + 前端TS错误修复
This commit is contained in:
1 parent
6fcd9465cf
commit
4c9427e546
105 files changed
+23680
-471
No files matched your search
@@ -0,0 +1,241 @@
|
||||
"""Schema 漂移检测 + 连接池管理 API 测试。"""
|
||||
import pytest
|
||||
from unittest.mock import patch, AsyncMock, MagicMock
|
||||
from httpx import AsyncClient
|
||||
|
||||
from tests.conftest import auth_header
|
||||
|
||||
|
||||
# ════════════════════════════════════════════════════════════════════
|
||||
# Schema Drift Tests
|
||||
# ════════════════════════════════════════════════════════════════════
|
||||
|
||||
|
||||
class TestSchemaDriftCapture:
|
||||
"""捕获 Schema 快照。"""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_capture_success(self, client, admin_token, test_instance):
|
||||
with patch("app.api.v1.endpoints.schema_drift.SchemaDriftDetector") as MockDet:
|
||||
instance_mock = MagicMock()
|
||||
instance_mock.capture_snapshot = AsyncMock(return_value=[{"col": "id"}])
|
||||
instance_mock.save_baseline = AsyncMock()
|
||||
MockDet.return_value = instance_mock
|
||||
|
||||
resp = await client.post(
|
||||
f"/api/v1/schema-drift/instances/{test_instance.id}/capture",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
body = resp.json()
|
||||
assert body["message"] == "Schema 快照已捕获"
|
||||
assert body["total_columns"] == 1
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_capture_instance_not_found(self, client, admin_token):
|
||||
resp = await client.post(
|
||||
"/api/v1/schema-drift/instances/9999/capture",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_capture_developer_forbidden(self, client, developer_token, test_instance):
|
||||
resp = await client.post(
|
||||
f"/api/v1/schema-drift/instances/{test_instance.id}/capture",
|
||||
headers=auth_header(developer_token),
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_capture_dba_allowed(self, client, dba_token, test_instance):
|
||||
with patch("app.api.v1.endpoints.schema_drift.SchemaDriftDetector") as MockDet:
|
||||
instance_mock = MagicMock()
|
||||
instance_mock.capture_snapshot = AsyncMock(return_value=[])
|
||||
instance_mock.save_baseline = AsyncMock()
|
||||
MockDet.return_value = instance_mock
|
||||
|
||||
resp = await client.post(
|
||||
f"/api/v1/schema-drift/instances/{test_instance.id}/capture",
|
||||
headers=auth_header(dba_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
|
||||
|
||||
class TestSchemaDriftCheck:
|
||||
"""检测 Schema 漂移。"""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_check_success(self, client, admin_token, test_instance):
|
||||
with patch("app.api.v1.endpoints.schema_drift.SchemaDriftDetector") as MockDet:
|
||||
report = MagicMock()
|
||||
report.to_dict.return_value = {"drifted": False, "changes": []}
|
||||
instance_mock = MagicMock()
|
||||
instance_mock.detect_drift = AsyncMock(return_value=report)
|
||||
MockDet.return_value = instance_mock
|
||||
|
||||
resp = await client.post(
|
||||
f"/api/v1/schema-drift/instances/{test_instance.id}/check",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
body = resp.json()
|
||||
assert body["drifted"] is False
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_check_instance_not_found(self, client, admin_token):
|
||||
resp = await client.post(
|
||||
"/api/v1/schema-drift/instances/9999/check",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
|
||||
|
||||
class TestSchemaHistory:
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_history_returns_data(self, client, admin_token, test_instance):
|
||||
resp = await client.get(
|
||||
f"/api/v1/schema-drift/instances/{test_instance.id}/history",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
body = resp.json()
|
||||
assert body["instance_id"] == test_instance.id
|
||||
assert "history" in body
|
||||
|
||||
|
||||
# ════════════════════════════════════════════════════════════════════
|
||||
# Connection Pool Tests
|
||||
# ════════════════════════════════════════════════════════════════════
|
||||
|
||||
|
||||
class TestPoolStats:
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stats_not_initialized(self, client, admin_token, test_instance):
|
||||
with patch("app.api.v1.endpoints.pool.get_pool_manager") as mock_pm:
|
||||
manager = MagicMock()
|
||||
manager.get_pool_stats.return_value = None
|
||||
mock_pm.return_value = manager
|
||||
|
||||
resp = await client.get(
|
||||
f"/api/v1/pools/instances/{test_instance.id}/stats",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
body = resp.json()
|
||||
assert body["status"] == "not_initialized"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stats_with_data(self, client, admin_token, test_instance):
|
||||
with patch("app.api.v1.endpoints.pool.get_pool_manager") as mock_pm:
|
||||
stats = MagicMock()
|
||||
stats.to_dict.return_value = {"active": 2, "idle": 3}
|
||||
manager = MagicMock()
|
||||
manager.get_pool_stats.return_value = stats
|
||||
mock_pm.return_value = manager
|
||||
|
||||
resp = await client.get(
|
||||
f"/api/v1/pools/instances/{test_instance.id}/stats",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
assert resp.json()["active"] == 2
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stats_instance_not_found(self, client, admin_token):
|
||||
resp = await client.get(
|
||||
"/api/v1/pools/instances/9999/stats",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stats_developer_forbidden(self, client, developer_token, test_instance):
|
||||
resp = await client.get(
|
||||
f"/api/v1/pools/instances/{test_instance.id}/stats",
|
||||
headers=auth_header(developer_token),
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
|
||||
class TestPoolReset:
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_reset_success(self, client, admin_token, test_instance):
|
||||
with patch("app.api.v1.endpoints.pool.get_pool_manager") as mock_pm:
|
||||
manager = MagicMock()
|
||||
manager.reset_pool = AsyncMock()
|
||||
mock_pm.return_value = manager
|
||||
|
||||
resp = await client.post(
|
||||
f"/api/v1/pools/instances/{test_instance.id}/reset",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
assert "已重置" in resp.json()["message"]
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_reset_dba_allowed(self, client, dba_token, test_instance):
|
||||
with patch("app.api.v1.endpoints.pool.get_pool_manager") as mock_pm:
|
||||
manager = MagicMock()
|
||||
manager.reset_pool = AsyncMock()
|
||||
mock_pm.return_value = manager
|
||||
|
||||
resp = await client.post(
|
||||
f"/api/v1/pools/instances/{test_instance.id}/reset",
|
||||
headers=auth_header(dba_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_reset_developer_forbidden(self, client, developer_token, test_instance):
|
||||
resp = await client.post(
|
||||
f"/api/v1/pools/instances/{test_instance.id}/reset",
|
||||
headers=auth_header(developer_token),
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
|
||||
class TestPoolsOverview:
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_overview_empty(self, client, admin_token):
|
||||
with patch("app.api.v1.endpoints.pool.get_pool_manager") as mock_pm:
|
||||
manager = MagicMock()
|
||||
manager.get_all_stats.return_value = {}
|
||||
mock_pm.return_value = manager
|
||||
|
||||
resp = await client.get(
|
||||
"/api/v1/pools/overview",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
assert resp.json()["total_pools"] == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_overview_with_pools(self, client, admin_token):
|
||||
with patch("app.api.v1.endpoints.pool.get_pool_manager") as mock_pm:
|
||||
stats = MagicMock()
|
||||
stats.to_dict.return_value = {"instance_id": 1, "active": 1}
|
||||
manager = MagicMock()
|
||||
manager.get_all_stats.return_value = {1: stats}
|
||||
mock_pm.return_value = manager
|
||||
|
||||
resp = await client.get(
|
||||
"/api/v1/pools/overview",
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
body = resp.json()
|
||||
assert body["total_pools"] == 1
|
||||
assert len(body["pools"]) == 1
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_overview_guest_forbidden(self, client, guest_token):
|
||||
resp = await client.get(
|
||||
"/api/v1/pools/overview",
|
||||
headers=auth_header(guest_token),
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
Reference in new issue
Block a user