tests/api/test_admin_routes_coverage.py

"""
Admin Routes Coverage Tests - Comprehensive Test Suite

Tests for admin routes (api/admin_routes.py) covering all 22 endpoints across 5 categories:
- Admin user CRUD (6): list, get, create, update, delete, update last login
- Admin role CRUD (5): list, get, create, update, delete
- WebSocket management (4): status, reconnect, disable, enable
- Rating sync (3): trigger sync, failed uploads, retry
- Conflict resolution (4): list, get detail, resolve, bulk resolve

Coverage target: 75%+ line coverage on api/admin_routes.py
Test patterns follow Phase 177/178 conventions
"""

import pytest
import uuid
from datetime import datetime, timezone, timedelta
from typing import List
from unittest.mock import MagicMock, AsyncMock, patch
from fastapi import FastAPI
from fastapi.testclient import TestClient
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, Session
from sqlalchemy.pool import StaticPool

# Import admin routes router
from api.admin_routes import router

# Import models
from core.models import (
    Base, User, AdminUser, AdminRole, WebSocketState,
    FailedRatingUpload, SkillRating, ConflictLog, SkillCache
)

# Import auth for password hashing
from core.auth import get_password_hash


# ============================================================================
# Test Database Setup
# ============================================================================

@pytest.fixture(scope="function")
def test_db():
    """Create in-memory SQLite database for testing."""
    engine = create_engine(
        "sqlite:///:memory:",
        connect_args={"check_same_thread": False},
        poolclass=StaticPool
    )

    # Create only the tables we need for admin routes testing
    # This avoids JSONB/SQLite incompatibility issues with other models
    from core.models import (
        AdminUser, AdminRole, User, WebSocketState,
        FailedRatingUpload, SkillRating, ConflictLog, SkillCache,
        CustomRole, Tenant, WorkflowTemplate
    )

    # Create tables individually using Table.create()
    Tenant.__table__.create(bind=engine, checkfirst=True)
    CustomRole.__table__.create(bind=engine, checkfirst=True)
    User.__table__.create(bind=engine, checkfirst=True)
    WorkflowTemplate.__table__.create(bind=engine, checkfirst=True)
    AdminUser.__table__.create(bind=engine, checkfirst=True)
    AdminRole.__table__.create(bind=engine, checkfirst=True)
    WebSocketState.__table__.create(bind=engine, checkfirst=True)
    FailedRatingUpload.__table__.create(bind=engine, checkfirst=True)
    SkillRating.__table__.create(bind=engine, checkfirst=True)
    ConflictLog.__table__.create(bind=engine, checkfirst=True)
    SkillCache.__table__.create(bind=engine, checkfirst=True)

    # Create session
    TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
    db = TestingSessionLocal()

    yield db

    # Cleanup
    db.close()
    # Drop tables individually
    SkillCache.__table__.drop(bind=engine)
    ConflictLog.__table__.drop(bind=engine)
    SkillRating.__table__.drop(bind=engine)
    FailedRatingUpload.__table__.drop(bind=engine)
    WebSocketState.__table__.drop(bind=engine)
    AdminUser.__table__.drop(bind=engine)
    AdminRole.__table__.drop(bind=engine)
    WorkflowTemplate.__table__.drop(bind=engine)
    User.__table__.drop(bind=engine)
    CustomRole.__table__.drop(bind=engine)
    Tenant.__table__.drop(bind=engine)


@pytest.fixture(scope="function")
def test_app(test_db: Session):
    """Create FastAPI app with admin routes for testing."""
    app = FastAPI()
    app.include_router(router)

    # Override get_db dependency
    from core.database import get_db

    def override_get_db():
        try:
            yield test_db
        finally:
            pass

    app.dependency_overrides[get_db] = override_get_db

    yield app

    # Clean up overrides
    app.dependency_overrides.clear()


@pytest.fixture(scope="function")
def client(test_app: FastAPI):
    """Create TestClient for testing."""
    return TestClient(test_app)


@pytest.fixture(scope="function")
def super_admin_user(test_db: Session) -> User:
    """Create super_admin user for authorization tests."""
    user = User(
        id=str(uuid.uuid4()),
        email="superadmin@test.com",
        first_name="Super",
        last_name="Admin",
        name="Super Admin",
        role="super_admin",
        status="active",
        email_verified=True
    )
    test_db.add(user)
    test_db.commit()
    test_db.refresh(user)
    return user


@pytest.fixture(scope="function")
def regular_user(test_db: Session) -> User:
    """Create regular user for auth testing."""
    user = User(
        id=str(uuid.uuid4()),
        email="member@test.com",
        first_name="Regular",
        last_name="Member",
        name="Regular Member",
        role="member",
        status="active",
        email_verified=True
    )
    test_db.add(user)
    test_db.commit()
    test_db.refresh(user)
    return user


@pytest.fixture(scope="function")
def authenticated_admin_client(client: TestClient, super_admin_user: User):
    """Create authenticated TestClient with super admin."""
    # Mock get_current_user to return super_admin
    from core.auth import get_current_user

    def override_get_current_user():
        return super_admin_user

    # Also mock require_super_admin
    from api.admin_routes import require_super_admin

    def override_require_super_admin():
        return super_admin_user

    client.app.dependency_overrides[get_current_user] = override_get_current_user
    client.app.dependency_overrides[require_super_admin] = override_require_super_admin

    yield client

    # Clean up
    client.app.dependency_overrides.clear()


@pytest.fixture(scope="function")
def unauthenticated_client(client: TestClient):
    """Create client without auth override."""
    return client


@pytest.fixture(scope="function")
def test_admin_role(test_db: Session) -> AdminRole:
    """Create test admin role."""
    role = AdminRole(
        id=str(uuid.uuid4()),
        name="Test Admin Role",
        permissions={"users": True, "workflows": False, "reports": True},
        description="Test role for coverage tests"
    )
    test_db.add(role)
    test_db.commit()
    test_db.refresh(role)
    return role


@pytest.fixture(scope="function")
def test_admin_user(test_db: Session, test_admin_role: AdminRole) -> AdminUser:
    """Create test admin user."""
    admin = AdminUser(
        id=str(uuid.uuid4()),
        email="testadmin@test.com",
        name="Test Admin User",
        password_hash=get_password_hash("TestPass123!"),
        role_id=test_admin_role.id,
        status="active",
        last_login=None,
        created_at=datetime.now(timezone.utc)
    )
    test_db.add(admin)
    test_db.commit()
    test_db.refresh(admin)
    return admin


@pytest.fixture(scope="function")
def mock_rating_sync_service():
    """Mock RatingSyncService for testing."""
    mock_service = AsyncMock()
    mock_service._sync_in_progress = False
    mock_service.get_pending_ratings = MagicMock(return_value=[])
    mock_service.sync_ratings = AsyncMock(return_value={
        "success": True, "uploaded": 10, "failed": 0, "skipped": 0
    })
    mock_service.upload_rating = AsyncMock(return_value={
        "success": True, "rating_id": "remote-123"
    })
    mock_service.mark_as_synced = MagicMock()
    return mock_service


@pytest.fixture(scope="function")
def mock_saas_client():
    """Mock AtomSaaSClient for testing."""
    mock_client = AsyncMock()
    return mock_client


@pytest.fixture(scope="function")
def mock_conflict_resolver():
    """Mock ConflictResolutionService for testing."""
    mock_resolver = MagicMock()
    mock_resolver.get_unresolved_conflicts = MagicMock(return_value=[])
    mock_resolver.get_conflict_by_id = MagicMock(return_value=None)
    mock_resolver.resolve_conflict = MagicMock(return_value=None)
    return mock_resolver


# ============================================================================
# Task 2: Admin User List and Get Tests
# ============================================================================

class TestAdminUserList:
    """Tests for GET /api/admin/users (list admin users)"""

    def test_list_admin_users_success(self, authenticated_admin_client: TestClient,
                                      test_db: Session, test_admin_role: AdminRole):
        """Test listing all admin users successfully."""
        # Create multiple admin users
        admin1 = AdminUser(
            id=str(uuid.uuid4()),
            email="admin1@test.com",
            name="Admin One",
            password_hash=get_password_hash("Pass123!"),
            role_id=test_admin_role.id,
            status="active"
        )
        admin2 = AdminUser(
            id=str(uuid.uuid4()),
            email="admin2@test.com",
            name="Admin Two",
            password_hash=get_password_hash("Pass456!"),
            role_id=test_admin_role.id,
            status="active"
        )
        test_db.add(admin1)
        test_db.add(admin2)
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/users")

        assert response.status_code == 200
        data = response.json()
        assert isinstance(data, list)
        assert len(data) >= 2

        # Verify response structure
        for user in data:
            assert "id" in user
            assert "email" in user
            assert "name" in user
            assert "role_id" in user
            assert "role_name" in user
            assert "permissions" in user
            assert "status" in user
            assert "last_login" in user

    def test_list_admin_users_empty(self, authenticated_admin_client: TestClient,
                                    test_db: Session):
        """Test listing admin users when none exist."""
        # Ensure no admin users exist
        test_db.query(AdminUser).delete()
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/users")

        assert response.status_code == 200
        data = response.json()
        assert isinstance(data, list)
        assert len(data) == 0

    def test_list_admin_users_includes_permissions(self, authenticated_admin_client: TestClient,
                                                   test_db: Session, test_admin_role: AdminRole):
        """Test that permissions are included in response."""
        admin = AdminUser(
            id=str(uuid.uuid4()),
            email="permissionadmin@test.com",
            name="Permission Test Admin",
            password_hash=get_password_hash("Pass123!"),
            role_id=test_admin_role.id,
            status="active"
        )
        test_db.add(admin)
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/users")

        assert response.status_code == 200
        data = response.json()
        assert len(data) >= 1

        # Find our test admin
        test_admin = next((u for u in data if u["email"] == "permissionadmin@test.com"), None)
        assert test_admin is not None
        assert test_admin["role_name"] == test_admin_role.name
        assert test_admin["permissions"] == test_admin_role.permissions


class TestAdminUserGet:
    """Tests for GET /api/admin/users/{admin_id} (get specific admin user)"""

    def test_get_admin_user_success(self, authenticated_admin_client: TestClient,
                                    test_admin_user: AdminUser):
        """Test getting a specific admin user successfully."""
        response = authenticated_admin_client.get(f"/api/admin/users/{test_admin_user.id}")

        assert response.status_code == 200
        data = response.json()
        assert data["id"] == test_admin_user.id
        assert data["email"] == test_admin_user.email
        assert data["name"] == test_admin_user.name
        assert data["role_id"] == test_admin_user.role_id
        assert data["status"] == test_admin_user.status
        assert "last_login" in data
        assert "created_at" in data

    def test_get_admin_user_not_found(self, authenticated_admin_client: TestClient):
        """Test getting a non-existent admin user."""
        fake_id = str(uuid.uuid4())

        response = authenticated_admin_client.get(f"/api/admin/users/{fake_id}")

        assert response.status_code == 404

    def test_get_admin_user_includes_created_at(self, authenticated_admin_client: TestClient,
                                                test_admin_user: AdminUser):
        """Test that created_at field is returned."""
        response = authenticated_admin_client.get(f"/api/admin/users/{test_admin_user.id}")

        assert response.status_code == 200
        data = response.json()
        assert "created_at" in data
        assert data["created_at"] is not None


# ============================================================================
# Task 3: Admin User Create, Update, Delete Tests
# ============================================================================

class TestAdminUserCreate:
    """Tests for POST /api/admin/users (create admin user)"""

    def test_create_admin_user_success(self, authenticated_admin_client: TestClient,
                                       test_db: Session, test_admin_role: AdminRole):
        """Test successful admin user creation."""
        request_data = {
            "email": "newadmin@test.com",
            "name": "New Admin",
            "password": "SecurePass123!",
            "role_id": test_admin_role.id
        }

        response = authenticated_admin_client.post("/api/admin/users", json=request_data)

        assert response.status_code == 201
        data = response.json()
        assert data["email"] == "newadmin@test.com"
        assert data["name"] == "New Admin"
        assert data["role_id"] == test_admin_role.id
        assert data["status"] == "active"
        assert "id" in data

        # Verify user created in DB
        admin = test_db.query(AdminUser).filter(AdminUser.email == "newadmin@test.com").first()
        assert admin is not None
        assert admin.password_hash != "SecurePass123!"  # Password should be hashed

    def test_create_admin_user_role_not_found(self, authenticated_admin_client: TestClient):
        """Test creation with non-existent role."""
        request_data = {
            "email": "test@test.com",
            "name": "Test Admin",
            "password": "SecurePass123!",
            "role_id": str(uuid.uuid4())  # Non-existent role
        }

        response = authenticated_admin_client.post("/api/admin/users", json=request_data)

        assert response.status_code == 404

    def test_create_admin_user_duplicate_email(self, authenticated_admin_client: TestClient,
                                              test_admin_user: AdminUser):
        """Test creation with duplicate email."""
        request_data = {
            "email": test_admin_user.email,  # Already exists
            "name": "Duplicate Admin",
            "password": "SecurePass123!",
            "role_id": test_admin_user.role_id
        }

        response = authenticated_admin_client.post("/api/admin/users", json=request_data)

        assert response.status_code == 409

    def test_create_admin_user_password_hashed(self, authenticated_admin_client: TestClient,
                                              test_db: Session, test_admin_role: AdminRole):
        """Test that password is hashed."""
        request_data = {
            "email": "hashed@test.com",
            "name": "Hashed Admin",
            "password": "PlainPassword123!",
            "role_id": test_admin_role.id
        }

        response = authenticated_admin_client.post("/api/admin/users", json=request_data)

        assert response.status_code == 201

        # Verify password hashed
        admin = test_db.query(AdminUser).filter(AdminUser.email == "hashed@test.com").first()
        assert admin is not None
        assert admin.password_hash != "PlainPassword123!"
        assert admin.password_hash.startswith("$2b$") or len(admin.password_hash) > 50

    def test_create_admin_user_default_status(self, authenticated_admin_client: TestClient,
                                             test_db: Session, test_admin_role: AdminRole):
        """Test that status defaults to active."""
        request_data = {
            "email": "defaultstatus@test.com",
            "name": "Default Status Admin",
            "password": "SecurePass123!",
            "role_id": test_admin_role.id
        }

        response = authenticated_admin_client.post("/api/admin/users", json=request_data)

        assert response.status_code == 201
        assert response.json()["status"] == "active"


class TestAdminUserUpdate:
    """Tests for PATCH /api/admin/users/{admin_id} (update admin user)"""

    def test_update_admin_user_name(self, authenticated_admin_client: TestClient,
                                    test_db: Session, test_admin_user: AdminUser):
        """Test updating only name."""
        response = authenticated_admin_client.patch(
            f"/api/admin/users/{test_admin_user.id}",
            json={"name": "Updated Name"}
        )

        assert response.status_code == 200
        data = response.json()
        assert data["name"] == "Updated Name"

        # Verify in DB
        test_db.refresh(test_admin_user)
        assert test_admin_user.name == "Updated Name"

    def test_update_admin_user_role(self, authenticated_admin_client: TestClient,
                                    test_db: Session, test_admin_user: AdminRole):
        """Test updating role."""
        # Create second role
        role2 = AdminRole(
            id=str(uuid.uuid4()),
            name="Second Role",
            permissions={"test": True},
            description="Second role"
        )
        test_db.add(role2)
        test_db.commit()

        response = authenticated_admin_client.patch(
            f"/api/admin/users/{test_admin_user.id}",
            json={"role_id": role2.id}
        )

        assert response.status_code == 200
        data = response.json()
        assert data["role_id"] == role2.id

    def test_update_admin_user_role_not_found(self, authenticated_admin_client: TestClient,
                                             test_admin_user: AdminUser):
        """Test updating with non-existent role."""
        response = authenticated_admin_client.patch(
            f"/api/admin/users/{test_admin_user.id}",
            json={"role_id": str(uuid.uuid4())}  # Non-existent role
        )

        assert response.status_code == 404

    def test_update_admin_user_status(self, authenticated_admin_client: TestClient,
                                     test_db: Session, test_admin_user: AdminUser):
        """Test updating status."""
        response = authenticated_admin_client.patch(
            f"/api/admin/users/{test_admin_user.id}",
            json={"status": "inactive"}
        )

        assert response.status_code == 200

        # Verify in DB
        test_db.refresh(test_admin_user)
        assert test_admin_user.status == "inactive"

    def test_update_admin_user_not_found(self, authenticated_admin_client: TestClient):
        """Test updating non-existent user."""
        response = authenticated_admin_client.patch(
            f"/api/admin/users/{uuid.uuid4()}",
            json={"name": "Test"}
        )

        assert response.status_code == 404


class TestAdminUserDelete:
    """Tests for DELETE /api/admin/users/{admin_id} (delete admin user)"""

    def test_delete_admin_user_success(self, authenticated_admin_client: TestClient,
                                       test_db: Session, test_admin_user: AdminUser):
        """Test successful deletion."""
        admin_id = test_admin_user.id

        response = authenticated_admin_client.delete(f"/api/admin/users/{admin_id}")

        assert response.status_code == 200
        data = response.json()
        assert "deleted" in data["message"].lower()

        # Verify deleted from DB
        admin = test_db.query(AdminUser).filter(AdminUser.id == admin_id).first()
        assert admin is None

    def test_delete_admin_user_not_found(self, authenticated_admin_client: TestClient):
        """Test deleting non-existent user."""
        response = authenticated_admin_client.delete(f"/api/admin/users/{uuid.uuid4()}")

        assert response.status_code == 404


# ============================================================================
# Task 4: Admin User Last Login and Admin Role Tests
# ============================================================================

class TestAdminUserLastLogin:
    """Tests for PATCH /api/admin/users/{admin_id}/last-login"""

    def test_update_last_login_success(self, authenticated_admin_client: TestClient,
                                       test_db: Session, test_admin_user: AdminUser):
        """Test updating last login timestamp."""
        assert test_admin_user.last_login is None

        response = authenticated_admin_client.patch(
            f"/api/admin/users/{test_admin_user.id}/last-login"
        )

        assert response.status_code == 200
        data = response.json()
        assert "updated" in data["message"].lower()

        # Verify updated in DB
        test_db.refresh(test_admin_user)
        assert test_admin_user.last_login is not None
        assert isinstance(test_admin_user.last_login, datetime)

    def test_update_last_login_not_found(self, authenticated_admin_client: TestClient):
        """Test updating non-existent user."""
        response = authenticated_admin_client.patch(
            f"/api/admin/users/{uuid.uuid4()}/last-login"
        )

        assert response.status_code == 404

    def test_update_last_login_sets_timestamp(self, authenticated_admin_client: TestClient,
                                             test_db: Session, test_admin_user: AdminUser):
        """Test that timestamp is set correctly."""
        old_login = test_admin_user.last_login
        import time
        time.sleep(0.1)  # Small delay

        response = authenticated_admin_client.patch(
            f"/api/admin/users/{test_admin_user.id}/last-login"
        )

        assert response.status_code == 200

        # Verify last_login updated
        test_db.refresh(test_admin_user)
        assert test_admin_user.last_login is not None
        if old_login:
            assert test_admin_user.last_login > old_login


class TestAdminRoleList:
    """Tests for GET /api/admin/roles (list admin roles)"""

    def test_list_admin_roles_success(self, authenticated_admin_client: TestClient,
                                     test_db: Session, test_admin_role: AdminRole):
        """Test listing all admin roles."""
        # Create second role
        role2 = AdminRole(
            id=str(uuid.uuid4()),
            name="Another Role",
            permissions={"test": True},
            description="Another test role"
        )
        test_db.add(role2)
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/roles")

        assert response.status_code == 200
        data = response.json()
        assert isinstance(data, list)
        assert len(data) >= 2

        # Verify response structure
        for role in data:
            assert "id" in role
            assert "name" in role
            assert "permissions" in role
            assert "description" in role

    def test_list_admin_roles_empty(self, authenticated_admin_client: TestClient,
                                    test_db: Session):
        """Test listing when no roles exist."""
        test_db.query(AdminRole).delete()
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/roles")

        assert response.status_code == 200
        data = response.json()
        assert isinstance(data, list)
        assert len(data) == 0


class TestAdminRoleGet:
    """Tests for GET /api/admin/roles/{role_id} (get specific admin role)"""

    def test_get_admin_role_success(self, authenticated_admin_client: TestClient,
                                    test_admin_role: AdminRole):
        """Test getting specific role."""
        response = authenticated_admin_client.get(f"/api/admin/roles/{test_admin_role.id}")

        assert response.status_code == 200
        data = response.json()
        assert data["id"] == test_admin_role.id
        assert data["name"] == test_admin_role.name
        assert data["permissions"] == test_admin_role.permissions
        assert data["description"] == test_admin_role.description

    def test_get_admin_role_not_found(self, authenticated_admin_client: TestClient):
        """Test getting non-existent role."""
        response = authenticated_admin_client.get(f"/api/admin/roles/{uuid.uuid4()}")

        assert response.status_code == 404


class TestAdminRoleCreate:
    """Tests for POST /api/admin/roles (create admin role)"""

    def test_create_admin_role_success(self, authenticated_admin_client: TestClient):
        """Test successful role creation."""
        request_data = {
            "name": "New Role",
            "permissions": {"users": True, "reports": False},
            "description": "Test role"
        }

        response = authenticated_admin_client.post("/api/admin/roles", json=request_data)

        assert response.status_code == 201
        data = response.json()
        assert data["name"] == "New Role"
        assert data["permissions"] == {"users": True, "reports": False}
        assert data["description"] == "Test role"
        assert "id" in data

    def test_create_admin_role_duplicate_name(self, authenticated_admin_client: TestClient,
                                             test_admin_role: AdminRole):
        """Test creation with duplicate name."""
        request_data = {
            "name": test_admin_role.name,  # Already exists
            "permissions": {"users": True},
            "description": "Duplicate"
        }

        response = authenticated_admin_client.post("/api/admin/roles", json=request_data)

        assert response.status_code == 409


# ============================================================================
# Task 5: Admin Role Update and Delete Tests
# ============================================================================

class TestAdminRoleUpdate:
    """Tests for PATCH /api/admin/roles/{role_id} (update admin role)"""

    def test_update_admin_role_name(self, authenticated_admin_client: TestClient,
                                    test_db: Session, test_admin_role: AdminRole):
        """Test updating role name."""
        response = authenticated_admin_client.patch(
            f"/api/admin/roles/{test_admin_role.id}",
            json={"name": "Updated Name"}
        )

        assert response.status_code == 200
        data = response.json()
        assert data["name"] == "Updated Name"

        # Verify in DB
        test_db.refresh(test_admin_role)
        assert test_admin_role.name == "Updated Name"

    def test_update_admin_role_name_conflict(self, authenticated_admin_client: TestClient,
                                            test_db: Session, test_admin_role: AdminRole):
        """Test updating name to existing name."""
        # Create second role
        role2 = AdminRole(
            id=str(uuid.uuid4()),
            name="Another Role",
            permissions={"test": True},
            description="Second role"
        )
        test_db.add(role2)
        test_db.commit()

        response = authenticated_admin_client.patch(
            f"/api/admin/roles/{test_admin_role.id}",
            json={"name": role2.name}  # Duplicate name
        )

        assert response.status_code == 409

    def test_update_admin_role_permissions(self, authenticated_admin_client: TestClient,
                                         test_db: Session, test_admin_role: AdminRole):
        """Test updating permissions."""
        new_permissions = {"users": False, "admin": True}
        response = authenticated_admin_client.patch(
            f"/api/admin/roles/{test_admin_role.id}",
            json={"permissions": new_permissions}
        )

        assert response.status_code == 200

        # Verify in DB
        test_db.refresh(test_admin_role)
        assert test_admin_role.permissions == new_permissions

    def test_update_admin_role_description(self, authenticated_admin_client: TestClient,
                                         test_db: Session, test_admin_role: AdminRole):
        """Test updating description."""
        response = authenticated_admin_client.patch(
            f"/api/admin/roles/{test_admin_role.id}",
            json={"description": "New description"}
        )

        assert response.status_code == 200

        # Verify in DB
        test_db.refresh(test_admin_role)
        assert test_admin_role.description == "New description"

    def test_update_admin_role_not_found(self, authenticated_admin_client: TestClient):
        """Test updating non-existent role."""
        response = authenticated_admin_client.patch(
            f"/api/admin/roles/{uuid.uuid4()}",
            json={"name": "Test"}
        )

        assert response.status_code == 404


class TestAdminRoleDelete:
    """Tests for DELETE /api/admin/roles/{role_id} (delete admin role)"""

    def test_delete_admin_role_success(self, authenticated_admin_client: TestClient,
                                      test_db: Session):
        """Test successful role deletion."""
        # Create role with no users
        role = AdminRole(
            id=str(uuid.uuid4()),
            name="Deletable Role",
            permissions={"test": True},
            description="Role for deletion test"
        )
        test_db.add(role)
        test_db.commit()
        role_id = role.id

        response = authenticated_admin_client.delete(f"/api/admin/roles/{role_id}")

        assert response.status_code == 200
        data = response.json()
        assert "deleted" in data["message"].lower()

        # Verify deleted from DB
        deleted_role = test_db.query(AdminRole).filter(AdminRole.id == role_id).first()
        assert deleted_role is None

    def test_delete_admin_role_with_users_fails(self, authenticated_admin_client: TestClient,
                                               test_admin_role: AdminRole,
                                               test_admin_user: AdminUser):
        """Test deletion when role assigned to users."""
        # test_admin_user has test_admin_role assigned
        response = authenticated_admin_client.delete(f"/api/admin/roles/{test_admin_role.id}")

        assert response.status_code == 409

    def test_delete_admin_role_not_found(self, authenticated_admin_client: TestClient):
        """Test deleting non-existent role."""
        response = authenticated_admin_client.delete(f"/api/admin/roles/{uuid.uuid4()}")

        assert response.status_code == 404


# ============================================================================
# Task 6: WebSocket Management Tests
# ============================================================================

class TestWebSocketStatus:
    """Tests for GET /api/admin/websocket/status"""

    def test_get_websocket_status_connected(self, authenticated_admin_client: TestClient,
                                           test_db: Session):
        """Test WebSocket status when connected."""
        # Create WebSocket state
        ws_state = WebSocketState(
            id=1,
            connected=True,
            ws_url="wss://api.example.com/ws",
            last_connected_at=datetime.now(timezone.utc),
            last_message_at=datetime.now(timezone.utc),
            reconnect_attempts=3,
            consecutive_failures=0,
            disconnect_reason=None,
            fallback_to_polling=False
        )
        test_db.add(ws_state)
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/websocket/status")

        assert response.status_code == 200
        data = response.json()
        assert data["connected"] is True
        assert data["ws_url"] == "wss://api.example.com/ws"
        assert data["reconnect_attempts"] == 3
        assert data["consecutive_failures"] == 0
        assert data["fallback_to_polling"] is False
        assert data["rate_limit_messages_per_sec"] == 100

    def test_get_websocket_status_no_state(self, authenticated_admin_client: TestClient):
        """Test WebSocket status when no state exists."""
        response = authenticated_admin_client.get("/api/admin/websocket/status")

        assert response.status_code == 200
        data = response.json()
        assert data["connected"] is False
        assert data["reconnect_attempts"] == 0
        assert data["consecutive_failures"] == 0
        assert data["fallback_to_polling"] is False

    def test_get_websocket_status_disconnected(self, authenticated_admin_client: TestClient,
                                              test_db: Session):
        """Test WebSocket status when disconnected."""
        ws_state = WebSocketState(
            id=1,
            connected=False,
            reconnect_attempts=5,
            consecutive_failures=3,
            disconnect_reason="connection_lost",
            fallback_to_polling=True
        )
        test_db.add(ws_state)
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/websocket/status")

        assert response.status_code == 200
        data = response.json()
        assert data["connected"] is False
        assert data["last_disconnect_reason"] == "connection_lost"
        assert data["fallback_to_polling"] is True


class TestWebSocketReconnect:
    """Tests for POST /api/admin/websocket/reconnect"""

    def test_trigger_websocket_reconnect(self, authenticated_admin_client: TestClient,
                                        test_db: Session):
        """Test forcing WebSocket reconnection."""
        # Create WebSocket state
        ws_state = WebSocketState(
            id=1,
            connected=False,
            reconnect_attempts=5,
            consecutive_failures=3,
            fallback_to_polling=True
        )
        test_db.add(ws_state)
        test_db.commit()

        response = authenticated_admin_client.post("/api/admin/websocket/reconnect")

        assert response.status_code == 200
        data = response.json()
        assert data["reconnect_triggered"] is True
        assert "Reconnection triggered" in data["message"]

    def test_trigger_websocket_reconnect_resets_counters(self, authenticated_admin_client: TestClient,
                                                         test_db: Session):
        """Test that reconnect resets counters."""
        ws_state = WebSocketState(
            id=1,
            connected=False,
            reconnect_attempts=5,
            consecutive_failures=3,
            fallback_to_polling=True
        )
        test_db.add(ws_state)
        test_db.commit()

        response = authenticated_admin_client.post("/api/admin/websocket/reconnect")

        assert response.status_code == 200

        # Verify DB updated
        test_db.refresh(ws_state)
        assert ws_state.reconnect_attempts == 0
        assert ws_state.consecutive_failures == 0
        assert ws_state.fallback_to_polling is False


class TestWebSocketToggle:
    """Tests for POST /api/admin/websocket/disable and /enable"""

    def test_disable_websocket(self, authenticated_admin_client: TestClient, test_db: Session):
        """Test disabling WebSocket."""
        ws_state = WebSocketState(
            id=1,
            connected=True,
            fallback_to_polling=False
        )
        test_db.add(ws_state)
        test_db.commit()

        response = authenticated_admin_client.post("/api/admin/websocket/disable")

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is True
        assert data["websocket_enabled"] is False

        # Verify DB updated
        test_db.refresh(ws_state)
        assert ws_state.fallback_to_polling is True
        assert ws_state.connected is False
        assert ws_state.disconnect_reason == "disabled_by_admin"

    def test_enable_websocket(self, authenticated_admin_client: TestClient, test_db: Session):
        """Test enabling WebSocket."""
        ws_state = WebSocketState(
            id=1,
            fallback_to_polling=True,
            next_ws_attempt_at=datetime.now(timezone.utc) + timedelta(hours=1),
            reconnect_attempts=5
        )
        test_db.add(ws_state)
        test_db.commit()

        response = authenticated_admin_client.post("/api/admin/websocket/enable")

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is True
        assert data["websocket_enabled"] is True

        # Verify DB updated
        test_db.refresh(ws_state)
        assert ws_state.fallback_to_polling is False
        assert ws_state.next_ws_attempt_at is None
        assert ws_state.reconnect_attempts == 0

    def test_disable_websocket_sets_disconnect_reason(self, authenticated_admin_client: TestClient,
                                                     test_db: Session):
        """Test disable sets disconnect reason."""
        ws_state = WebSocketState(id=1, connected=True)
        test_db.add(ws_state)
        test_db.commit()

        response = authenticated_admin_client.post("/api/admin/websocket/disable")

        assert response.status_code == 200

        # Verify disconnect_reason set
        test_db.refresh(ws_state)
        assert ws_state.disconnect_reason == "disabled_by_admin"


# ============================================================================
# Task 7: Rating Sync Tests
# ============================================================================

class TestRatingSync:
    """Tests for POST /api/admin/sync/ratings"""

    def test_trigger_rating_sync_success(self, authenticated_admin_client: TestClient,
                                        test_db: Session, mock_rating_sync_service):
        """Test triggering rating sync successfully."""
        # Create pending ratings
        for i in range(5):
            rating = SkillRating(
                id=f"rating_{i}",
                skill_id=f"skill_{i}",
                user_id="test_user",
                tenant_id="test_tenant",
                rating=5,
                synced_at=None
            )
            test_db.add(rating)
        test_db.commit()

        with patch('core.rating_sync_service.RatingSyncService', return_value=mock_rating_sync_service):
            response = authenticated_admin_client.post(
                "/api/admin/sync/ratings",
                json={"upload_all": False}
            )

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is True
        assert data["uploaded"] == 10

    def test_trigger_rating_sync_upload_all(self, authenticated_admin_client: TestClient,
                                          mock_rating_sync_service):
        """Test sync with upload_all flag."""
        with patch('core.rating_sync_service.RatingSyncService', return_value=mock_rating_sync_service):
            response = authenticated_admin_client.post(
                "/api/admin/sync/ratings",
                json={"upload_all": True}
            )

        assert response.status_code == 200
        assert response.json()["uploaded"] == 10

    def test_trigger_rating_sync_in_progress(self, authenticated_admin_client: TestClient):
        """Test sync when already in progress."""
        mock_service = MagicMock()
        mock_service._sync_in_progress = True
        mock_service.get_pending_ratings = MagicMock(return_value=[])

        with patch('core.rating_sync_service.RatingSyncService', return_value=mock_service):
            response = authenticated_admin_client.post(
                "/api/admin/sync/ratings",
                json={"upload_all": False}
            )

        assert response.status_code == 503


class TestFailedRatingUploads:
    """Tests for GET /api/admin/ratings/failed-uploads"""

    def test_get_failed_rating_uploads(self, authenticated_admin_client: TestClient,
                                      test_db: Session):
        """Test listing failed uploads."""
        # Create failed uploads
        for i in range(3):
            failed = FailedRatingUpload(
                id=f"failed_{i}",
                rating_id=f"rating_{i}",
                error_message=f"Error {i}",
                failed_at=datetime.now(timezone.utc),
                retry_count=i,
                last_retry_at=datetime.now(timezone.utc) if i > 0 else None
            )
            test_db.add(failed)
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/ratings/failed-uploads")

        assert response.status_code == 200
        data = response.json()
        assert len(data) == 3
        assert data[0]["rating_id"] == "rating_0"
        assert data[0]["retry_count"] == 0

    def test_get_failed_rating_uploads_empty(self, authenticated_admin_client: TestClient):
        """Test listing when no failed uploads."""
        response = authenticated_admin_client.get("/api/admin/ratings/failed-uploads")

        assert response.status_code == 200
        data = response.json()
        assert data == []

    def test_get_failed_uploads_limited(self, authenticated_admin_client: TestClient,
                                       test_db: Session):
        """Test that results are limited to 100."""
        # Create 150 failed uploads
        for i in range(150):
            failed = FailedRatingUpload(
                id=f"failed_{i}",
                rating_id=f"rating_{i}",
                error_message=f"Error {i}",
                failed_at=datetime.now(timezone.utc),
                retry_count=0
            )
            test_db.add(failed)
        test_db.commit()

        response = authenticated_admin_client.get("/api/admin/ratings/failed-uploads")

        assert response.status_code == 200
        data = response.json()
        assert len(data) <= 100  # Should be limited


class TestRetryRatingUpload:
    """Tests for POST /api/admin/ratings/failed-uploads/{failed_id}/retry"""

    def test_retry_failed_upload_success(self, authenticated_admin_client: TestClient,
                                        test_db: Session):
        """Test retrying failed upload successfully."""
        # Create rating
        rating = SkillRating(
            id="rating_1",
            skill_id="skill_1",
            user_id="test_user",
            tenant_id="test_tenant",
            rating=5,
            synced_at=None
        )
        test_db.add(rating)

        # Create failed upload
        failed = FailedRatingUpload(
            id="failed_1",
            rating_id="rating_1",
            error_message="Network error",
            failed_at=datetime.now(timezone.utc),
            retry_count=0
        )
        test_db.add(failed)
        test_db.commit()

        # Mock service
        mock_service = MagicMock()
        mock_service.upload_rating = AsyncMock(return_value={
            "success": True, "rating_id": "remote-123"
        })
        mock_service.mark_as_synced = MagicMock()

        with patch('core.rating_sync_service.RatingSyncService', return_value=mock_service):
            response = authenticated_admin_client.post(
                "/api/admin/ratings/failed-uploads/failed_1/retry"
            )

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is True
        assert data["retry_triggered"] is True

    def test_retry_failed_upload_rating_deleted(self, authenticated_admin_client: TestClient,
                                               test_db: Session):
        """Test retry when rating was deleted."""
        # Create failed upload without rating
        failed = FailedRatingUpload(
            id="failed_1",
            rating_id="deleted_rating",
            error_message="Network error",
            failed_at=datetime.now(timezone.utc),
            retry_count=0
        )
        test_db.add(failed)
        test_db.commit()

        response = authenticated_admin_client.post(
            "/api/admin/ratings/failed-uploads/failed_1/retry"
        )

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is False

    def test_retry_failed_upload_fails_again(self, authenticated_admin_client: TestClient,
                                            test_db: Session):
        """Test retry that fails again."""
        # Create rating
        rating = SkillRating(
            id="rating_1",
            skill_id="skill_1",
            user_id="test_user",
            tenant_id="test_tenant",
            rating=5
        )
        test_db.add(rating)

        # Create failed upload
        failed = FailedRatingUpload(
            id="failed_1",
            rating_id="rating_1",
            error_message="Network error",
            failed_at=datetime.now(timezone.utc),
            retry_count=1
        )
        test_db.add(failed)
        test_db.commit()

        # Mock service to fail
        mock_service = MagicMock()
        mock_service.upload_rating = AsyncMock(return_value={
            "success": False, "error": "API error"
        })

        with patch('core.rating_sync_service.RatingSyncService', return_value=mock_service):
            response = authenticated_admin_client.post(
                "/api/admin/ratings/failed-uploads/failed_1/retry"
            )

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is False

        # Verify retry count incremented
        test_db.refresh(failed)
        assert failed.retry_count == 2

    def test_retry_failed_upload_not_found(self, authenticated_admin_client: TestClient):
        """Test retrying non-existent failed upload."""
        response = authenticated_admin_client.post(
            "/api/admin/ratings/failed-uploads/nonexistent/retry"
        )

        assert response.status_code == 404


# ============================================================================
# Task 8: Conflict Resolution Tests
# ============================================================================

class TestConflictList:
    """Tests for GET /api/admin/conflicts"""

    def test_list_conflicts(self, authenticated_admin_client: TestClient, test_db: Session):
        """Test listing unresolved conflicts."""
        # Create conflicts
        for i in range(3):
            conflict = ConflictLog(
                id=i + 1,
                skill_id=f"skill_{i}",
                conflict_type="version_mismatch",
                severity="high",
                local_data={"version": "1.0"},
                remote_data={"version": "2.0"},
                created_at=datetime.now(timezone.utc)
            )
            test_db.add(conflict)
        test_db.commit()

        mock_resolver = MagicMock()
        mock_resolver.get_unresolved_conflicts = MagicMock(return_value=test_db.query(ConflictLog).all())

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.get("/api/admin/conflicts")

        assert response.status_code == 200
        data = response.json()
        assert "conflicts" in data
        assert data["total_count"] == 3
        assert data["page"] == 1

    def test_list_conflicts_with_filters(self, authenticated_admin_client: TestClient):
        """Test listing with severity and type filters."""
        mock_resolver = MagicMock()
        mock_resolver.get_unresolved_conflicts = MagicMock(return_value=[])

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.get(
                "/api/admin/conflicts?severity=HIGH&conflict_type=skill_data"
            )

        assert response.status_code == 200

    def test_list_conflicts_with_pagination(self, authenticated_admin_client: TestClient):
        """Test pagination parameters."""
        mock_resolver = MagicMock()
        mock_resolver.get_unresolved_conflicts = MagicMock(return_value=[])

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.get(
                "/api/admin/conflicts?page=2&page_size=25"
            )

        assert response.status_code == 200
        data = response.json()
        assert data["page"] == 2
        assert data["page_size"] == 25


class TestConflictGet:
    """Tests for GET /api/admin/conflicts/{conflict_id}"""

    def test_get_conflict_success(self, authenticated_admin_client: TestClient, test_db: Session):
        """Test getting conflict by ID."""
        conflict = ConflictLog(
            id=1,
            skill_id="skill_1",
            conflict_type="version_mismatch",
            severity="high",
            local_data={"version": "1.0"},
            remote_data={"version": "2.0"},
            created_at=datetime.now(timezone.utc)
        )
        test_db.add(conflict)
        test_db.commit()

        mock_resolver = MagicMock()
        mock_resolver.get_conflict_by_id = MagicMock(return_value=conflict)

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.get("/api/admin/conflicts/1")

        assert response.status_code == 200
        data = response.json()
        assert data["id"] == 1
        assert data["skill_id"] == "skill_1"

    def test_get_conflict_not_found(self, authenticated_admin_client: TestClient):
        """Test getting non-existent conflict."""
        mock_resolver = MagicMock()
        mock_resolver.get_conflict_by_id = MagicMock(return_value=None)

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.get("/api/admin/conflicts/999")

        assert response.status_code == 404


class TestConflictResolve:
    """Tests for POST /api/admin/conflicts/{conflict_id}/resolve"""

    def test_resolve_conflict_remote_wins(self, authenticated_admin_client: TestClient,
                                         test_db: Session):
        """Test resolving with remote_wins strategy."""
        conflict = ConflictLog(
            id=1,
            skill_id="skill_1",
            conflict_type="version_mismatch",
            severity="high",
            local_data={"version": "1.0"},
            remote_data={"version": "2.0"},
            created_at=datetime.now(timezone.utc)
        )
        test_db.add(conflict)
        test_db.commit()

        resolved_data = {"skill_id": "skill_1", "version": "2.0"}

        mock_resolver = MagicMock()
        mock_resolver.resolve_conflict = MagicMock(return_value=resolved_data)

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.post(
                "/api/admin/conflicts/1/resolve",
                json={"strategy": "remote_wins", "resolved_by": "admin"}
            )

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is True
        assert data["resolved_data"] == resolved_data

    def test_resolve_conflict_local_wins(self, authenticated_admin_client: TestClient):
        """Test resolving with local_wins strategy."""
        resolved_data = {"skill_id": "skill_1", "version": "1.0"}

        mock_resolver = MagicMock()
        mock_resolver.resolve_conflict = MagicMock(return_value=resolved_data)

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.post(
                "/api/admin/conflicts/1/resolve",
                json={"strategy": "local_wins", "resolved_by": "admin"}
            )

        assert response.status_code == 200

    def test_resolve_conflict_merge(self, authenticated_admin_client: TestClient):
        """Test resolving with merge strategy."""
        resolved_data = {"skill_id": "skill_1", "version": "2.0", "local_changes": "preserved"}

        mock_resolver = MagicMock()
        mock_resolver.resolve_conflict = MagicMock(return_value=resolved_data)

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.post(
                "/api/admin/conflicts/1/resolve",
                json={"strategy": "merge", "resolved_by": "admin"}
            )

        assert response.status_code == 200

    def test_resolve_conflict_invalid_strategy(self, authenticated_admin_client: TestClient):
        """Test with invalid strategy."""
        response = authenticated_admin_client.post(
            "/api/admin/conflicts/1/resolve",
            json={"strategy": "invalid", "resolved_by": "admin"}
        )

        assert response.status_code == 422

    def test_resolve_conflict_not_found(self, authenticated_admin_client: TestClient):
        """Test resolving non-existent conflict."""
        mock_resolver = MagicMock()
        mock_resolver.resolve_conflict = MagicMock(return_value=None)

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.post(
                "/api/admin/conflicts/999/resolve",
                json={"strategy": "remote_wins", "resolved_by": "admin"}
            )

        assert response.status_code == 200
        data = response.json()
        assert data["success"] is False


class TestBulkConflictResolve:
    """Tests for POST /api/admin/conflicts/bulk-resolve"""

    def test_bulk_resolve_conflicts_success(self, authenticated_admin_client: TestClient):
        """Test bulk resolving conflicts successfully."""
        mock_resolver = MagicMock()
        mock_resolver.resolve_conflict = MagicMock(
            return_value={"skill_id": "skill_1", "version": "2.0"}
        )

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.post(
                "/api/admin/conflicts/bulk-resolve",
                json={
                    "conflict_ids": [1, 2, 3],
                    "strategy": "remote_wins",
                    "resolved_by": "admin"
                }
            )

        assert response.status_code == 200
        data = response.json()
        assert data["resolved_count"] == 3
        assert data["failed_count"] == 0

    def test_bulk_resolve_with_failures(self, authenticated_admin_client: TestClient):
        """Test bulk resolve with some failures."""
        mock_resolver = MagicMock()

        # Make second resolve fail
        def side_effect(conflict_id, strategy, resolved_by):
            if conflict_id == 2:
                return None
            return {"skill_id": f"skill_{conflict_id}", "version": "2.0"}

        mock_resolver.resolve_conflict.side_effect = side_effect

        with patch('core.conflict_resolution_service.ConflictResolutionService', return_value=mock_resolver):
            response = authenticated_admin_client.post(
                "/api/admin/conflicts/bulk-resolve",
                json={
                    "conflict_ids": [1, 2, 3],
                    "strategy": "remote_wins",
                    "resolved_by": "admin"
                }
            )

        assert response.status_code == 200
        data = response.json()
        assert data["resolved_count"] == 2
        assert data["failed_count"] == 1

    def test_bulk_resolve_invalid_strategy(self, authenticated_admin_client: TestClient):
        """Test bulk resolve with invalid strategy."""
        response = authenticated_admin_client.post(
            "/api/admin/conflicts/bulk-resolve",
            json={
                "conflict_ids": [1, 2, 3],
                "strategy": "invalid",
                "resolved_by": "admin"
            }
        )

        assert response.status_code == 422

    def test_bulk_resolve_empty_conflict_ids(self, authenticated_admin_client: TestClient):
        """Test with empty conflict_ids list."""
        response = authenticated_admin_client.post(
            "/api/admin/conflicts/bulk-resolve",
            json={
                "conflict_ids": [],
                "strategy": "remote_wins",
                "resolved_by": "admin"
            }
        )

        assert response.status_code == 422

    def test_bulk_resolve_too_many_conflicts(self, authenticated_admin_client: TestClient):
        """Test exceeding max conflict_ids."""
        conflict_ids = list(range(101))

        response = authenticated_admin_client.post(
            "/api/admin/conflicts/bulk-resolve",
            json={
                "conflict_ids": conflict_ids,
                "strategy": "remote_wins",
                "resolved_by": "admin"
            }
        )

        assert response.status_code == 422


# ============================================================================
# Task 9: Authentication and Governance Tests
# ============================================================================

class TestAdminRoutesAuth:
    """Tests for super_admin requirement on all endpoints"""

    def test_unauthenticated_request_fails(self, unauthenticated_client: TestClient):
        """Test unauthenticated request."""
        response = unauthenticated_client.get("/api/admin/users")

        # Should return 401 or 403
        assert response.status_code in [401, 403]

    def test_inactive_admin_blocked(self, authenticated_admin_client: TestClient,
                                    test_db: Session, test_admin_role: AdminRole):
        """Test that inactive admin is blocked."""
        # Create inactive super_admin
        inactive_super = User(
            id=str(uuid.uuid4()),
            email="inactive@test.com",
            name="Inactive Super",
            role="super_admin",
            status="inactive",
            email_verified=True
        )
        test_db.add(inactive_super)
        test_db.commit()

        # Override auth to return inactive super
        def override_get_current_user():
            return inactive_super

        def override_require_super_admin():
            return inactive_super

        authenticated_admin_client.app.dependency_overrides[
            authenticated_admin_client.app.dependencies[0].dependency
        ] = override_get_current_user

        from api.admin_routes import require_super_admin
        authenticated_admin_client.app.dependency_overrides[require_super_admin] = override_require_super_admin

        # Should still work if role is super_admin regardless of status
        # (this is implementation-specific - adjust if different)
        response = authenticated_admin_client.get("/api/admin/users")
        # May pass or fail depending on implementation
        assert response.status_code in [200, 403]


class TestGovernanceEnforcement:
    """Tests for AUTONOMOUS maturity governance on critical operations"""

    def test_create_user_governance_CRITICAL(self, authenticated_admin_client: TestClient,
                                            test_admin_role: AdminRole):
        """Test CRITICAL complexity governance for create user."""
        # Mock governance to fail
        with patch('core.agent_governance_service.GovernanceCache') as mock_cache:
            mock_instance = MagicMock()
            mock_instance.can_perform_action.return_value = (
                False, "PENDING_APPROVAL", "Not AUTONOMOUS"
            )
            mock_cache.return_value = mock_instance

            # Add agent_id to simulate agent request
            # This test documents governance requirement
            # Actual governance enforcement depends on implementation
            request_data = {
                "email": "governance@test.com",
                "name": "Governance Test",
                "password": "SecurePass123!",
                "role_id": test_admin_role.id
            }

            # May pass for human users, fail for agents
            response = authenticated_admin_client.post("/api/admin/users", json=request_data)
            # Human users (no agent_id) should pass
            assert response.status_code in [201, 403]

    def test_AUTONOMOUS_passes_all_checks(self, authenticated_admin_client: TestClient,
                                         test_admin_role: AdminRole):
        """Test AUTONOMOUS maturity passes all checks."""
        # Mock governance to succeed
        with patch('core.agent_governance_service.GovernanceCache') as mock_cache:
            mock_instance = MagicMock()
            mock_instance.can_perform_action.return_value = (
                True, "AUTONOMOUS", "All checks passed"
            )
            mock_cache.return_value = mock_instance

            request_data = {
                "email": "autonomous@test.com",
                "name": "Autonomous Test",
                "password": "SecurePass123!",
                "role_id": test_admin_role.id
            }

            # Should succeed
            response = authenticated_admin_client.post("/api/admin/users", json=request_data)
            assert response.status_code == 201