62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from unittest import TestCase
|
|
from unittest.mock import patch
|
|
|
|
from fastapi import APIRouter
|
|
from fastapi.testclient import TestClient
|
|
from sqlalchemy import text
|
|
|
|
from govoplan_core.core.registry import PlatformRegistry
|
|
from govoplan_core.db.query_metrics import collect_query_metrics
|
|
from govoplan_core.db.session import create_database_engine
|
|
from govoplan_core.server.fastapi import create_govoplan_app
|
|
|
|
|
|
class QueryMetricsTests(TestCase):
|
|
def test_collect_query_metrics_counts_instrumented_engine_queries(self) -> None:
|
|
engine = create_database_engine("sqlite:///:memory:")
|
|
try:
|
|
with collect_query_metrics() as metrics:
|
|
with engine.connect() as connection:
|
|
self.assertEqual(1, connection.execute(text("select 1")).scalar_one())
|
|
|
|
self.assertEqual(1, metrics.query_count)
|
|
self.assertGreaterEqual(metrics.total_ms, 0.0)
|
|
self.assertGreaterEqual(metrics.slowest_ms, 0.0)
|
|
self.assertEqual(0, metrics.error_count)
|
|
finally:
|
|
engine.dispose()
|
|
|
|
def test_slow_request_log_includes_query_metrics(self) -> None:
|
|
engine = create_database_engine("sqlite:///:memory:")
|
|
router = APIRouter()
|
|
|
|
@router.get("/query")
|
|
def query_route() -> dict[str, bool]:
|
|
with engine.connect() as connection:
|
|
connection.execute(text("select 1")).scalar_one()
|
|
return {"ok": True}
|
|
|
|
try:
|
|
with patch.dict(os.environ, {"GOVOPLAN_SLOW_REQUEST_MS": "0.001"}):
|
|
app = create_govoplan_app(
|
|
title="query metrics test",
|
|
version="0",
|
|
registry=PlatformRegistry(),
|
|
api_router=router,
|
|
)
|
|
|
|
with TestClient(app) as client, self.assertLogs("govoplan.request", level="WARNING") as logs:
|
|
response = client.get("/query")
|
|
|
|
self.assertEqual(200, response.status_code, response.text)
|
|
output = "\n".join(logs.output)
|
|
self.assertIn("db_query_count=1", output)
|
|
self.assertIn("db_time_ms=", output)
|
|
self.assertIn("db_slowest_ms=", output)
|
|
self.assertIn("db_error_count=0", output)
|
|
finally:
|
|
engine.dispose()
|