"""Integration tests for admin-gated routes. Covers, for each affected route, the regression matrix: - admin JWT -> success - ADMIN_API_KEY -> success (no DB user required) - AUTH_DISABLED -> success - member JWT -> 403 - no auth -> 401 """ from __future__ import annotations # --- GET /configure --- def test_get_configure_admin_jwt(client, auth_admin_header): response = client.get("/configure", headers=auth_admin_header) assert response.status_code == 200 def test_get_configure_admin_api_key(client, admin_api_key_env): response = client.get("/configure", headers=admin_api_key_env) assert response.status_code == 200 def test_get_configure_auth_disabled(client, auth_disabled_env): response = client.get("/configure") assert response.status_code == 200 def test_get_configure_member_forbidden(client, auth_member_header): response = client.get("/configure", headers=auth_member_header) assert response.status_code == 403 def test_get_configure_no_auth_unauthorized(client): response = client.get("/configure") assert response.status_code == 401 # --- POST /configure --- def _valid_config(): return { "vector_store": {"provider": "pgvector", "config": {"host": "h"}}, "llm": {"provider": "openai", "config": {"api_key": "x", "model": "m"}}, "embedder": {"provider": "openai", "config": {"api_key": "x", "model": "e"}}, } def test_post_configure_admin_jwt(client, auth_admin_header): response = client.post("/configure", headers=auth_admin_header, json=_valid_config()) assert response.status_code == 200 def test_post_configure_admin_api_key(client, admin_api_key_env): response = client.post("/configure", headers=admin_api_key_env, json=_valid_config()) assert response.status_code == 200 def test_post_configure_auth_disabled(client, auth_disabled_env): response = client.post("/configure", json=_valid_config()) assert response.status_code == 200 def test_post_configure_member_forbidden(client, auth_member_header): response = client.post("/configure", headers=auth_member_header, json=_valid_config()) assert response.status_code == 403 def test_post_configure_no_auth_unauthorized(client): response = client.post("/configure", json=_valid_config()) assert response.status_code == 401 # --- POST /reset --- def test_post_reset_admin_jwt(client, auth_admin_header): response = client.post("/reset", headers=auth_admin_header) assert response.status_code == 200 def test_post_reset_admin_api_key(client, admin_api_key_env): response = client.post("/reset", headers=admin_api_key_env) assert response.status_code == 200 def test_post_reset_auth_disabled(client, auth_disabled_env): response = client.post("/reset") assert response.status_code == 200 def test_post_reset_member_forbidden(client, auth_member_header): response = client.post("/reset", headers=auth_member_header) assert response.status_code == 403 def test_post_reset_no_auth_unauthorized(client): response = client.post("/reset") assert response.status_code == 401 # --- GET /entities --- def test_get_entities_admin_jwt(client, auth_admin_header): response = client.get("/entities", headers=auth_admin_header) assert response.status_code == 200 def test_get_entities_admin_api_key(client, admin_api_key_env): response = client.get("/entities", headers=admin_api_key_env) assert response.status_code == 200 def test_get_entities_auth_disabled(client, auth_disabled_env): response = client.get("/entities") assert response.status_code == 200 def test_get_entities_member_forbidden(client, auth_member_header): response = client.get("/entities", headers=auth_member_header) assert response.status_code == 403 def test_get_entities_no_auth_unauthorized(client): response = client.get("/entities") assert response.status_code == 401 # --- DELETE /entities/{type}/{id} --- def test_delete_entity_admin_jwt(client, auth_admin_header): response = client.delete("/entities/user/alice", headers=auth_admin_header) assert response.status_code == 200 def test_delete_entity_admin_api_key(client, admin_api_key_env): response = client.delete("/entities/user/alice", headers=admin_api_key_env) assert response.status_code == 200 def test_delete_entity_auth_disabled(client, auth_disabled_env): response = client.delete("/entities/user/alice") assert response.status_code == 200 def test_delete_entity_member_forbidden(client, auth_member_header): response = client.delete("/entities/user/alice", headers=auth_member_header) assert response.status_code == 403 def test_delete_entity_no_auth_unauthorized(client): response = client.delete("/entities/user/alice") assert response.status_code == 401 # --- GET /requests --- def test_get_requests_admin_jwt(client, auth_admin_header): response = client.get("/requests", headers=auth_admin_header) assert response.status_code == 200 def test_get_requests_admin_api_key(client, admin_api_key_env): response = client.get("/requests", headers=admin_api_key_env) assert response.status_code == 200 def test_get_requests_auth_disabled(client, auth_disabled_env): response = client.get("/requests") assert response.status_code == 200 def test_get_requests_member_forbidden(client, auth_member_header): response = client.get("/requests", headers=auth_member_header) assert response.status_code == 403 def test_get_requests_no_auth_unauthorized(client): response = client.get("/requests") assert response.status_code == 401 # --- GET /memories (no filter branch only) --- def test_get_memories_no_filter_admin_jwt(client, auth_admin_header): response = client.get("/memories", headers=auth_admin_header) assert response.status_code == 200 def test_get_memories_no_filter_admin_api_key(client, admin_api_key_env): response = client.get("/memories", headers=admin_api_key_env) assert response.status_code == 200 def test_get_memories_no_filter_auth_disabled(client, auth_disabled_env): response = client.get("/memories") assert response.status_code == 200 def test_get_memories_no_filter_member_forbidden(client, auth_member_header): """The info-disclosure branch (_list_all_memories) is admin-only.""" response = client.get("/memories", headers=auth_member_header) assert response.status_code == 403 def test_get_memories_filtered_member_succeeds(client, auth_member_header): """Filtered queries are unchanged — member can still call them.""" response = client.get("/memories?user_id=alice", headers=auth_member_header) assert response.status_code == 200 def test_get_memories_no_auth_unauthorized(client): response = client.get("/memories") assert response.status_code == 401