131 lines
3.8 KiB
Python
131 lines
3.8 KiB
Python
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
import app.main as main
|
|
|
|
|
|
class DummySettings(SimpleNamespace):
|
|
@property
|
|
def auth_enabled(self):
|
|
return bool(self.auth_token)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def reset_settings(monkeypatch):
|
|
monkeypatch.setattr(
|
|
main,
|
|
"settings",
|
|
DummySettings(
|
|
env="test",
|
|
test_ui_enabled=True,
|
|
auth_token="",
|
|
auth_header_name="X-API-Key",
|
|
),
|
|
)
|
|
|
|
|
|
@pytest.fixture()
|
|
def client():
|
|
return TestClient(main.app)
|
|
|
|
|
|
def test_health_reflects_settings(client):
|
|
resp = client.get("/health")
|
|
assert resp.status_code == 200
|
|
assert resp.json() == {
|
|
"ok": True,
|
|
"env": "test",
|
|
"test_ui_enabled": True,
|
|
"auth_enabled": False,
|
|
"auth_header": None,
|
|
}
|
|
|
|
|
|
def test_index_shows_disabled_message_when_ui_off(monkeypatch):
|
|
monkeypatch.setattr(
|
|
main,
|
|
"settings",
|
|
DummySettings(
|
|
env="test",
|
|
test_ui_enabled=False,
|
|
auth_token="",
|
|
auth_header_name="X-API-Key",
|
|
),
|
|
)
|
|
resp = TestClient(main.app).get("/")
|
|
assert resp.status_code == 200
|
|
assert "Test UI is disabled." in resp.text
|
|
|
|
|
|
def test_focus_accepts_bearer_auth_and_returns_payload(monkeypatch, client):
|
|
monkeypatch.setattr(
|
|
main,
|
|
"settings",
|
|
DummySettings(
|
|
env="test",
|
|
test_ui_enabled=True,
|
|
auth_token="secret",
|
|
auth_header_name="X-API-Key",
|
|
),
|
|
)
|
|
|
|
def fake_process_image(image_bytes, filename, buffer_ratio=0.15, detector="subject"):
|
|
assert image_bytes == b"image-bytes"
|
|
assert filename == "sample.jpg"
|
|
assert buffer_ratio == 0.2
|
|
assert detector == "face"
|
|
return {
|
|
"filename": filename,
|
|
"detector": detector,
|
|
"method": "face_cascade",
|
|
"buffer_ratio": buffer_ratio,
|
|
"detected_bbox": {"x": 1, "y": 2, "w": 3, "h": 4},
|
|
"square_bbox": {"x": 0, "y": 0, "w": 8, "h": 8},
|
|
"source_size": {"width": 10, "height": 12},
|
|
"crop_data_url": "data:image/jpeg;base64,AAA=",
|
|
"annotated_data_url": "data:image/jpeg;base64,BBB=",
|
|
"mime_type": "image/jpeg",
|
|
"crop_bytes": b"crop",
|
|
}
|
|
|
|
monkeypatch.setattr("app.vision.process_image", fake_process_image)
|
|
|
|
resp = client.post(
|
|
"/api/focus",
|
|
headers={"Authorization": "Bearer secret"},
|
|
files={"file": ("sample.jpg", b"image-bytes", "image/jpeg")},
|
|
data={"buffer_ratio": "0.2", "detector": "face"},
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.json()["method"] == "face_cascade"
|
|
assert resp.json()["detected_bbox"] == {"x": 1, "y": 2, "w": 3, "h": 4}
|
|
|
|
|
|
def test_focus_image_returns_jpeg_stream(monkeypatch, client):
|
|
def fake_process_image(image_bytes, filename, buffer_ratio=0.15, detector="subject"):
|
|
return {
|
|
"filename": filename,
|
|
"detector": detector,
|
|
"method": "subject_contour",
|
|
"buffer_ratio": buffer_ratio,
|
|
"detected_bbox": {"x": 1, "y": 2, "w": 3, "h": 4},
|
|
"square_bbox": {"x": 0, "y": 0, "w": 8, "h": 8},
|
|
"source_size": {"width": 10, "height": 12},
|
|
"crop_data_url": "data:image/jpeg;base64,AAA=",
|
|
"annotated_data_url": "data:image/jpeg;base64,BBB=",
|
|
"mime_type": "image/jpeg",
|
|
"crop_bytes": b"crop-bytes",
|
|
}
|
|
|
|
monkeypatch.setattr("app.vision.process_image", fake_process_image)
|
|
|
|
resp = client.post(
|
|
"/api/focus/image",
|
|
files={"file": ("sample.jpg", b"image-bytes", "image/jpeg")},
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.headers["content-type"] == "image/jpeg"
|
|
assert resp.content == b"crop-bytes"
|