mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 17:30:23 +08:00
feat: 扩展 Rust gateway 全功能模块,新增 billing/crypto/wallet crate 及完整数据层
- 新增 aether-billing、aether-crypto、aether-wallet 独立 crate - aether-data 扩展 repository 层:announcements、auth_modules、billing、 candidate_selection、gemini_file_mappings、global_models、management_tokens、 oauth_providers、proxy_nodes、quota、users、wallet 等模块 - aether-gateway 新增 api/auth/billing/control/middleware/scheduler/usage/ video_tasks/hooks/maintenance/model_fetch/provider_transport 等功能模块 - 重构 executor decision 和 gateway state 为模块目录结构 - 新增 gateway router、frontdoor 路由层及对应测试 - Python 侧 API 路由重构,新增 compat/support 模块 - 前端 Logo 组件更新及 Provider 管理页面调整
This commit is contained in:
130
tests/api/public/test_models_shell.py
Normal file
130
tests/api/public/test_models_shell.py
Normal file
@@ -0,0 +1,130 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
import pytest
|
||||
from fastapi import FastAPI
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from src.database import get_db
|
||||
|
||||
|
||||
def _build_app(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
*,
|
||||
pipeline_result: Any,
|
||||
) -> tuple[TestClient, list[dict[str, Any]]]:
|
||||
from src.api.public import models as mod
|
||||
|
||||
app = FastAPI()
|
||||
app.include_router(mod.router)
|
||||
app.dependency_overrides[get_db] = lambda: MagicMock()
|
||||
|
||||
calls: list[dict[str, Any]] = []
|
||||
|
||||
async def _fake_pipeline_run(
|
||||
*,
|
||||
adapter: Any,
|
||||
http_request: object,
|
||||
db: object,
|
||||
mode: object,
|
||||
api_format_hint: str | None = None,
|
||||
path_params: dict[str, Any] | None = None,
|
||||
) -> Any:
|
||||
del http_request, db, api_format_hint, path_params
|
||||
calls.append(
|
||||
{
|
||||
"adapter_type": type(adapter).__name__,
|
||||
"mode": getattr(mode, "value", mode),
|
||||
"adapter_state": dict(getattr(adapter, "__dict__", {})),
|
||||
}
|
||||
)
|
||||
return pipeline_result
|
||||
|
||||
monkeypatch.setattr(mod.pipeline, "run", _fake_pipeline_run)
|
||||
return TestClient(app), calls
|
||||
|
||||
|
||||
def test_public_openai_models_route_is_pipeline_shell(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
client, calls = _build_app(monkeypatch, pipeline_result={"object": "list", "data": []})
|
||||
|
||||
response = client.get("/v1/models?after_id=model-a&limit=12")
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.json() == {"object": "list", "data": []}
|
||||
assert calls == [
|
||||
{
|
||||
"adapter_type": "PublicModelsListAdapter",
|
||||
"mode": "public",
|
||||
"adapter_state": {
|
||||
"before_id": None,
|
||||
"after_id": "model-a",
|
||||
"limit": 12,
|
||||
"page_size": 50,
|
||||
"page_token": None,
|
||||
},
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
def test_public_model_detail_route_is_pipeline_shell(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
client, calls = _build_app(monkeypatch, pipeline_result={"id": "gpt-5"})
|
||||
|
||||
response = client.get("/v1/models/gpt-5")
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.json() == {"id": "gpt-5"}
|
||||
assert calls == [
|
||||
{
|
||||
"adapter_type": "PublicModelDetailAdapter",
|
||||
"mode": "public",
|
||||
"adapter_state": {
|
||||
"model_id": "gpt-5",
|
||||
"force_gemini_name": False,
|
||||
},
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
def test_public_gemini_models_route_is_pipeline_shell(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
client, calls = _build_app(monkeypatch, pipeline_result={"models": []})
|
||||
|
||||
response = client.get("/v1beta/models?pageSize=25&pageToken=next-1")
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.json() == {"models": []}
|
||||
assert calls == [
|
||||
{
|
||||
"adapter_type": "PublicModelsListAdapter",
|
||||
"mode": "public",
|
||||
"adapter_state": {
|
||||
"before_id": None,
|
||||
"after_id": None,
|
||||
"limit": 20,
|
||||
"page_size": 25,
|
||||
"page_token": "next-1",
|
||||
},
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
def test_public_gemini_model_detail_route_is_pipeline_shell(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
client, calls = _build_app(monkeypatch, pipeline_result={"name": "models/gemini-2.5-pro"})
|
||||
|
||||
response = client.get("/v1beta/models/models/gemini-2.5-pro")
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.json() == {"name": "models/gemini-2.5-pro"}
|
||||
assert calls == [
|
||||
{
|
||||
"adapter_type": "PublicModelDetailAdapter",
|
||||
"mode": "public",
|
||||
"adapter_state": {
|
||||
"model_id": "models/gemini-2.5-pro",
|
||||
"force_gemini_name": True,
|
||||
},
|
||||
}
|
||||
]
|
||||
Reference in New Issue
Block a user