mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-01 17:00:21 +08:00
feat: Provider 异步删除、可配置密码策略、Hub 超时优化及多项改进
- 新增 Provider 异步删除任务系统,后台分阶段删除子资源并清理残留引用 - 新增可配置密码策略等级(weak/medium/strong),支持系统设置面板调整 - aether-hub 升级至 0.1.4,idle timeout 支持禁用(设为 0),worker 默认超时调整为 120s - OAuth 手动续期增加 Redis 分布式锁,防止并发刷新冲突 - ProxyNode 心跳检测改为 asyncio.to_thread,避免阻塞事件循环 - 删除 ModelMultiSelect 和 useInvalidModels,MultiSelect 组件通用化 - 明确 allowed_providers/allowed_api_formats 的 NULL 与空数组语义 - 前端 StandaloneKeyFormDialog、UserFormDialog 等多处 UI 优化 - 新增 Alembic 迁移脚本清理 Provider 删除后的残留引用 - 补充相关测试用例
This commit is contained in:
@@ -140,8 +140,14 @@ def test_sync_delete_reports_progress_after_each_batch(
|
||||
|
||||
session = _FakeSession()
|
||||
progress_updates: list[int] = []
|
||||
cleanup_batches: list[list[str]] = []
|
||||
|
||||
monkeypatch.setattr(taskmod, "_CLEANUP_BATCH_SIZE", 2)
|
||||
monkeypatch.setattr(
|
||||
taskmod,
|
||||
"cleanup_key_references",
|
||||
lambda _db, batch: cleanup_batches.append(list(batch)),
|
||||
)
|
||||
monkeypatch.setattr("src.database.create_session", lambda: session)
|
||||
monkeypatch.setattr(
|
||||
"src.models.database.ProviderAPIKey",
|
||||
@@ -157,5 +163,6 @@ def test_sync_delete_reports_progress_after_each_batch(
|
||||
|
||||
assert affected == 3
|
||||
assert progress_updates == [2, 3]
|
||||
assert cleanup_batches == [["key-1", "key-2"], ["key-3"]]
|
||||
assert session.commits == 2
|
||||
assert session.closed is True
|
||||
|
||||
109
tests/services/test_provider_auth_detached.py
Normal file
109
tests/services/test_provider_auth_detached.py
Normal file
@@ -0,0 +1,109 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import sys
|
||||
import types
|
||||
from datetime import datetime, timezone
|
||||
from types import SimpleNamespace
|
||||
from typing import Any
|
||||
|
||||
import pytest
|
||||
|
||||
from src.services.provider import auth as module
|
||||
|
||||
|
||||
class _FakeQuery:
|
||||
def __init__(self, row: Any | None) -> None:
|
||||
self._row = row
|
||||
|
||||
def filter(self, *_args: Any, **_kwargs: Any) -> "_FakeQuery":
|
||||
return self
|
||||
|
||||
def first(self) -> Any | None:
|
||||
return self._row
|
||||
|
||||
|
||||
class _FakeDB:
|
||||
def __init__(self, row: Any | None) -> None:
|
||||
self.row = row
|
||||
self.committed = False
|
||||
|
||||
def query(self, _model: Any) -> _FakeQuery:
|
||||
return _FakeQuery(self.row)
|
||||
|
||||
def commit(self) -> None:
|
||||
self.committed = True
|
||||
|
||||
|
||||
class _FakeSessionCtx:
|
||||
def __init__(self, db: _FakeDB) -> None:
|
||||
self.db = db
|
||||
|
||||
def __enter__(self) -> _FakeDB:
|
||||
return self.db
|
||||
|
||||
def __exit__(self, exc_type: Any, exc: Any, tb: Any) -> bool:
|
||||
_ = exc_type, exc, tb
|
||||
return False
|
||||
|
||||
|
||||
def _install_module(monkeypatch: pytest.MonkeyPatch, name: str, attrs: dict[str, Any]) -> None:
|
||||
fake_module = types.ModuleType(name)
|
||||
for key, value in attrs.items():
|
||||
setattr(fake_module, key, value)
|
||||
monkeypatch.setitem(sys.modules, name, fake_module)
|
||||
|
||||
|
||||
def test_persist_refreshed_token_detached_key_does_not_raise(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
key = SimpleNamespace(
|
||||
id="key-1",
|
||||
api_key="old-api",
|
||||
auth_config="old-config",
|
||||
oauth_invalid_at=datetime.now(timezone.utc),
|
||||
oauth_invalid_reason="[REFRESH_FAILED] stale",
|
||||
)
|
||||
|
||||
monkeypatch.setattr(
|
||||
module, "object_session", lambda _key: (_ for _ in ()).throw(RuntimeError())
|
||||
)
|
||||
monkeypatch.setattr(module.crypto_service, "encrypt", lambda value: f"enc:{value}")
|
||||
|
||||
module._persist_refreshed_token(key, "new-token", {"refresh_token": "rt-2"})
|
||||
|
||||
assert key.api_key == "enc:new-token"
|
||||
assert key.auth_config == 'enc:{"refresh_token": "rt-2"}'
|
||||
assert key.oauth_invalid_at is None
|
||||
assert key.oauth_invalid_reason is None
|
||||
|
||||
|
||||
def test_mark_refresh_token_invalid_persists_detached_key(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
key = SimpleNamespace(id="key-1")
|
||||
row = SimpleNamespace(id="key-1", oauth_invalid_at=None, oauth_invalid_reason=None)
|
||||
fake_db = _FakeDB(row)
|
||||
|
||||
monkeypatch.setattr(
|
||||
module, "object_session", lambda _key: (_ for _ in ()).throw(RuntimeError())
|
||||
)
|
||||
_install_module(
|
||||
monkeypatch, "src.database", {"create_session": lambda: _FakeSessionCtx(fake_db)}
|
||||
)
|
||||
_install_module(
|
||||
monkeypatch,
|
||||
"src.models.database",
|
||||
{"ProviderAPIKey": type("ProviderAPIKey", (), {"id": "id"})},
|
||||
)
|
||||
|
||||
module._mark_refresh_token_invalid(
|
||||
key,
|
||||
401,
|
||||
'{"error": {"code": "refresh_token_reused", "message": "used"}}',
|
||||
)
|
||||
|
||||
assert fake_db.committed is True
|
||||
assert key.oauth_invalid_at is not None
|
||||
assert row.oauth_invalid_at is not None
|
||||
assert str(key.oauth_invalid_reason).startswith("[REFRESH_FAILED] Token 续期失败 (401)")
|
||||
assert "refresh_token_reused" in str(row.oauth_invalid_reason)
|
||||
159
tests/services/test_provider_delete_cleanup.py
Normal file
159
tests/services/test_provider_delete_cleanup.py
Normal file
@@ -0,0 +1,159 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
import pytest
|
||||
|
||||
from src.services.provider import delete_cleanup as cleanup_module
|
||||
from src.services.provider.delete_cleanup import (
|
||||
delete_provider_tree,
|
||||
prune_allowed_provider_list,
|
||||
prune_allowed_provider_refs,
|
||||
)
|
||||
|
||||
|
||||
def _query_mock(
|
||||
*,
|
||||
rows: list[object] | None = None,
|
||||
update_count: int | None = None,
|
||||
delete_count: int | None = None,
|
||||
) -> MagicMock:
|
||||
query = MagicMock()
|
||||
filtered = query.filter.return_value
|
||||
if rows is not None:
|
||||
filtered.all.return_value = rows
|
||||
if update_count is not None:
|
||||
filtered.update.return_value = update_count
|
||||
if delete_count is not None:
|
||||
filtered.delete.return_value = delete_count
|
||||
return query
|
||||
|
||||
|
||||
def test_prune_allowed_provider_list_removes_target_and_normalizes_empty() -> None:
|
||||
next_allowed, changed = prune_allowed_provider_list(["provider-a", "provider-b"], "provider-a")
|
||||
|
||||
assert changed is True
|
||||
assert next_allowed == ["provider-b"]
|
||||
|
||||
next_allowed, changed = prune_allowed_provider_list(["provider-a"], "provider-a")
|
||||
|
||||
assert changed is True
|
||||
assert next_allowed == []
|
||||
|
||||
|
||||
def test_prune_allowed_provider_refs_updates_matching_records_only() -> None:
|
||||
records = [
|
||||
SimpleNamespace(allowed_providers=["provider-a", "provider-b"]),
|
||||
SimpleNamespace(allowed_providers=["provider-b"]),
|
||||
SimpleNamespace(allowed_providers=None),
|
||||
]
|
||||
|
||||
updated = prune_allowed_provider_refs(records, "provider-a")
|
||||
|
||||
assert updated == 1
|
||||
assert records[0].allowed_providers == ["provider-b"]
|
||||
assert records[1].allowed_providers == ["provider-b"]
|
||||
assert records[2].allowed_providers is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_cleanup_deleted_provider_references_cleans_large_fanout_tables(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
db = MagicMock()
|
||||
|
||||
user_records = [SimpleNamespace(allowed_providers=["provider-a", "provider-b"])]
|
||||
api_key_records = [SimpleNamespace(allowed_providers=["provider-a"])]
|
||||
key_cleanup_calls: list[list[str]] = []
|
||||
|
||||
monkeypatch.setattr(
|
||||
cleanup_module,
|
||||
"cleanup_key_references",
|
||||
lambda _db, key_ids: key_cleanup_calls.append(list(key_ids)),
|
||||
)
|
||||
|
||||
db.query.side_effect = [
|
||||
_query_mock(rows=[("endpoint-1",), ("endpoint-2",)]),
|
||||
_query_mock(rows=[("key-1",), ("key-2",)]),
|
||||
_query_mock(rows=user_records),
|
||||
_query_mock(rows=api_key_records),
|
||||
_query_mock(update_count=2),
|
||||
_query_mock(update_count=3),
|
||||
_query_mock(update_count=4),
|
||||
_query_mock(update_count=5),
|
||||
_query_mock(update_count=6),
|
||||
_query_mock(delete_count=7),
|
||||
_query_mock(delete_count=8),
|
||||
]
|
||||
|
||||
stats = cleanup_module.cleanup_deleted_provider_references(db, "provider-a")
|
||||
|
||||
assert stats == {
|
||||
"users": 1,
|
||||
"api_keys": 1,
|
||||
"user_preferences": 2,
|
||||
"usage_provider": 3,
|
||||
"usage_endpoint": 5,
|
||||
"video_tasks_provider": 4,
|
||||
"video_tasks_endpoint": 6,
|
||||
"request_candidates_provider": 8,
|
||||
"request_candidates_endpoint": 7,
|
||||
}
|
||||
assert user_records[0].allowed_providers == ["provider-b"]
|
||||
assert api_key_records[0].allowed_providers == []
|
||||
assert key_cleanup_calls == [["key-1", "key-2"]]
|
||||
|
||||
|
||||
def test_delete_provider_tree_deletes_children_before_provider(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
db = MagicMock()
|
||||
|
||||
cleanup_mock = MagicMock(
|
||||
return_value={
|
||||
"users": 1,
|
||||
"api_keys": 2,
|
||||
"user_preferences": 3,
|
||||
"usage_provider": 4,
|
||||
"usage_endpoint": 5,
|
||||
"video_tasks_provider": 6,
|
||||
"video_tasks_endpoint": 7,
|
||||
"request_candidates_provider": 8,
|
||||
"request_candidates_endpoint": 9,
|
||||
}
|
||||
)
|
||||
monkeypatch.setattr(cleanup_module, "cleanup_deleted_provider_references", cleanup_mock)
|
||||
|
||||
db.query.side_effect = [
|
||||
_query_mock(rows=[("endpoint-1",), ("endpoint-2",)]),
|
||||
_query_mock(rows=[("key-1",), ("key-2",), ("key-3",)]),
|
||||
_query_mock(delete_count=10),
|
||||
_query_mock(delete_count=11),
|
||||
_query_mock(delete_count=12),
|
||||
_query_mock(delete_count=13),
|
||||
_query_mock(delete_count=14),
|
||||
_query_mock(delete_count=1),
|
||||
]
|
||||
|
||||
stats = delete_provider_tree(db, "provider-a")
|
||||
|
||||
cleanup_mock.assert_called_once_with(
|
||||
db,
|
||||
"provider-a",
|
||||
endpoint_ids=["endpoint-1", "endpoint-2"],
|
||||
key_ids=["key-1", "key-2", "key-3"],
|
||||
)
|
||||
assert stats == {
|
||||
"cleanup": cleanup_mock.return_value,
|
||||
"deleted": {
|
||||
"api_key_mappings": 10,
|
||||
"usage_tracking": 11,
|
||||
"models": 12,
|
||||
"api_keys": 13,
|
||||
"endpoints": 14,
|
||||
"providers": 1,
|
||||
},
|
||||
"key_count": 3,
|
||||
"endpoint_count": 2,
|
||||
}
|
||||
151
tests/services/test_provider_delete_task.py
Normal file
151
tests/services/test_provider_delete_task.py
Normal file
@@ -0,0 +1,151 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
from collections.abc import Coroutine
|
||||
from concurrent.futures import Future
|
||||
from unittest.mock import AsyncMock
|
||||
|
||||
import pytest
|
||||
|
||||
from src.services.provider import delete_task as taskmod
|
||||
|
||||
|
||||
class _FakeRedis:
|
||||
def __init__(self) -> None:
|
||||
self.store: dict[str, str] = {}
|
||||
|
||||
async def get(self, key: str) -> str | None:
|
||||
return self.store.get(key)
|
||||
|
||||
async def setex(self, key: str, _ttl: int, value: str) -> None:
|
||||
self.store[key] = value
|
||||
|
||||
|
||||
async def _fake_get_redis_client(*, require_redis: bool = False) -> _FakeRedis:
|
||||
_ = require_redis
|
||||
return _FakeRedis()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_provider_delete_waits_for_progress_updates_before_completion(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
release_progress = asyncio.Event()
|
||||
updates: list[dict[str, object]] = []
|
||||
|
||||
async def fake_update_task_field(
|
||||
task_id: str,
|
||||
r: object | None = None,
|
||||
**fields: object,
|
||||
) -> None:
|
||||
_ = task_id, r
|
||||
updates.append(dict(fields))
|
||||
|
||||
def fake_sync_delete_provider(
|
||||
provider_id: str,
|
||||
progress_callback: object | None = None,
|
||||
) -> dict[str, object]:
|
||||
_ = provider_id
|
||||
assert callable(progress_callback)
|
||||
progress_callback(
|
||||
{
|
||||
"stage": "deleting_keys",
|
||||
"total_keys": 4,
|
||||
"deleted_keys": 2,
|
||||
"message": "deleted key batch 1/2",
|
||||
}
|
||||
)
|
||||
return {
|
||||
"total_keys": 4,
|
||||
"deleted_keys": 4,
|
||||
"total_endpoints": 2,
|
||||
"deleted_endpoints": 2,
|
||||
"elapsed_seconds": 1.2,
|
||||
}
|
||||
|
||||
def fake_run_coroutine_threadsafe(
|
||||
coro: Coroutine[object, object, object],
|
||||
loop: asyncio.AbstractEventLoop,
|
||||
) -> Future[object]:
|
||||
future: Future[object] = Future()
|
||||
|
||||
async def runner() -> None:
|
||||
await release_progress.wait()
|
||||
try:
|
||||
result = await coro
|
||||
except Exception as exc:
|
||||
future.set_exception(exc)
|
||||
else:
|
||||
future.set_result(result)
|
||||
|
||||
loop.call_soon_threadsafe(lambda: asyncio.create_task(runner()))
|
||||
return future
|
||||
|
||||
monkeypatch.setattr(taskmod, "_update_task_field", fake_update_task_field)
|
||||
monkeypatch.setattr(taskmod, "_sync_delete_provider", fake_sync_delete_provider)
|
||||
monkeypatch.setattr(taskmod.asyncio, "run_coroutine_threadsafe", fake_run_coroutine_threadsafe)
|
||||
monkeypatch.setattr(taskmod, "get_redis_client", _fake_get_redis_client)
|
||||
monkeypatch.setattr(taskmod, "invalidate_models_list_cache", AsyncMock())
|
||||
monkeypatch.setattr(taskmod.ModelCacheService, "invalidate_all_resolve_cache", AsyncMock())
|
||||
monkeypatch.setattr(taskmod.ProviderCacheService, "invalidate_provider_cache", AsyncMock())
|
||||
|
||||
task = asyncio.create_task(taskmod._run_provider_delete("task-1", "provider-1"))
|
||||
|
||||
await asyncio.sleep(0.05)
|
||||
|
||||
assert not task.done()
|
||||
assert updates == [
|
||||
{"status": taskmod.STATUS_RUNNING, "stage": "queued", "message": "delete task started"}
|
||||
]
|
||||
|
||||
release_progress.set()
|
||||
await task
|
||||
|
||||
assert updates == [
|
||||
{"status": taskmod.STATUS_RUNNING, "stage": "queued", "message": "delete task started"},
|
||||
{
|
||||
"stage": "deleting_keys",
|
||||
"total_keys": 4,
|
||||
"deleted_keys": 2,
|
||||
"message": "deleted key batch 1/2",
|
||||
},
|
||||
{
|
||||
"status": taskmod.STATUS_COMPLETED,
|
||||
"stage": "completed",
|
||||
"total_keys": 4,
|
||||
"deleted_keys": 4,
|
||||
"total_endpoints": 2,
|
||||
"deleted_endpoints": 2,
|
||||
"message": "provider deleted: keys=4, endpoints=2",
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_submit_provider_delete_reuses_running_task(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
redis = _FakeRedis()
|
||||
started = asyncio.Event()
|
||||
finish = asyncio.Event()
|
||||
|
||||
async def fake_get_redis(*, require_redis: bool = False) -> _FakeRedis:
|
||||
_ = require_redis
|
||||
return redis
|
||||
|
||||
async def fake_run_provider_delete(task_id: str, provider_id: str) -> None:
|
||||
_ = task_id, provider_id
|
||||
started.set()
|
||||
await finish.wait()
|
||||
|
||||
monkeypatch.setattr(taskmod, "get_redis_client", fake_get_redis)
|
||||
monkeypatch.setattr(taskmod, "_run_provider_delete", fake_run_provider_delete)
|
||||
|
||||
task_id_1 = await taskmod.submit_provider_delete("provider-1")
|
||||
await started.wait()
|
||||
task_id_2 = await taskmod.submit_provider_delete("provider-1")
|
||||
|
||||
assert task_id_2 == task_id_1
|
||||
|
||||
finish.set()
|
||||
await asyncio.sleep(0)
|
||||
@@ -57,6 +57,18 @@ class _FakeClearOAuthDB:
|
||||
self.commit_count += 1
|
||||
|
||||
|
||||
class _FakeQueryAll:
|
||||
def __init__(self, rows: list[Any]) -> None:
|
||||
self._rows = rows
|
||||
|
||||
def filter(self, *args: Any, **kwargs: Any) -> _FakeQueryAll:
|
||||
_ = args, kwargs
|
||||
return self
|
||||
|
||||
def all(self) -> list[Any]:
|
||||
return self._rows
|
||||
|
||||
|
||||
def _build_key(**overrides: Any) -> SimpleNamespace:
|
||||
base: dict[str, Any] = {
|
||||
"auto_fetch_models": False,
|
||||
@@ -228,3 +240,121 @@ async def test_run_delete_key_side_effects_skip_disassociate(
|
||||
|
||||
assert captured["provider_id"] == "provider-1"
|
||||
assert captured["skip_disassociate"] is True
|
||||
|
||||
|
||||
def test_cleanup_key_references_preserves_usage_and_video_tasks(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
class _FakeDeleteStatement:
|
||||
def __init__(self, model: Any) -> None:
|
||||
self.kind = "delete"
|
||||
self.model = model
|
||||
self.conditions: list[Any] = []
|
||||
|
||||
def where(self, *conditions: Any) -> _FakeDeleteStatement:
|
||||
self.conditions.extend(conditions)
|
||||
return self
|
||||
|
||||
class _FakeUpdateStatement:
|
||||
def __init__(self, model: Any) -> None:
|
||||
self.kind = "update"
|
||||
self.model = model
|
||||
self.conditions: list[Any] = []
|
||||
self.values_dict: dict[str, Any] = {}
|
||||
|
||||
def where(self, *conditions: Any) -> _FakeUpdateStatement:
|
||||
self.conditions.extend(conditions)
|
||||
return self
|
||||
|
||||
def values(self, **values: Any) -> _FakeUpdateStatement:
|
||||
self.values_dict.update(values)
|
||||
return self
|
||||
|
||||
class _FakeDB:
|
||||
def __init__(self) -> None:
|
||||
self.statements: list[Any] = []
|
||||
|
||||
def execute(self, statement: Any) -> None:
|
||||
self.statements.append(statement)
|
||||
|
||||
def get_bind(self) -> Any:
|
||||
return SimpleNamespace(dialect=SimpleNamespace(name="postgresql"))
|
||||
|
||||
monkeypatch.setattr(side_effects_module, "sa_delete", lambda model: _FakeDeleteStatement(model))
|
||||
monkeypatch.setattr(side_effects_module, "sa_update", lambda model: _FakeUpdateStatement(model))
|
||||
|
||||
db = _FakeDB()
|
||||
side_effects_module.cleanup_key_references(cast(Any, db), ["key-1", "key-2"])
|
||||
|
||||
assert [(stmt.kind, stmt.model.__name__) for stmt in db.statements] == [
|
||||
("delete", "RequestCandidate"),
|
||||
("delete", "GeminiFileMapping"),
|
||||
("update", "Usage"),
|
||||
("update", "VideoTask"),
|
||||
]
|
||||
assert db.statements[2].values_dict == {"provider_api_key_id": None}
|
||||
assert db.statements[3].values_dict == {"key_id": None}
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_batch_delete_endpoint_keys_response_cleans_related_references(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
cleanup_calls: list[list[str]] = []
|
||||
side_effect_calls: list[str | None] = []
|
||||
|
||||
class _FakeBatchDeleteDB:
|
||||
def __init__(self, keys: list[Any]) -> None:
|
||||
self._keys = keys
|
||||
self.commit_count = 0
|
||||
self.executed: list[Any] = []
|
||||
|
||||
def query(self, model: Any) -> _FakeQueryAll:
|
||||
_ = model
|
||||
return _FakeQueryAll(self._keys)
|
||||
|
||||
def execute(self, statement: Any) -> None:
|
||||
self.executed.append(statement)
|
||||
|
||||
def commit(self) -> None:
|
||||
self.commit_count += 1
|
||||
|
||||
def rollback(self) -> None:
|
||||
raise AssertionError("rollback should not be called")
|
||||
|
||||
async def _fake_run_delete_key_side_effects(
|
||||
db: Any,
|
||||
provider_id: str | None,
|
||||
deleted_key_allowed_models: list[str] | None,
|
||||
) -> None:
|
||||
_ = db, deleted_key_allowed_models
|
||||
side_effect_calls.append(provider_id)
|
||||
|
||||
monkeypatch.setattr(
|
||||
command_module,
|
||||
"cleanup_key_references",
|
||||
lambda _db, key_ids: cleanup_calls.append(list(key_ids)),
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
command_module,
|
||||
"run_delete_key_side_effects",
|
||||
_fake_run_delete_key_side_effects,
|
||||
)
|
||||
|
||||
keys = [
|
||||
SimpleNamespace(id="key-1", provider_id="provider-1"),
|
||||
SimpleNamespace(id="key-2", provider_id="provider-1"),
|
||||
]
|
||||
db = _FakeBatchDeleteDB(keys)
|
||||
|
||||
result = await command_module.batch_delete_endpoint_keys_response(
|
||||
cast(Any, db),
|
||||
["key-1", "key-2"],
|
||||
)
|
||||
|
||||
assert result["success_count"] == 2
|
||||
assert result["failed_count"] == 0
|
||||
assert cleanup_calls == [["key-1", "key-2"]] or cleanup_calls == [["key-2", "key-1"]]
|
||||
assert side_effect_calls == ["provider-1"]
|
||||
assert db.commit_count == 1
|
||||
assert len(db.executed) == 1
|
||||
|
||||
Reference in New Issue
Block a user