diff --git a/server.py b/server.py index c59b63e..5824795 100644 --- a/server.py +++ b/server.py @@ -44,7 +44,6 @@ from starlette.templating import Jinja2Templates # Import core modules from core import ( - EventType, LogLevel, ToolGenerator, get_api_key_manager, @@ -4258,10 +4257,7 @@ def create_multi_endpoint_app(transport: str = "streamable-http"): pass # 4. Master API key (fallback — no distinguishing prefix) - if auth_manager.validate_master_key(token): - return True - - return False + return bool(auth_manager.validate_master_key(token)) class DashboardCSRFMiddleware(BaseHTTPMiddleware): """ diff --git a/tests/test_dynamic_endpoints_integration.py b/tests/test_dynamic_endpoints_integration.py index 10c0525..1185674 100644 --- a/tests/test_dynamic_endpoints_integration.py +++ b/tests/test_dynamic_endpoints_integration.py @@ -3,12 +3,11 @@ Integration tests for the per-user dynamic MCP endpoints. Uses Starlette TestClient to simulate real HTTP requests through the server. """ -import pytest -from starlette.testclient import TestClient -from unittest.mock import patch, MagicMock, AsyncMock -import json +from unittest.mock import AsyncMock, MagicMock, patch +import pytest from server import create_multi_endpoint_app +from starlette.testclient import TestClient app = create_multi_endpoint_app() client = TestClient(app) diff --git a/tests/test_site_api.py b/tests/test_site_api.py index ae19157..6733c8c 100644 --- a/tests/test_site_api.py +++ b/tests/test_site_api.py @@ -2,7 +2,6 @@ from unittest.mock import AsyncMock, MagicMock, patch -import aiohttp import pytest from core.site_api import ( diff --git a/tests/test_tenant_isolation.py b/tests/test_tenant_isolation.py index 6fc7451..c653229 100644 --- a/tests/test_tenant_isolation.py +++ b/tests/test_tenant_isolation.py @@ -1,8 +1,9 @@ """Tests for tenant isolation in Dashboard routes.""" import pytest + from core.dashboard.routes import get_all_projects -from core.site_manager import SiteManager, SiteConfig +from core.site_manager import SiteConfig, SiteManager @pytest.mark.asyncio