feat: Ipad app features and Dream Weaver for Velocity WebOS
This commit is contained in:
40
backend/tests/test_migrations_and_observability.py
Normal file
40
backend/tests/test_migrations_and_observability.py
Normal file
@@ -0,0 +1,40 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
from fastapi import FastAPI
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
os.environ.setdefault("VELOCITY_JWT_SECRET", "test-secret")
|
||||
|
||||
from backend.migrations.runner import discover_migrations
|
||||
from backend.observability import RequestObservabilityMiddleware
|
||||
|
||||
|
||||
def test_migration_discovery_is_ordered_and_checksummed() -> None:
|
||||
migrations = discover_migrations(Path("backend/migrations/versions"))
|
||||
|
||||
assert migrations
|
||||
assert migrations == sorted(migrations, key=lambda migration: migration.version)
|
||||
assert all(len(migration.checksum) == 64 for migration in migrations)
|
||||
assert len({migration.version for migration in migrations}) == len(migrations)
|
||||
|
||||
|
||||
def test_observability_middleware_adds_request_headers_and_snapshot() -> None:
|
||||
app = FastAPI()
|
||||
app.add_middleware(RequestObservabilityMiddleware)
|
||||
|
||||
@app.get("/ping")
|
||||
async def ping() -> dict[str, str]:
|
||||
return {"status": "ok"}
|
||||
|
||||
client = TestClient(app)
|
||||
response = client.get("/ping", headers={"X-Request-ID": "req-test"})
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.headers["X-Request-ID"] == "req-test"
|
||||
assert "X-Response-Time-Ms" in response.headers
|
||||
assert app.state.request_metrics[-1].request_id == "req-test"
|
||||
assert app.state.request_metrics[-1].path == "/ping"
|
||||
|
||||
Reference in New Issue
Block a user