mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 09:50:21 +08:00
46737d32f8
- 新增 StatusSnapshot 模型，聚合 OAuth / 账号 / 配额三维状态 - 新增 StatusSnapshotStore 负责快照的持久化与查询 - 重构 response_builder / endpoint_models，基于 snapshot 输出状态字段 - 前端抽取 providerKeyStatus / oauthRefreshFeedback 工具函数， 统一 PoolManagement、ProviderDetailDrawer、BatchDialog 的状态展示 - errorParser 增加已知 OAuth 错误的友好提示 - refresher 适配 snapshot 写入，account_state 扩展状态分类 - 新增 alembic 迁移及存量数据回填脚本 - 补充前后端单元测试
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from src.models.database import Provider, ProviderAPIKey, _provider_api_key_before_insert
|
|
from src.services.provider_keys import status_snapshot_store as module
|
|
|
|
|
|
def test_provider_api_key_before_insert_populates_status_snapshot(
|
|
monkeypatch: "pytest.MonkeyPatch",
|
|
) -> None:
|
|
provider = Provider(
|
|
id="provider-1",
|
|
name="Codex Pool",
|
|
provider_type="codex",
|
|
)
|
|
key = ProviderAPIKey(
|
|
id="key-1",
|
|
provider_id="provider-1",
|
|
provider=provider, # type: ignore[arg-type]
|
|
api_key="enc-access-token",
|
|
auth_type="oauth",
|
|
auth_config='{"expires_at":2100000000}',
|
|
name="codex-user",
|
|
oauth_invalid_reason=(
|
|
"[ACCOUNT_BLOCK] 工作区已停用 (deactivated_workspace)\n"
|
|
"[REFRESH_FAILED] refresh_token_reused"
|
|
),
|
|
)
|
|
|
|
monkeypatch.setattr(module.crypto_service, "decrypt", lambda value: value)
|
|
|
|
_provider_api_key_before_insert(None, None, key)
|
|
|
|
assert isinstance(key.status_snapshot, dict)
|
|
assert key.status_snapshot["oauth"]["code"] == "invalid"
|
|
assert key.status_snapshot["oauth"]["label"] == "已失效"
|
|
assert key.status_snapshot["oauth"]["reason"] == "refresh_token_reused"
|
|
assert key.status_snapshot["account"]["code"] == "workspace_deactivated"
|
|
assert key.status_snapshot["account"]["blocked"] is True
|
|
|
|
|
|
def test_resolve_provider_key_status_snapshot_prefers_persisted_snapshot_layers(
|
|
monkeypatch: "pytest.MonkeyPatch",
|
|
) -> None:
|
|
provider = Provider(
|
|
id="provider-1",
|
|
name="Codex Pool",
|
|
provider_type="codex",
|
|
)
|
|
key = ProviderAPIKey(
|
|
id="key-2",
|
|
provider_id="provider-1",
|
|
provider=provider, # type: ignore[arg-type]
|
|
api_key="enc-access-token",
|
|
auth_type="oauth",
|
|
auth_config='{"expires_at":100}',
|
|
name="codex-user",
|
|
upstream_metadata=None,
|
|
oauth_invalid_reason=None,
|
|
status_snapshot={
|
|
"oauth": {
|
|
"code": "valid",
|
|
"label": "有效",
|
|
"expires_at": 100,
|
|
},
|
|
"account": {
|
|
"code": "workspace_deactivated",
|
|
"label": "工作区停用",
|
|
"reason": "persisted",
|
|
"blocked": True,
|
|
"source": "persisted",
|
|
},
|
|
"quota": {
|
|
"code": "exhausted",
|
|
"label": "额度耗尽",
|
|
"reason": "persisted quota",
|
|
"exhausted": True,
|
|
"usage_ratio": 1.0,
|
|
},
|
|
},
|
|
)
|
|
|
|
monkeypatch.setattr(module.crypto_service, "decrypt", lambda value: value)
|
|
|
|
snapshot = module.resolve_provider_key_status_snapshot(
|
|
key,
|
|
now_ts=200,
|
|
)
|
|
|
|
assert snapshot.oauth.code == "expired"
|
|
assert snapshot.oauth.label == "已过期"
|
|
assert snapshot.account.code == "workspace_deactivated"
|
|
assert snapshot.account.reason == "persisted"
|
|
assert snapshot.quota.code == "exhausted"
|
|
assert snapshot.quota.reason == "persisted quota"
|