init
This commit is contained in:
1 parent
c7fccf99c7
commit
8471f368eb
33 files changed
+444
-2095
No files matched your search
@@ -17,7 +17,6 @@ from app.models.announcement import Announcement
|
||||
from app.models.notification import NotificationChannel
|
||||
from app.models.ldap_config import LdapConfig
|
||||
from app.models.backup import Backup
|
||||
from app.models.slow_log import SlowLog
|
||||
from app.models.scheduled_task import ScheduledTask
|
||||
from app.models.schema_revision import SchemaRevision
|
||||
from app.models.masking_rule import MaskingRule
|
||||
|
||||
@@ -177,28 +177,6 @@ class TestShowGrants:
|
||||
assert len(data["grants"]) >= 1
|
||||
|
||||
|
||||
class TestSlowQueries:
|
||||
async def test_slow_queries_success(self, client: AsyncClient, test_instance, admin_token):
|
||||
mock_conn, _ = _mock_mysql_connection([
|
||||
{"start_time": "2024-01-01 00:00:00", "user_host": "root[root]@localhost",
|
||||
"query_time": "00:00:05", "lock_time": "00:00:00", "rows_sent": 1,
|
||||
"rows_examined": 1000, "sql_text": "SELECT * FROM big_table"},
|
||||
])
|
||||
|
||||
mock_aiomysql = _make_aiomysql_mock()
|
||||
mock_aiomysql.connect = AsyncMock(return_value=mock_conn)
|
||||
|
||||
with patch.dict(sys.modules, {"aiomysql": mock_aiomysql}):
|
||||
resp = await client.get(
|
||||
"/api/v1/instances/" + str(test_instance.id) + "/detail/slow-queries",
|
||||
params={"database": "test_db"},
|
||||
headers=auth_header(admin_token),
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert isinstance(data, list)
|
||||
|
||||
|
||||
class TestProcesslist:
|
||||
async def test_show_processlist_success(self, client: AsyncClient, test_instance, admin_token):
|
||||
mock_conn, _ = _mock_mysql_connection([
|
||||
|
||||
@@ -267,8 +267,8 @@ async def test_create_ldap_config_empty_password(client: AsyncClient, admin_toke
|
||||
# ────────────────────────────────────────────────────────────────────
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_ldap_connection_mock_success(client: AsyncClient, admin_token: str):
|
||||
"""配置存在时 POST /test 在 mock 模式下应返回 success=True"""
|
||||
async def test_ldap_connection_real_failure(client: AsyncClient, admin_token: str):
|
||||
"""配置存在时 POST /test 应尝试真实连接,测试环境无 LDAP 服务器则返回 success=False"""
|
||||
await client.put(
|
||||
"/api/v1/ldap-config",
|
||||
json=VALID_LDAP_PAYLOAD,
|
||||
@@ -282,11 +282,10 @@ async def test_ldap_connection_mock_success(client: AsyncClient, admin_token: st
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["success"] is True
|
||||
# mock 模式消息中包含 Mock 字样
|
||||
assert "Mock" in data["message"]
|
||||
# 消息中应包含主机和测试用户信息
|
||||
assert "zhangsan" in data["message"]
|
||||
# 测试环境无真实 LDAP 服务器,连接应失败
|
||||
assert data["success"] is False
|
||||
# 消息中应包含连接错误信息
|
||||
assert any(kw in data["message"] for kw in ["error", "fail", "未安装", "连接失败", "LDAP"])
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
"""Comprehensive tests for NotificationChannel CRUD, permissions, config masking, and test-send."""
|
||||
import pytest
|
||||
from unittest.mock import AsyncMock, patch
|
||||
from httpx import AsyncClient
|
||||
|
||||
from tests.conftest import auth_header
|
||||
@@ -268,7 +269,9 @@ async def test_delete_nonexistent_channel(client: AsyncClient, admin_token: str)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_dingtalk_success(client: AsyncClient, admin_token: str):
|
||||
@patch("app.services.notification_sender.NotificationSender.send", new_callable=AsyncMock)
|
||||
async def test_send_dingtalk_success(mock_send, client: AsyncClient, admin_token: str):
|
||||
mock_send.return_value = {"success": True, "message": "钉钉测试消息发送成功"}
|
||||
create = await client.post(BASE, json=DINGTALK_PAYLOAD, headers=auth_header(admin_token))
|
||||
cid = create.json()["id"]
|
||||
|
||||
@@ -297,7 +300,9 @@ async def test_send_dingtalk_missing_webhook(client: AsyncClient, admin_token: s
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_feishu_success(client: AsyncClient, admin_token: str):
|
||||
@patch("app.services.notification_sender.NotificationSender.send", new_callable=AsyncMock)
|
||||
async def test_send_feishu_success(mock_send, client: AsyncClient, admin_token: str):
|
||||
mock_send.return_value = {"success": True, "message": "飞书测试消息发送成功"}
|
||||
create = await client.post(BASE, json=FEISHU_PAYLOAD, headers=auth_header(admin_token))
|
||||
cid = create.json()["id"]
|
||||
|
||||
@@ -325,7 +330,9 @@ async def test_send_feishu_missing_webhook(client: AsyncClient, admin_token: str
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_email_success(client: AsyncClient, admin_token: str):
|
||||
@patch("app.services.notification_sender.NotificationSender.send", new_callable=AsyncMock)
|
||||
async def test_send_email_success(mock_send, client: AsyncClient, admin_token: str):
|
||||
mock_send.return_value = {"success": True, "message": "邮件测试消息发送成功"}
|
||||
create = await client.post(BASE, json=EMAIL_PAYLOAD, headers=auth_header(admin_token))
|
||||
cid = create.json()["id"]
|
||||
|
||||
@@ -351,9 +358,7 @@ async def test_send_email_missing_smtp_config(client: AsyncClient, admin_token:
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["success"] is False
|
||||
assert "smtp_port" in data["message"]
|
||||
assert "smtp_user" in data["message"]
|
||||
assert "from_addr" in data["message"]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
|
||||
@@ -2,6 +2,7 @@
|
||||
import json
|
||||
|
||||
import pytest
|
||||
from unittest.mock import AsyncMock, patch
|
||||
from httpx import AsyncClient
|
||||
|
||||
from tests.conftest import auth_header
|
||||
@@ -194,13 +195,15 @@ async def test_delete_notification_channel(client: AsyncClient, admin_token: str
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_notification_test_endpoint(client: AsyncClient, admin_token: str):
|
||||
@patch("app.services.notification_sender.NotificationSender.send", new_callable=AsyncMock)
|
||||
async def test_notification_test_endpoint(mock_send, client: AsyncClient, admin_token: str):
|
||||
mock_send.return_value = {"success": True, "message": "钉钉测试消息发送成功"}
|
||||
create = await client.post(
|
||||
"/api/v1/notifications",
|
||||
json={
|
||||
"name": "钉钉测试",
|
||||
"channel_type": "dingtalk",
|
||||
"config": {"webhook_url": "https://oapi.dingtalk.com/robot/send?access_token=test"},
|
||||
"config": {"webhook_url": "https://oapi.dingtalk.com/robot/send?access_token=***"},
|
||||
"is_enabled": True,
|
||||
},
|
||||
headers=auth_header(admin_token),
|
||||
@@ -365,9 +368,8 @@ async def test_ldap_test_connection(client: AsyncClient, admin_token: str):
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
# Mock mode should return success
|
||||
assert data["success"] is True
|
||||
assert "Mock" in data["message"]
|
||||
# 测试环境无真实 LDAP 服务器,应返回失败
|
||||
assert data["success"] is False
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
|
||||
@@ -1,433 +0,0 @@
|
||||
"""慢日志 API 端点测试 — P1 审计闭环。
|
||||
|
||||
测试范围:
|
||||
- 慢日志列表查看
|
||||
- 慢日志收集(需mock数据库连接)
|
||||
- 慢日志分析报告
|
||||
- 仪表盘 TOP 10 查询
|
||||
- 权限控制(仅 admin/dba)
|
||||
"""
|
||||
import pytest
|
||||
import pytest_asyncio
|
||||
from unittest.mock import patch, AsyncMock, MagicMock
|
||||
from httpx import AsyncClient
|
||||
|
||||
from app.core.security import create_access_token, hash_password
|
||||
from app.models.user import User
|
||||
from app.models.instance import Instance
|
||||
from app.services.slow_log_collector import SlowLogCollector, SlowLogEntry, SlowLogReport
|
||||
|
||||
|
||||
@pytest_asyncio.fixture
|
||||
async def admin_headers(admin_token: str) -> dict:
|
||||
return {"Authorization": f"Bearer {admin_token}"}
|
||||
|
||||
|
||||
@pytest_asyncio.fixture
|
||||
async def dba_headers(dba_token: str) -> dict:
|
||||
return {"Authorization": f"Bearer {dba_token}"}
|
||||
|
||||
|
||||
@pytest_asyncio.fixture
|
||||
async def developer_headers(developer_token: str) -> dict:
|
||||
return {"Authorization": f"Bearer {developer_token}"}
|
||||
|
||||
|
||||
@pytest_asyncio.fixture
|
||||
async def guest_headers(guest_token: str) -> dict:
|
||||
return {"Authorization": f"Bearer {guest_token}"}
|
||||
|
||||
|
||||
# ── SlowLogEntry 和 SlowLogReport 单元测试 ──
|
||||
|
||||
|
||||
class TestSlowLogModels:
|
||||
"""慢日志数据模型测试。"""
|
||||
|
||||
def test_slow_log_entry_to_dict(self):
|
||||
"""测试 SlowLogEntry 序列化。"""
|
||||
entry = SlowLogEntry(
|
||||
sql_digest="abc123",
|
||||
sql_text="SELECT * FROM users WHERE id = ?",
|
||||
execution_count=100,
|
||||
avg_time_ms=250.456,
|
||||
max_time_ms=1000.789,
|
||||
min_time_ms=50.123,
|
||||
rows_examined=10000,
|
||||
rows_sent=100,
|
||||
first_seen="2025-01-01 00:00:00",
|
||||
last_seen="2025-01-02 00:00:00",
|
||||
)
|
||||
d = entry.to_dict()
|
||||
assert d["sql_digest"] == "abc123"
|
||||
assert d["execution_count"] == 100
|
||||
assert d["avg_time_ms"] == 250.46 # rounded
|
||||
assert d["rows_examined"] == 10000
|
||||
assert d["first_seen"] == "2025-01-01 00:00:00"
|
||||
|
||||
def test_slow_log_entry_long_sql_truncated(self):
|
||||
"""测试长SQL被截断。"""
|
||||
long_sql = "SELECT * FROM t WHERE " + "id = 1 AND " * 100
|
||||
entry = SlowLogEntry(
|
||||
sql_digest="x", sql_text=long_sql,
|
||||
execution_count=1, avg_time_ms=100,
|
||||
max_time_ms=100, min_time_ms=100,
|
||||
rows_examined=1, rows_sent=1,
|
||||
)
|
||||
d = entry.to_dict()
|
||||
assert len(d["sql_text"]) <= 200
|
||||
|
||||
def test_slow_log_report_to_dict(self):
|
||||
"""测试 SlowLogReport 序列化。"""
|
||||
entry = SlowLogEntry(
|
||||
sql_digest="abc", sql_text="SELECT 1",
|
||||
execution_count=1, avg_time_ms=100,
|
||||
max_time_ms=100, min_time_ms=100,
|
||||
rows_examined=1, rows_sent=1,
|
||||
)
|
||||
report = SlowLogReport(
|
||||
total_slow_queries=1,
|
||||
avg_execution_time_ms=100.555,
|
||||
max_execution_time_ms=100.0,
|
||||
top_queries=[entry],
|
||||
time_distribution={"00-06": 0, "06-12": 1, "12-18": 0, "18-24": 0},
|
||||
)
|
||||
d = report.to_dict()
|
||||
assert d["total_slow_queries"] == 1
|
||||
assert d["avg_execution_time_ms"] == 100.56
|
||||
assert len(d["top_queries"]) == 1
|
||||
assert d["time_distribution"]["06-12"] == 1
|
||||
|
||||
|
||||
# ── SlowLogCollector 单元测试 ──
|
||||
|
||||
|
||||
class TestSlowLogCollector:
|
||||
"""慢日志收集器单元测试。"""
|
||||
|
||||
def test_save_and_get_slow_logs(self):
|
||||
"""测试保存和获取慢日志。"""
|
||||
collector = SlowLogCollector()
|
||||
collector._logs = {}
|
||||
|
||||
entries = [
|
||||
SlowLogEntry(
|
||||
sql_digest=f"digest_{i}", sql_text=f"SELECT {i}",
|
||||
execution_count=i + 1, avg_time_ms=100 * (i + 1),
|
||||
max_time_ms=200 * (i + 1), min_time_ms=10,
|
||||
rows_examined=1000 * (i + 1), rows_sent=10,
|
||||
)
|
||||
for i in range(3)
|
||||
]
|
||||
|
||||
import asyncio
|
||||
asyncio.get_event_loop().run_until_complete(
|
||||
collector.save_slow_logs(1, entries)
|
||||
)
|
||||
|
||||
logs = collector.get_slow_logs(1)
|
||||
assert len(logs) == 3
|
||||
|
||||
def test_get_slow_logs_empty(self):
|
||||
"""测试获取不存在的实例慢日志。"""
|
||||
collector = SlowLogCollector()
|
||||
collector._logs = {}
|
||||
logs = collector.get_slow_logs(999)
|
||||
assert logs == []
|
||||
|
||||
def test_get_report_empty(self):
|
||||
"""测试获取空报告。"""
|
||||
collector = SlowLogCollector()
|
||||
collector._logs = {}
|
||||
report = collector.get_report(999)
|
||||
assert report.total_slow_queries == 0
|
||||
|
||||
def test_get_report_with_data(self):
|
||||
"""测试获取有数据的报告。"""
|
||||
collector = SlowLogCollector()
|
||||
entries = [
|
||||
SlowLogEntry(
|
||||
sql_digest="d1", sql_text="SELECT * FROM users",
|
||||
execution_count=10, avg_time_ms=500,
|
||||
max_time_ms=1000, min_time_ms=200,
|
||||
rows_examined=50000, rows_sent=100,
|
||||
),
|
||||
SlowLogEntry(
|
||||
sql_digest="d2", sql_text="SELECT * FROM orders",
|
||||
execution_count=5, avg_time_ms=300,
|
||||
max_time_ms=600, min_time_ms=100,
|
||||
rows_examined=30000, rows_sent=50,
|
||||
),
|
||||
]
|
||||
|
||||
import asyncio
|
||||
asyncio.get_event_loop().run_until_complete(
|
||||
collector.save_slow_logs(1, entries)
|
||||
)
|
||||
|
||||
report = collector.get_report(1)
|
||||
assert report.total_slow_queries == 2
|
||||
assert report.max_execution_time_ms == 1000
|
||||
assert len(report.top_queries) <= 20 # TOP 20 max
|
||||
|
||||
|
||||
# ── API 端点权限测试 ──
|
||||
|
||||
|
||||
class TestSlowLogAPIPermissions:
|
||||
"""慢日志 API 权限控制测试。"""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_slow_logs_unauthorized(self, client: AsyncClient):
|
||||
"""测试未认证访问。"""
|
||||
resp = await client.get("/api/v1/slow-logs/instances/1")
|
||||
assert resp.status_code == 401
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_slow_logs_developer_forbidden(
|
||||
self, client: AsyncClient, developer_headers: dict
|
||||
):
|
||||
"""测试开发者无权查看慢日志。"""
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/instances/1",
|
||||
headers=developer_headers,
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_slow_logs_guest_forbidden(
|
||||
self, client: AsyncClient, guest_headers: dict
|
||||
):
|
||||
"""测试访客无权查看慢日志。"""
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/instances/1",
|
||||
headers=guest_headers,
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_collect_slow_logs_developer_forbidden(
|
||||
self, client: AsyncClient, developer_headers: dict
|
||||
):
|
||||
"""测试开发者无权收集慢日志。"""
|
||||
resp = await client.post(
|
||||
"/api/v1/slow-logs/instances/1/collect",
|
||||
headers=developer_headers,
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_report_developer_forbidden(
|
||||
self, client: AsyncClient, developer_headers: dict
|
||||
):
|
||||
"""测试开发者无权查看报告。"""
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/instances/1/report",
|
||||
headers=developer_headers,
|
||||
)
|
||||
assert resp.status_code == 403
|
||||
|
||||
|
||||
# ── API 端点功能测试 ──
|
||||
|
||||
|
||||
class TestSlowLogAPIFunctional:
|
||||
"""慢日志 API 功能测试。"""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_slow_logs_not_found(
|
||||
self, client: AsyncClient, admin_headers: dict, db
|
||||
):
|
||||
"""测试查看不存在实例的慢日志。"""
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/instances/999",
|
||||
headers=admin_headers,
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_slow_logs_empty(
|
||||
self, client: AsyncClient, admin_headers: dict, test_instance
|
||||
):
|
||||
"""测试查看实例慢日志(空列表)。"""
|
||||
collector = SlowLogCollector()
|
||||
collector._logs = {}
|
||||
|
||||
with patch.object(SlowLogCollector, '_logs', {}):
|
||||
resp = await client.get(
|
||||
f"/api/v1/slow-logs/instances/{test_instance.id}",
|
||||
headers=admin_headers,
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["instance_id"] == test_instance.id
|
||||
assert data["total"] == 0
|
||||
assert data["logs"] == []
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_slow_logs_with_data(
|
||||
self, client: AsyncClient, admin_headers: dict, test_instance
|
||||
):
|
||||
"""测试查看实例慢日志(有数据)。"""
|
||||
entries = [
|
||||
SlowLogEntry(
|
||||
sql_digest="d1", sql_text="SELECT * FROM users",
|
||||
execution_count=10, avg_time_ms=500,
|
||||
max_time_ms=1000, min_time_ms=200,
|
||||
rows_examined=50000, rows_sent=100,
|
||||
),
|
||||
]
|
||||
|
||||
collector = SlowLogCollector()
|
||||
collector._logs = {test_instance.id: entries}
|
||||
|
||||
with patch.object(SlowLogCollector, '_logs', {test_instance.id: entries}):
|
||||
resp = await client.get(
|
||||
f"/api/v1/slow-logs/instances/{test_instance.id}",
|
||||
headers=admin_headers,
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["total"] == 1
|
||||
assert data["logs"][0]["sql_digest"] == "d1"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_collect_slow_logs_unsupported_type(
|
||||
self, client: AsyncClient, admin_headers: dict, db
|
||||
):
|
||||
"""测试收集不支持类型的慢日志。"""
|
||||
# 创建Redis实例(redis不支持慢日志收集)
|
||||
instance = Instance(
|
||||
name="测试Redis", db_type="redis",
|
||||
host="127.0.0.1", port=6379,
|
||||
username="", password="test123",
|
||||
database="0", env="development",
|
||||
status="disconnected",
|
||||
)
|
||||
db.add(instance)
|
||||
await db.commit()
|
||||
await db.refresh(instance)
|
||||
|
||||
# endpoint先解密密码再检查类型,可能返回400或500
|
||||
resp = await client.post(
|
||||
f"/api/v1/slow-logs/instances/{instance.id}/collect",
|
||||
headers=admin_headers,
|
||||
)
|
||||
# 接受400(不支持类型)或500(解密/连接错误)
|
||||
assert resp.status_code in (400, 500)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_collect_slow_logs_not_found(
|
||||
self, client: AsyncClient, admin_headers: dict
|
||||
):
|
||||
"""测试收集不存在实例的慢日志。"""
|
||||
resp = await client.post(
|
||||
"/api/v1/slow-logs/instances/999/collect",
|
||||
headers=admin_headers,
|
||||
)
|
||||
assert resp.status_code == 404
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_report_not_found(
|
||||
self, client: AsyncClient, admin_headers: dict
|
||||
):
|
||||
"""测试获取不存在实例的报告。"""
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/instances/999/report",
|
||||
headers=admin_headers,
|
||||
)
|
||||
# 报告端点不检查实例存在性,返回空报告
|
||||
assert resp.status_code == 200
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_dashboard_top_queries(
|
||||
self, client: AsyncClient, admin_headers: dict
|
||||
):
|
||||
"""测试仪表盘 TOP 查询。"""
|
||||
collector = SlowLogCollector()
|
||||
collector._logs = {}
|
||||
|
||||
with patch.object(SlowLogCollector, '_logs', {}):
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/dashboard/top",
|
||||
headers=admin_headers,
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert "total_instances" in data
|
||||
assert "top_queries" in data
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_dashboard_top_queries_with_data(
|
||||
self, client: AsyncClient, admin_headers: dict
|
||||
):
|
||||
"""测试仪表盘 TOP 查询(有数据)。"""
|
||||
entries = [
|
||||
SlowLogEntry(
|
||||
sql_digest=f"d{i}", sql_text=f"SELECT {i}",
|
||||
execution_count=i + 1, avg_time_ms=100 * (i + 1),
|
||||
max_time_ms=200 * (i + 1), min_time_ms=10,
|
||||
rows_examined=1000, rows_sent=10,
|
||||
)
|
||||
for i in range(15)
|
||||
]
|
||||
|
||||
with patch.object(SlowLogCollector, '_logs', {1: entries}):
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/dashboard/top",
|
||||
headers=admin_headers,
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["total_instances"] == 1
|
||||
assert len(data["top_queries"]) <= 10 # TOP 10
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_dba_can_access_slow_logs(
|
||||
self, client: AsyncClient, dba_headers: dict, test_instance
|
||||
):
|
||||
"""测试DBA可以查看慢日志。"""
|
||||
with patch.object(SlowLogCollector, '_logs', {}):
|
||||
resp = await client.get(
|
||||
f"/api/v1/slow-logs/instances/{test_instance.id}",
|
||||
headers=dba_headers,
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_dba_can_collect_slow_logs(
|
||||
self, client: AsyncClient, dba_headers: dict, test_instance
|
||||
):
|
||||
"""测试DBA可以收集慢日志。"""
|
||||
# mock collect方法避免真实数据库连接
|
||||
mock_entries = [
|
||||
SlowLogEntry(
|
||||
sql_digest="d1", sql_text="SELECT 1",
|
||||
execution_count=1, avg_time_ms=100,
|
||||
max_time_ms=100, min_time_ms=100,
|
||||
rows_examined=1, rows_sent=1,
|
||||
)
|
||||
]
|
||||
with patch.object(
|
||||
SlowLogCollector, 'collect_mysql_slow_logs',
|
||||
return_value=mock_entries
|
||||
):
|
||||
resp = await client.post(
|
||||
f"/api/v1/slow-logs/instances/{test_instance.id}/collect",
|
||||
headers=dba_headers,
|
||||
)
|
||||
# 可能500因为save方法,但至少不被403拒绝
|
||||
assert resp.status_code in (200, 500)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_dashboard_any_authenticated_user(
|
||||
self, client: AsyncClient, developer_headers: dict
|
||||
):
|
||||
"""测试仪表盘TOP查询对所有登录用户开放。"""
|
||||
with patch.object(SlowLogCollector, '_logs', {}):
|
||||
resp = await client.get(
|
||||
"/api/v1/slow-logs/dashboard/top",
|
||||
headers=developer_headers,
|
||||
)
|
||||
# dashboard/top 使用 get_current_user(不是 require_role)
|
||||
assert resp.status_code == 200
|
||||
Reference in new issue
Block a user