fix(mcp): simplify health_check tool and refactor system utils (#36063)

This commit is contained in:
Amin Ghadersohi
2025-11-13 04:28:09 +10:00
committed by GitHub
parent bb2e2a5ed6
commit c244e7f847
8 changed files with 314 additions and 177 deletions

View File

@@ -0,0 +1,55 @@
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
"""Tests for health_check MCP tool."""
from superset.mcp_service.system.schemas import HealthCheckResponse
def test_health_check_response_schema():
"""Test that HealthCheckResponse has required fields."""
response = HealthCheckResponse(
status="healthy",
timestamp="2025-11-10T19:00:00",
service="Test MCP Service",
version="4.0.0",
python_version="3.11.0",
platform="Darwin",
)
assert response.status == "healthy"
assert response.service == "Test MCP Service"
assert response.version == "4.0.0"
assert response.python_version == "3.11.0"
assert response.platform == "Darwin"
assert response.timestamp is not None
assert response.uptime_seconds is None # Optional field
def test_health_check_response_with_uptime():
"""Test HealthCheckResponse with optional uptime field."""
response = HealthCheckResponse(
status="healthy",
timestamp="2025-11-10T19:00:00",
service="Test MCP Service",
version="4.0.0",
python_version="3.11.0",
platform="Darwin",
uptime_seconds=123.45,
)
assert response.uptime_seconds == 123.45