mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 09:20:22 +08:00
7c580e843f
- 移除 token/latency Prometheus 指标的 model 标签，避免 provider x model 笛卡尔积 - HealthMonitor 滑动窗口从 DB JSON 迁移至进程内存，减少写放大 - ModelCostService 三层缓存增加 500 条上限，超限时清空 - StickyPriority 粘性缓存和健康状态字典增加容量淘汰 - AffinityManager 请求锁字典增加 500 条上限，淘汰空闲锁 - 配额刷新/探测查询使用 defer/load_only 避免加载大 JSON 列 - Alembic 迁移清理 DB 中遗留的 request_results_window 数据 - 同步更新测试适配 batch_get_cooldowns 返回值和批量删除异步化
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from src.api.admin.pool.routes import AdminBatchActionKeysAdapter
|
|
from src.api.admin.pool.schemas import BatchActionRequest
|
|
|
|
|
|
def _build_context(db: MagicMock) -> SimpleNamespace:
|
|
return SimpleNamespace(
|
|
db=db,
|
|
user=SimpleNamespace(username="admin-1"),
|
|
add_audit_metadata=lambda **_: None,
|
|
)
|
|
|
|
|
|
def _mock_provider_lookup(db: MagicMock, provider_id: str) -> None:
|
|
db.query.return_value.filter.return_value.first.return_value = SimpleNamespace(id=provider_id)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_batch_delete_submits_async_task(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""Delete action now submits an async batch-delete task instead of
|
|
executing SQL synchronously."""
|
|
db = MagicMock()
|
|
provider_id = "provider-1"
|
|
_mock_provider_lookup(db, provider_id)
|
|
|
|
mock_submit = AsyncMock(return_value="task-abc-123")
|
|
monkeypatch.setattr(
|
|
"src.services.provider_keys.batch_delete_task.submit_batch_delete",
|
|
mock_submit,
|
|
)
|
|
|
|
key_ids = [f"key-{idx}" for idx in range(1200)]
|
|
adapter = AdminBatchActionKeysAdapter(
|
|
provider_id=provider_id,
|
|
body=BatchActionRequest(
|
|
key_ids=key_ids,
|
|
action="delete",
|
|
),
|
|
)
|
|
|
|
result = await adapter.handle(_build_context(db))
|
|
|
|
# Async task returns affected=0 and a task_id
|
|
assert result.affected == 0
|
|
assert result.task_id == "task-abc-123"
|
|
assert "1200" in result.message
|
|
mock_submit.assert_awaited_once_with(provider_id, list(dict.fromkeys(key_ids)))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_batch_delete_deduplicates_key_ids(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""Duplicate key IDs should be deduplicated before submission."""
|
|
db = MagicMock()
|
|
provider_id = "provider-2"
|
|
_mock_provider_lookup(db, provider_id)
|
|
|
|
mock_submit = AsyncMock(return_value="task-xyz-456")
|
|
monkeypatch.setattr(
|
|
"src.services.provider_keys.batch_delete_task.submit_batch_delete",
|
|
mock_submit,
|
|
)
|
|
|
|
adapter = AdminBatchActionKeysAdapter(
|
|
provider_id=provider_id,
|
|
body=BatchActionRequest(
|
|
key_ids=["key-1", "key-2", "key-1", "key-3", "key-2"],
|
|
action="delete",
|
|
),
|
|
)
|
|
|
|
result = await adapter.handle(_build_context(db))
|
|
|
|
assert result.affected == 0
|
|
assert result.task_id == "task-xyz-456"
|
|
# Deduplicated: 3 unique keys
|
|
submitted_ids = mock_submit.call_args[0][1]
|
|
assert len(submitted_ids) == 3
|
|
assert submitted_ids == ["key-1", "key-2", "key-3"]
|