Fix Unit Tests to Handle UI-Disabled Mode closes #378 (#379)

Signed-off-by: Mihai Criveti <[email protected]>

Author: crivetimihai

tests/unit/mcpgateway/test_main.py

@@ -18,6 +18,7 @@
 import pytest
 
 # First-Party
+from mcpgateway.config import settings
 from mcpgateway.models import InitializeResult, ResourceContent, ServerCapabilities
 from mcpgateway.schemas import (
     PromptRead,
@@ -209,10 +210,20 @@ def test_ready_check(self, test_client):
         assert response.json()["status"] == "ready"
 
     def test_root_redirect(self, test_client):
-        """Test that root path redirects to admin UI."""
+        """Test that root path behavior depends on UI configuration."""
         response = test_client.get("/", follow_redirects=False)
-        assert response.status_code == 303
-        assert response.headers["location"] == "/admin"
+        
+        # Check if UI is enabled
+        if settings.mcpgateway_ui_enabled:
+            # When UI is enabled, should redirect to admin
+            assert response.status_code == 303
+            assert response.headers["location"] == "/admin"
+        else:
+            # When UI is disabled, should return API info
+            assert response.status_code == 200
+            data = response.json()
+            assert data["name"] == "MCP_Gateway"
+            assert data["ui_enabled"] is False
 
     def test_static_files(self, test_client):
         """Test static file serving (when files don't exist)."""

tests/unit/mcpgateway/test_ui_version.py

@@ -67,10 +67,12 @@ def auth_headers() -> Dict[str, str]:
 #     assert "App:" in html or "Application:" in html
 
 
+@pytest.mark.skipif(
+    not settings.mcpgateway_ui_enabled,
+    reason="Admin UI tests require MCPGATEWAY_UI_ENABLED=true"
+)
 def test_admin_ui_contains_version_tab(test_client: TestClient, auth_headers: Dict[str, str]):
-    """
-    The Admin dashboard must contain the "Version & Environment Info" tab.
-    """
+    """The Admin dashboard must contain the "Version & Environment Info" tab."""
     resp = test_client.get("/admin", headers=auth_headers)
     assert resp.status_code == 200
     assert 'id="tab-version-info"' in resp.text