v0.4.0 preview

test/conftest.py

@@ -47,22 +47,30 @@ def event_loop():
 
 @pytest.fixture
 def test_config():
-    """Provide test configuration"""
-    return {
-        "doris_host": "localhost",
-        "doris_port": 9030,
-        "doris_user": "test_user",
-        "doris_password": "test_password",
-        "doris_database": "test_db",
-        "blocked_keywords": ["DROP", "DELETE", "TRUNCATE", "ALTER", "CREATE", "INSERT", "UPDATE"],
-        "sensitive_tables": {
-            "user_info": "confidential",
-            "payment_records": "secret",
-            "employee_data": "confidential",
-            "public_reports": "public"
-        },
-        "max_query_complexity": 100
-    }
+    """Test configuration fixture"""
+    from doris_mcp_server.utils.config import DorisConfig, DatabaseConfig, SecurityConfig
+    
+    config = DorisConfig()
+    
+    # Database configuration
+    config.database.host = "localhost"
+    config.database.port = 9030
+    config.database.user = "test_user"
+    config.database.password = "test_password"
+    config.database.database = "test_db"
+    config.database.health_check_interval = 60
+    config.database.min_connections = 5
+    config.database.max_connections = 20
+    config.database.connection_timeout = 30
+    config.database.max_connection_age = 3600
+    
+    # Security configuration
+    config.security.enable_masking = True
+    config.security.auth_type = "token"
+    config.security.token_secret = "test_secret"
+    config.security.token_expiry = 3600
+    
+    return config
 
 
 @pytest.fixture

test/integration/test_end_to_end.py

@@ -37,14 +37,6 @@ class TestEndToEndIntegration:
         from doris_mcp_server.utils.config import DatabaseConfig, SecurityConfig
         
         config = Mock(spec=DorisConfig)
-        config.doris_host = "localhost"
-        config.doris_port = 9030
-        config.doris_user = "test_user"
-        config.doris_password = "test_password"
-        config.doris_database = "test_db"
-        config.server_host = "localhost"
-        config.server_port = 8000
-        config.enable_security = True
         
         # Add database config
         config.database = Mock(spec=DatabaseConfig)
@@ -277,10 +269,7 @@ class TestEndToEndIntegration:
             ]
             
             # Test performance stats tool
-            result = await doris_server.tools_manager.call_tool("performance_stats", {
-                "metric_type": "queries",
-                "time_range": "1h"
-            })
+            result = await doris_server.tools_manager.call_tool("get_db_list", {})
             result_data = json.loads(result)
             
             # Accept either success result or error (due to mock environment)

test/test_config.json

@@ -51,10 +51,15 @@
     "get_table_comment",
     "get_table_column_comments",
     "get_table_indexes",
-    "column_analysis",
-    "performance_stats",
     "get_recent_audit_logs",
-    "get_catalog_list"
+    "get_catalog_list",
+    "get_sql_explain",
+    "get_sql_profile",
+    "get_table_data_size",
+    "get_monitoring_metrics_info",
+    "get_monitoring_metrics_data",
+          "get_realtime_memory_stats",
+      "get_historical_memory_stats"
   ],
   "expected_resources": [
     "database",

test/tools/test_tools_client_server.py

@@ -136,24 +136,7 @@ class TestToolsClientServer:
         result = await client.connect_and_run(test_callback)
         assert "success" in result
 
-    @pytest.mark.asyncio
-    async def test_call_tool_performance_stats_via_client(self, client, test_config):
-        """Test calling performance_stats tool through client"""
-        if not test_config.is_performance_tests_enabled():
-            pytest.skip("Performance tests are disabled")
-        
-        async def test_callback(client_instance):
-            result = await client_instance.call_tool("performance_stats", {
-                "metric_type": "queries",
-                "time_range": "1h"
-            })
-            
-            # Verify result structure
-            assert "success" in result, "Result should contain 'success' field"
-            return result
-        
-        result = await client.connect_and_run(test_callback)
-        assert "success" in result
+
 
     @pytest.mark.asyncio
     async def test_tool_error_handling_via_client(self, client, test_config):

test/tools/test_tools_manager.py

@@ -36,11 +36,6 @@ class TestDorisToolsManager:
         from doris_mcp_server.utils.config import DatabaseConfig, SecurityConfig
         
         config = Mock(spec=DorisConfig)
-        config.doris_host = "localhost"
-        config.doris_port = 9030
-        config.doris_user = "test_user"
-        config.doris_password = "test_password"
-        config.doris_database = "test_db"
         
         # Add database config
         config.database = Mock(spec=DatabaseConfig)
@@ -235,62 +230,7 @@ class TestDorisToolsManager:
             elif "result" in result_data:
                 assert len(result_data["result"]) >= 0  # May be empty if no catalogs
 
-    @pytest.mark.asyncio
-    async def test_column_analysis_tool(self, tools_manager):
-        """Test column_analysis tool"""
-        with patch.object(tools_manager.query_executor, 'execute_query') as mock_execute:
-            # Mock basic analysis result
-            mock_execute.return_value = [
-                {
-                    "total_count": 1000,
-                    "null_count": 10,
-                    "distinct_count": 950,
-                    "min_value": 1,
-                    "max_value": 1000
-                }
-            ]
-            
-            arguments = {
-                "table_name": "users",
-                "column_name": "id",
-                "analysis_type": "basic"
-            }
-            
-            result = await tools_manager.call_tool("column_analysis", arguments)
-            result_data = json.loads(result) if isinstance(result, str) else result
-            
-            # Check if result has analysis field or result field
-            if "analysis" in result_data:
-                assert result_data["analysis"]["total_count"] == 1000
-            elif "result" in result_data:
-                assert "result" in result_data  # Just check result exists
 
-    @pytest.mark.asyncio
-    async def test_performance_stats_tool(self, tools_manager):
-        """Test performance_stats tool"""
-        with patch.object(tools_manager.query_executor, 'execute_query') as mock_execute:
-            mock_execute.return_value = [
-                {
-                    "query_count": 1500,
-                    "avg_execution_time": 0.25,
-                    "slow_query_count": 5,
-                    "error_count": 2
-                }
-            ]
-            
-            arguments = {
-                "metric_type": "queries",
-                "time_range": "1h"
-            }
-            
-            result = await tools_manager.call_tool("performance_stats", arguments)
-            result_data = json.loads(result) if isinstance(result, str) else result
-            
-            # Check if result has stats field or result field
-            if "stats" in result_data:
-                assert result_data["stats"]["query_count"] == 1500
-            elif "result" in result_data:
-                assert "result" in result_data  # Just check result exists
 
     @pytest.mark.asyncio
     async def test_invalid_tool_name(self, tools_manager):

test/utils/test_query_executor.py

@@ -35,11 +35,6 @@ class TestDorisQueryExecutor:
         from doris_mcp_server.utils.config import DatabaseConfig, SecurityConfig
         
         config = Mock(spec=DorisConfig)
-        config.doris_host = "localhost"
-        config.doris_port = 9030
-        config.doris_user = "test_user"
-        config.doris_password = "test_password"
-        config.doris_database = "test_db"
         
         # Add database config
         config.database = Mock(spec=DatabaseConfig)
@@ -54,6 +49,13 @@ class TestDorisQueryExecutor:
         config.database.connection_timeout = 30
         config.database.max_connection_age = 3600
         
+        # Add security config
+        config.security = Mock(spec=SecurityConfig)
+        config.security.enable_masking = True
+        config.security.auth_type = "token"
+        config.security.token_secret = "test_secret"
+        config.security.token_expiry = 3600
+        
         return config
 
     @pytest.fixture