backend/tests/test_api_keys.py

"""Tests for API key management and RBAC."""
import pytest
from unittest.mock import patch


@pytest.fixture(autouse=True)
def temp_keys_file(tmp_path):
    temp = tmp_path / "test_api_keys.jsonl"
    with patch("backend.services.api_key_manager.KEYS_PATH", temp):
        yield temp


def test_create_key_returns_raw_key():
    from backend.services.api_key_manager import create_key
    result = create_key("Test Key", role="analyst")
    assert result["key"].startswith("vfx_")
    assert "warning" in result
    assert result["role"] == "analyst"

def test_create_key_requires_name():
    from backend.services.api_key_manager import create_key
    assert "error" in create_key("")

def test_create_key_invalid_role():
    from backend.services.api_key_manager import create_key
    assert "error" in create_key("Test", role="superuser")

def test_verify_valid_key():
    from backend.services.api_key_manager import create_key, verify_key
    created = create_key("Verify Test")
    result  = verify_key(created["key"])
    assert result is not None
    assert result["role"] == "analyst"
    assert "key_hash" not in result

def test_verify_invalid_key():
    from backend.services.api_key_manager import verify_key
    assert verify_key("vfx_notvalid") is None
    assert verify_key("wrong_prefix") is None
    assert verify_key("") is None

def test_verify_increments_use_count():
    from backend.services.api_key_manager import create_key, verify_key
    created = create_key("Counter Test")
    for _ in range(3):
        verify_key(created["key"])
    result = verify_key(created["key"])
    assert result["use_count"] >= 4

def test_revoke_key():
    from backend.services.api_key_manager import create_key, revoke_key, verify_key
    created = create_key("Revoke Test")
    revoke_key(created["key_id"])
    assert verify_key(created["key"]) is None

def test_list_keys():
    from backend.services.api_key_manager import create_key, list_keys
    create_key("Key A", role="viewer")
    create_key("Key B", role="analyst")
    keys = list_keys()
    assert len(keys) == 2
    for k in keys:
        assert "key_hash" not in k

def test_list_keys_excludes_revoked():
    from backend.services.api_key_manager import create_key, revoke_key, list_keys
    c1 = create_key("Active")
    c2 = create_key("Revoked")
    revoke_key(c2["key_id"])
    active = list_keys(include_inactive=False)
    assert len(active) == 1
    assert active[0]["key_id"] == c1["key_id"]

def test_check_permission_viewer():
    from backend.services.api_key_manager import check_permission
    assert check_permission("viewer", "GET", "/api/v1/history") is True
    assert check_permission("viewer", "POST", "/api/v1/analyze/image") is False

def test_check_permission_analyst():
    from backend.services.api_key_manager import check_permission
    assert check_permission("analyst", "POST", "/api/v1/analyze/image") is True
    assert check_permission("analyst", "POST", "/api/v1/keys/") is False

def test_check_permission_admin():
    from backend.services.api_key_manager import check_permission
    assert check_permission("admin", "POST", "/api/v1/keys/") is True
    assert check_permission("admin", "DELETE", "/api/v1/keys/abc") is True

def test_api_create_key_without_auth(client):
    response = client.post("/api/v1/keys/", json={"name": "No Auth", "role": "analyst"})
    assert response.status_code == 401

def test_api_verify_endpoint_valid(client):
    from backend.services.api_key_manager import create_key
    created = create_key("API Test", role="analyst")
    response = client.get("/api/v1/keys/verify",
                          headers={"Authorization": f"Bearer {created['key']}"})
    assert response.status_code == 200
    assert response.json()["valid"] is True
    assert response.json()["role"] == "analyst"

def test_api_verify_endpoint_invalid(client):
    response = client.get("/api/v1/keys/verify",
                          headers={"Authorization": "Bearer vfx_invalid"})
    assert response.status_code == 401