mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 01:40:21 +08:00
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from fastapi import FastAPI
|
||
|
|
|
||
|
|
|
||
|
|
def _build_contract_app() -> FastAPI:
|
||
|
|
from src.api.public.claude import router as claude_router
|
||
|
|
from src.api.public.gemini import router as gemini_router
|
||
|
|
from src.api.public.openai import router as openai_router
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
app.include_router(claude_router)
|
||
|
|
app.include_router(openai_router)
|
||
|
|
app.include_router(gemini_router)
|
||
|
|
return app
|
||
|
|
|
||
|
|
|
||
|
|
def test_public_api_routes_contract_paths_and_tags() -> None:
|
||
|
|
app = _build_contract_app()
|
||
|
|
schema = app.openapi()
|
||
|
|
|
||
|
|
paths = schema.get("paths") or {}
|
||
|
|
|
||
|
|
expected = [
|
||
|
|
("/v1/messages", "post", "Claude API"),
|
||
|
|
("/v1/messages/count_tokens", "post", "Claude API"),
|
||
|
|
("/v1/chat/completions", "post", "OpenAI API"),
|
||
|
|
("/v1/responses", "post", "OpenAI API"),
|
||
|
|
("/v1beta/models/{model}:generateContent", "post", "Gemini API"),
|
||
|
|
("/v1beta/models/{model}:streamGenerateContent", "post", "Gemini API"),
|
||
|
|
("/v1/models/{model}:generateContent", "post", "Gemini API"),
|
||
|
|
("/v1/models/{model}:streamGenerateContent", "post", "Gemini API"),
|
||
|
|
]
|
||
|
|
|
||
|
|
for path, method, expected_tag in expected:
|
||
|
|
assert path in paths, f"missing path {path}"
|
||
|
|
operations = paths.get(path) or {}
|
||
|
|
assert method in operations, f"missing {method.upper()} {path}"
|
||
|
|
tags = operations.get(method, {}).get("tags") or []
|
||
|
|
assert expected_tag in tags, f"{method.upper()} {path} missing tag {expected_tag}"
|