qzl
97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import socket
|
|
import threading
|
|
import time
|
|
from pathlib import Path
|
|
|
|
from fastapi import FastAPI
|
|
from playwright.sync_api import sync_playwright
|
|
import uvicorn
|
|
|
|
from core.config.settings import Settings
|
|
from core.logging.config import configure_logging
|
|
from core.logging.middleware import (
|
|
RequestContextMiddleware,
|
|
register_exception_handlers,
|
|
)
|
|
|
|
|
|
def _read_json_lines(path: Path) -> list[dict[str, object]]:
|
|
return [json.loads(line) for line in path.read_text().splitlines() if line.strip()]
|
|
|
|
|
|
def _find_free_port() -> int:
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
|
|
sock.bind(("127.0.0.1", 0))
|
|
return sock.getsockname()[1]
|
|
|
|
|
|
def _wait_for_port(host: str, port: int, timeout: float = 5.0) -> None:
|
|
deadline = time.time() + timeout
|
|
while time.time() < deadline:
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
|
|
if sock.connect_ex((host, port)) == 0:
|
|
return
|
|
time.sleep(0.05)
|
|
raise RuntimeError("Server did not start in time")
|
|
|
|
|
|
def _start_server(app: FastAPI, host: str, port: int):
|
|
config = uvicorn.Config(app, host=host, port=port, log_level="info")
|
|
server = uvicorn.Server(config)
|
|
thread = threading.Thread(target=server.run, daemon=True)
|
|
thread.start()
|
|
_wait_for_port(host, port)
|
|
return server, thread
|
|
|
|
|
|
def test_e2e_error_logging(tmp_path: Path) -> None:
|
|
settings = Settings()
|
|
runtime = settings.runtime.model_copy(
|
|
update={
|
|
"log_dir": str(tmp_path),
|
|
"log_error_dir": str(tmp_path / "errors"),
|
|
"log_rotation": "size",
|
|
"log_rotation_max_bytes": 2048,
|
|
}
|
|
)
|
|
configure_logging(settings.model_copy(update={"runtime": runtime}))
|
|
|
|
app = FastAPI()
|
|
app.add_middleware(RequestContextMiddleware) # type: ignore[arg-type]
|
|
register_exception_handlers(app)
|
|
|
|
@app.get("/boom")
|
|
async def boom() -> dict[str, str]:
|
|
raise RuntimeError("boom")
|
|
|
|
host = "127.0.0.1"
|
|
port = _find_free_port()
|
|
server, thread = _start_server(app, host, port)
|
|
|
|
try:
|
|
with sync_playwright() as playwright:
|
|
request_context = playwright.request.new_context(
|
|
base_url=f"http://{host}:{port}"
|
|
)
|
|
response = request_context.get(
|
|
"/boom",
|
|
headers={"X-Request-ID": "e2e-5000"},
|
|
)
|
|
assert response.status == 500
|
|
request_context.dispose()
|
|
finally:
|
|
server.should_exit = True
|
|
thread.join(timeout=5)
|
|
|
|
error_entries = _read_json_lines(Path(tmp_path) / "errors" / "app.error.log")
|
|
entry = next(
|
|
item for item in error_entries if item.get("message") == "Unhandled exception"
|
|
)
|
|
|
|
assert entry["request_id"] == "e2e-5000"
|
|
exception = str(entry["exception"])
|
|
assert "Traceback" in exception
|