import pytest from backend.app.celery_app import celery from backend.app.models import TagKind from backend.app.services.tag_service import TagService pytestmark = pytest.mark.integration @pytest.fixture(autouse=True) def eager(): celery.conf.task_always_eager = True yield celery.conf.task_always_eager = False @pytest.mark.asyncio async def test_get_and_patch_settings(client): resp = await client.get("/api/ml/settings") assert resp.status_code == 200 body = await resp.get_json() assert body["head_min_positives"] == 8 # Retired tagger/suggestion-threshold columns are gone from the payload # (Camie retirement #1189/#1199). assert "suggestion_threshold_general" not in body assert "tagger_store_floor" not in body assert "tagger_model_version" not in body resp = await client.patch( "/api/ml/settings", json={"head_min_positives": 12} ) assert resp.status_code == 200 assert (await resp.get_json())["head_min_positives"] == 12 @pytest.mark.asyncio async def test_embedder_model_default_settable_and_empty_rejected(client): # #1203: fresh installs default to SigLIP 2 (migration 0069, no embeddings). # The model is operator-settable (a swap) and neither field may be blanked. body = await (await client.get("/api/ml/settings")).get_json() assert body["embedder_model_name"] == "google/siglip2-so400m-patch16-512" ok = await client.patch("/api/ml/settings", json={ "embedder_model_name": "google/siglip-so400m-patch14-384", "embedder_model_version": "siglip-so400m-patch14-384", }) assert ok.status_code == 200 out = await ok.get_json() assert out["embedder_model_name"] == "google/siglip-so400m-patch14-384" bad = await client.patch("/api/ml/settings", json={"embedder_model_name": " "}) assert bad.status_code == 400 @pytest.mark.asyncio async def test_detector_settings_defaults_patch_and_validation(client): # #134: crop-proposer config is exposed + editable here (announced to the # agent via the lease). Defaults are all-on with the pinned weights. body = await (await client.get("/api/ml/settings")).get_json() assert body["detector_person_enabled"] is True assert body["detector_anatomy_enabled"] is True assert "yolov11m_aa22" in body["detector_anatomy_weights"] assert body["detector_panel_weights"].endswith("::best.pt") assert body["detector_max_regions"] == 128 ok = await client.patch("/api/ml/settings", json={ "detector_anatomy_enabled": False, "detector_person_conf": 0.5, "detector_max_panels": 4, }) assert ok.status_code == 200 out = await ok.get_json() assert out["detector_anatomy_enabled"] is False assert out["detector_person_conf"] == 0.5 assert out["detector_max_panels"] == 4 # Out-of-range confidence + non-positive cap are rejected. assert (await client.patch( "/api/ml/settings", json={"detector_panel_conf": 1.5})).status_code == 400 assert (await client.patch( "/api/ml/settings", json={"detector_max_regions": 0})).status_code == 400 @pytest.mark.asyncio async def test_presentation_settings_defaults_patch_and_validation(client): # #141: presentation-chrome auto-hide knobs (banner/editor auto-apply + # the "also looks like content" conflict threshold) are exposed + editable. body = await (await client.get("/api/ml/settings")).get_json() assert body["presentation_auto_apply_enabled"] is True assert body["presentation_auto_apply_threshold"] == 0.90 assert body["presentation_conflict_threshold"] == 0.50 ok = await client.patch("/api/ml/settings", json={ "presentation_auto_apply_enabled": False, "presentation_auto_apply_threshold": 0.95, "presentation_conflict_threshold": 0.4, }) assert ok.status_code == 200 out = await ok.get_json() assert out["presentation_auto_apply_enabled"] is False assert out["presentation_auto_apply_threshold"] == 0.95 assert out["presentation_conflict_threshold"] == 0.4 # Auto-apply threshold below the floor + conflict cut above 1 are rejected. assert (await client.patch( "/api/ml/settings", json={"presentation_auto_apply_threshold": 0.4})).status_code == 400 assert (await client.patch( "/api/ml/settings", json={"presentation_conflict_threshold": 1.5})).status_code == 400 @pytest.mark.asyncio async def test_embedder_models_list(client): # #1203: the dropdown reads the supported-model list from the server. body = await (await client.get("/api/ml/embedder-models")).get_json() assert any( m["name"] == "google/siglip2-so400m-patch16-512" for m in body["models"] ) assert all({"name", "version", "label"} <= set(m) for m in body["models"]) @pytest.mark.asyncio async def test_video_settings_default_and_patch(client): """#747: video frame-sampling knobs are exposed + patchable.""" body = await (await client.get("/api/ml/settings")).get_json() assert body["video_frame_interval_seconds"] == pytest.approx(4.0) assert body["video_max_frames"] == 64 resp = await client.patch( "/api/ml/settings", json={"video_frame_interval_seconds": 5, "video_max_frames": 40}, ) assert resp.status_code == 200 out = await resp.get_json() assert out["video_frame_interval_seconds"] == pytest.approx(5.0) assert out["video_max_frames"] == 40 @pytest.mark.asyncio async def test_video_max_frames_below_one_rejected(client): resp = await client.patch( "/api/ml/settings", json={"video_max_frames": 0}, ) assert resp.status_code == 400 assert "video_max_frames" in (await resp.get_json())["error"] @pytest.mark.asyncio async def test_backfill_trigger(client): r1 = await client.post("/api/ml/backfill") assert r1.status_code == 202 @pytest.mark.asyncio async def test_rename_tag(client, db): tag = await TagService(db).find_or_create("oldname", TagKind.character) await db.commit() resp = await client.patch(f"/api/tags/{tag.id}", json={"name": "New Name"}) assert resp.status_code == 200 assert (await resp.get_json())["name"] == "New Name" @pytest.mark.asyncio async def test_rename_collision_409(client, db): svc = TagService(db) await svc.find_or_create("Taken", TagKind.character) other = await svc.find_or_create("Mover", TagKind.character) await db.commit() resp = await client.patch(f"/api/tags/{other.id}", json={"name": "Taken"}) assert resp.status_code == 409