julian
6933406ae3
Skeleton of the host-side vision service per the packaging decision: a Python/FastAPI app at apps/vision/, uv-managed, wired into the Turbo graph via a thin package.json shim (dev/lint/test/build → uv/uvicorn/ruff/pytest). A per-package turbo.json sets build outputs [] so the no-op build is warning-free. Endpoints: GET /health (readiness + model version) and POST /analyze (raw octet-stream body, so Node POSTs Snapshot.bytes directly; empty→400, oversize→413, recognizer-not-ready→503). The recognizer is a Protocol with a StubRecognizer (no models, boots/tests offline — the dev/CI default) and a FastAlprRecognizer (the real MIT YOLOv9+CCT/ONNX stack, lazily imported; missing models ⇒ ready=False, not a crash) — the device-adapter pattern applied to the model. fast-alpr + onnxruntime are an optional `alpr` extra, so `uv sync` needs no model download. Verified: turbo run lint|test|build includes @parking/vision and stays green; uv run mypy strict-clean; uvicorn boots and serves /health + /analyze live; pnpm workspace 6→7. Not built yet: the Node VisionClient adapter, a Dockerfile + model fetch, and Job 2 (vehicle verification). Updates the packaging decision (As-scaffolded) + log. Claude-Session: https://claude.ai/code/session_01Xcm6ikLgGoCxxHrxtjkk5V
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
"""Smoke tests for the vision service in STUB mode (no model weights needed)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from vision_service.app import app
|
|
|
|
|
|
def make_client() -> TestClient:
|
|
# TestClient runs the lifespan, building the (stub) recognizer.
|
|
return TestClient(app)
|
|
|
|
|
|
def test_health_ok_in_stub_mode() -> None:
|
|
with make_client() as client:
|
|
res = client.get("/health")
|
|
assert res.status_code == 200
|
|
body = res.json()
|
|
assert body["status"] == "ok"
|
|
assert body["ready"] is True
|
|
assert body["recognizer"] == "stub"
|
|
assert body["model_version"] == "stub-0"
|
|
|
|
|
|
def test_analyze_returns_contract_shape() -> None:
|
|
with make_client() as client:
|
|
# The stub recognizes nothing, but the response must match the contract.
|
|
res = client.post(
|
|
"/analyze",
|
|
content=b"\xff\xd8\xff\xe0not-a-real-jpeg",
|
|
headers={"content-type": "application/octet-stream"},
|
|
)
|
|
assert res.status_code == 200
|
|
body = res.json()
|
|
assert body["plate"] is None
|
|
assert body["plates"] == []
|
|
assert body["vehicle"] is None
|
|
assert body["low_confidence"] is False
|
|
assert body["model_version"] == "stub-0"
|
|
assert "took_ms" in body
|
|
|
|
|
|
def test_analyze_rejects_empty_body() -> None:
|
|
with make_client() as client:
|
|
res = client.post(
|
|
"/analyze",
|
|
content=b"",
|
|
headers={"content-type": "application/octet-stream"},
|
|
)
|
|
assert res.status_code == 400
|