mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-06 17:37:47 +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:
@@ -58,6 +58,8 @@ def _build_context(raw_body: bytes, headers: dict[str, str] | None = None) -> Ap
|
||||
api_key=None,
|
||||
request_id="req_test",
|
||||
start_time=0.0,
|
||||
request_method="POST",
|
||||
request_path="/v1/messages",
|
||||
client_ip="127.0.0.1",
|
||||
user_agent="pytest",
|
||||
original_headers=headers or {},
|
||||
@@ -67,6 +69,74 @@ def _build_context(raw_body: bytes, headers: dict[str, str] | None = None) -> Ap
|
||||
|
||||
|
||||
class TestApiRequestContextEnsureJsonBody:
|
||||
def test_build_prefers_request_state_request_id_over_trace_header(self) -> None:
|
||||
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
|
||||
request.state.request_id = "state-rid-001"
|
||||
|
||||
context = ApiRequestContext.build(
|
||||
request=request,
|
||||
db=None, # type: ignore[arg-type]
|
||||
user=None,
|
||||
api_key=None,
|
||||
raw_body=b"{}",
|
||||
)
|
||||
|
||||
assert context.request_id == "state-rid-001"
|
||||
assert request.state.request_id == "state-rid-001"
|
||||
|
||||
def test_build_prefers_trace_header_for_request_id(self) -> None:
|
||||
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
|
||||
|
||||
context = ApiRequestContext.build(
|
||||
request=request,
|
||||
db=None, # type: ignore[arg-type]
|
||||
user=None,
|
||||
api_key=None,
|
||||
raw_body=b"{}",
|
||||
)
|
||||
|
||||
assert context.request_id == "trace-frontdoor-123"
|
||||
assert request.state.request_id == "trace-frontdoor-123"
|
||||
|
||||
def test_build_snapshots_request_method_path_and_path_params(self) -> None:
|
||||
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
|
||||
request.scope["method"] = "GET"
|
||||
request.scope["path"] = "/v1beta/models/gemini-2.5-pro:generateContent"
|
||||
request.scope["raw_path"] = b"/v1beta/models/gemini-2.5-pro:generateContent"
|
||||
request.scope["path_params"] = {"model": "gemini-2.5-pro"}
|
||||
|
||||
context = ApiRequestContext.build(
|
||||
request=request,
|
||||
db=None, # type: ignore[arg-type]
|
||||
user=None,
|
||||
api_key=None,
|
||||
raw_body=b"{}",
|
||||
)
|
||||
|
||||
assert context.request_method == "GET"
|
||||
assert context.request_path == "/v1beta/models/gemini-2.5-pro:generateContent"
|
||||
assert context.path_params == {"model": "gemini-2.5-pro"}
|
||||
|
||||
def test_build_snapshots_request_runtime_state(self) -> None:
|
||||
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
|
||||
request.state.prefetched_balance_remaining = "12.5"
|
||||
request.state.gateway_execution_path = "public_proxy_after_executor_miss"
|
||||
request.state.rate_limit_scope = "user"
|
||||
request.state.tx_committed_by_route = True
|
||||
|
||||
context = ApiRequestContext.build(
|
||||
request=request,
|
||||
db=None, # type: ignore[arg-type]
|
||||
user=None,
|
||||
api_key=None,
|
||||
raw_body=b"{}",
|
||||
)
|
||||
|
||||
assert context.prefetched_balance_remaining == 12.5
|
||||
assert context.gateway_execution_path == "public_proxy_after_executor_miss"
|
||||
assert context.rate_limit_scope == "user"
|
||||
assert context.tx_committed_by_route is True
|
||||
|
||||
def test_decompresses_gzip_body(self) -> None:
|
||||
payload = {"message": "hello", "count": 2}
|
||||
raw_body = gzip.compress(json.dumps(payload).encode("utf-8"))
|
||||
@@ -88,6 +158,7 @@ class TestApiRequestContextEnsureJsonBody:
|
||||
def test_build_records_client_encoding_preferences(self) -> None:
|
||||
request = _build_request(
|
||||
headers={
|
||||
"content-type": "application/json",
|
||||
"content-encoding": "gzip",
|
||||
"accept-encoding": "gzip, deflate",
|
||||
}
|
||||
@@ -102,6 +173,31 @@ class TestApiRequestContextEnsureJsonBody:
|
||||
|
||||
assert context.client_content_encoding == "gzip"
|
||||
assert context.client_accept_encoding == "gzip, deflate"
|
||||
assert context.request_content_type == "application/json"
|
||||
|
||||
def test_build_records_perf_only_when_payload_not_empty(self) -> None:
|
||||
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
|
||||
request.state.perf_metrics = {}
|
||||
context = ApiRequestContext.build(
|
||||
request=request,
|
||||
db=None, # type: ignore[arg-type]
|
||||
user=None,
|
||||
api_key=None,
|
||||
raw_body=b"{}",
|
||||
)
|
||||
assert "perf" not in context.extra
|
||||
|
||||
request_with_perf = _build_request(headers={"x-trace-id": "trace-frontdoor-456"})
|
||||
request_with_perf.state.perf_metrics = {"pipeline": {"auth_ms": 3}}
|
||||
context_with_perf = ApiRequestContext.build(
|
||||
request=request_with_perf,
|
||||
db=None, # type: ignore[arg-type]
|
||||
user=None,
|
||||
api_key=None,
|
||||
raw_body=b"{}",
|
||||
)
|
||||
assert context_with_perf.extra["perf"] == {"pipeline": {"auth_ms": 3}}
|
||||
assert context_with_perf.perf_metrics == {"pipeline": {"auth_ms": 3}}
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_ensure_json_body_async_loads_body_lazily(self) -> None:
|
||||
|
||||
Reference in New Issue
Block a user