File size: 1,962 Bytes
9b5157d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 | """
Pytest fixtures for the photo verification API.
Uses dependency_overrides so tests run without loading 300MB of model weights.
"""
import numpy as np
import pytest
from fastapi.testclient import TestClient
from app.main import create_app
from app.services.face_analysis import FaceAnalysisService, FaceAnalysis
from app.services.liveness import LivenessService, LivenessResult
from app.core.dependencies import run_in_cv_executor
class _MockFaceService(FaceAnalysisService):
def __init__(self, result: FaceAnalysis):
super().__init__()
self._loaded = True
self._result = result
def analyze(self, image_bgr):
return self._result
class _MockLivenessService(LivenessService):
def __init__(self, result: LivenessResult):
super().__init__()
self._loaded = True
self._result = result
def check(self, image_bgr, face_bbox=None):
return self._result
@pytest.fixture
def verified_face():
return FaceAnalysis(
face_detected=True,
yaw=-20.0,
pitch=0.0,
roll=0.0,
smile_score=0.0,
face_bbox=(100, 100, 200, 200),
)
@pytest.fixture
def live_result():
return LivenessResult(is_live=True, score=0.9, is_real_prob=0.9, is_spoof_prob=0.1)
@pytest.fixture
def client(verified_face, live_result, monkeypatch):
import app.services.face_analysis as fa_module
import app.services.liveness as lv_module
monkeypatch.setattr(fa_module, "face_analysis_service", _MockFaceService(verified_face))
monkeypatch.setattr(lv_module, "liveness_service", _MockLivenessService(live_result))
# Make run_in_cv_executor run synchronously in tests
import app.core.dependencies as dep_module
import asyncio
async def sync_executor(fn, *args):
return fn(*args)
monkeypatch.setattr(dep_module, "run_in_cv_executor", sync_executor)
app_instance = create_app()
return TestClient(app_instance)
|