"""Unload GPU-resident tagger sessions via POST /api/models/unload."""

from __future__ import annotations

from fastapi.testclient import TestClient

from app.main import app
from app.storage import execute


def test_unload_models_clears_loaded_list():
    with TestClient(app) as client:
        resp = client.post("/api/models/unload")
        assert resp.status_code == 200
        body = resp.json()
        assert isinstance(body["unloaded"], list)
        assert "message" in body
        providers = client.get("/api/providers").json()
        assert providers["loaded_models"] == []


def test_unload_models_rejects_while_run_is_active():
    execute(
        """
        INSERT INTO runs (
            created_at, root_repo, categories_root, confidence_threshold,
            status, total_images, processed_images, failed_images
        ) VALUES ('2026-08-16T00:00:00+00:00', 'x', 'y', 0.45, 'running', 10, 1, 0)
        """
    )
    with TestClient(app) as client:
        resp = client.post("/api/models/unload")
        assert resp.status_code == 409
        assert "running" in resp.json()["detail"].lower()